2 * Copyright (c) 2014, 2015 Hewlett-Packard Development Company, L.P. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.aaa.idm.persistence;
11 import static org.junit.Assert.*;
12 import static org.mockito.Mockito.*;
13 import static org.mockito.Matchers.*;
15 import java.sql.Statement;
16 import java.sql.Connection;
17 import java.sql.DatabaseMetaData;
18 import java.sql.ResultSet;
19 import java.sql.SQLException;
21 import org.junit.After;
22 import org.junit.AfterClass;
23 import org.junit.Before;
24 import org.junit.BeforeClass;
25 import org.junit.Test;
26 import org.mockito.Mockito;
27 import org.opendaylight.aaa.idm.model.Roles;
29 public class RoleStoreTest {
31 Connection connectionMock = mock(Connection.class);
32 private final RoleStore RoleStoreUnderTest = new RoleStore();
36 RoleStoreUnderTest.dbConnection = connectionMock;
40 public void teardown() {
45 public void getRolesTest() throws SQLException, Exception {
47 String[] tableTypes = {"TABLE"};
48 Mockito.when(connectionMock.isClosed()).thenReturn(false);
49 DatabaseMetaData dbmMock = mock(DatabaseMetaData.class);
50 Mockito.when(connectionMock.getMetaData()).thenReturn(dbmMock);
51 ResultSet rsUserMock = mock(ResultSet.class);
52 Mockito.when(dbmMock.getTables(null,null,"ROLES",tableTypes)).thenReturn(rsUserMock);
53 Mockito.when(rsUserMock.next()).thenReturn(true);
55 Statement stmtMock = mock(Statement.class);
56 Mockito.when(connectionMock.createStatement()).thenReturn(stmtMock);
58 ResultSet rsMock = getMockedResultSet();
59 Mockito.when(stmtMock.executeQuery(anyString())).thenReturn(rsMock);
62 Roles roles = RoleStoreUnderTest.getRoles();
65 assertTrue(roles.getRoles().size() == 1);
66 verify(stmtMock).close();
71 public ResultSet getMockedResultSet(){
72 ResultSet rsMock = mock(ResultSet.class);
74 Mockito.when(rsMock.next()).thenReturn(true).thenReturn(false);
75 Mockito.when(rsMock.getInt(RoleStore.SQL_ID)).thenReturn(1);
76 Mockito.when(rsMock.getString(RoleStore.SQL_NAME)).thenReturn("RoleName_1");
77 Mockito.when(rsMock.getString(RoleStore.SQL_DESCR)).thenReturn("Desc_1");
80 catch (SQLException e) {
81 // TODO Auto-generated catch block