2 * Copyright (c) 2014, 2017 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
8 package org.opendaylight.aaa.datastore.h2;
10 import static org.junit.Assert.assertTrue;
11 import static org.mockito.ArgumentMatchers.anyString;
12 import static org.mockito.Mockito.mock;
13 import static org.mockito.Mockito.verify;
15 import java.sql.Connection;
16 import java.sql.DatabaseMetaData;
17 import java.sql.ResultSet;
18 import java.sql.SQLException;
19 import java.sql.Statement;
20 import org.junit.Test;
21 import org.mockito.Mockito;
22 import org.opendaylight.aaa.api.model.Roles;
24 public class RoleStoreTest {
26 private final Connection connectionMock = mock(Connection.class);
28 private final ConnectionProvider connectionFactoryMock = () -> connectionMock;
30 private final RoleStore roleStoreUnderTest = new RoleStore(connectionFactoryMock);
33 public void getRolesTest() throws SQLException, Exception {
34 // Setup Mock Behavior
35 String[] tableTypes = { "TABLE" };
36 Mockito.when(connectionMock.isClosed()).thenReturn(false);
37 DatabaseMetaData dbmMock = mock(DatabaseMetaData.class);
38 Mockito.when(connectionMock.getMetaData()).thenReturn(dbmMock);
39 ResultSet rsUserMock = mock(ResultSet.class);
40 Mockito.when(dbmMock.getTables(null, null, "ROLES", tableTypes)).thenReturn(rsUserMock);
41 Mockito.when(rsUserMock.next()).thenReturn(true);
43 Statement stmtMock = mock(Statement.class);
44 Mockito.when(connectionMock.createStatement()).thenReturn(stmtMock);
46 ResultSet rsMock = getMockedResultSet();
47 Mockito.when(stmtMock.executeQuery(anyString())).thenReturn(rsMock);
50 Roles roles = roleStoreUnderTest.getRoles();
53 assertTrue(roles.getRoles().size() == 1);
54 verify(stmtMock).close();
57 public ResultSet getMockedResultSet() throws SQLException {
58 ResultSet rsMock = mock(ResultSet.class);
59 Mockito.when(rsMock.next()).thenReturn(true).thenReturn(false);
60 Mockito.when(rsMock.getInt(RoleStore.SQL_ID)).thenReturn(1);
61 Mockito.when(rsMock.getString(RoleStore.SQL_NAME)).thenReturn("RoleName_1");
62 Mockito.when(rsMock.getString(RoleStore.SQL_DESCR)).thenReturn("Desc_1");