Examples with JDBCConnection com.yahoo.athenz.zms.store.jdbc.JDBCConnection used on opensource projects

Example 66 with JDBCConnection

use of com.yahoo.athenz.zms.store.jdbc.JDBCConnection in project athenz by yahoo.

the class JDBCConnectionTest method testDeletePrincipalSubDomainException.

@Test
public void testDeletePrincipalSubDomainException() throws Exception {
    JDBCConnection jdbcConn = new JDBCConnection(mockConn, true);
    Mockito.when(mockPrepStmt.executeUpdate()).thenReturn(1).thenThrow(new SQLException("failed operation", "state", 1001));
    try {
        jdbcConn.deletePrincipal("user.jake", true);
        fail();
    } catch (ResourceException ex) {
        assertEquals(ex.getCode(), ResourceException.INTERNAL_SERVER_ERROR);
    }
    jdbcConn.close();
}

Example 67 with JDBCConnection

use of com.yahoo.athenz.zms.store.jdbc.JDBCConnection in project athenz by yahoo.

the class JDBCConnectionTest method testListRoles.

@Test
public void testListRoles() throws Exception {
    JDBCConnection jdbcConn = new JDBCConnection(mockConn, true);
    // return domain id
    Mockito.doReturn(5).when(mockResultSet).getInt(1);
    Mockito.when(mockResultSet.next()).thenReturn(// this one is for domain id
    true).thenReturn(true).thenReturn(true).thenReturn(true).thenReturn(false);
    Mockito.when(mockResultSet.getString(1)).thenReturn("zrole").thenReturn("arole").thenReturn("brole");
    List<String> roles = jdbcConn.listRoles("my-domain");
    // data back is sorted
    assertEquals(3, roles.size());
    assertEquals("arole", roles.get(0));
    assertEquals("brole", roles.get(1));
    assertEquals("zrole", roles.get(2));
    jdbcConn.close();
}

Example 68 with JDBCConnection

use of com.yahoo.athenz.zms.store.jdbc.JDBCConnection in project athenz by yahoo.

the class JDBCConnectionTest method testUpdateDomainModTimestampFailure.

@Test
public void testUpdateDomainModTimestampFailure() throws Exception {
    JDBCConnection jdbcConn = new JDBCConnection(mockConn, true);
    Mockito.doReturn(0).when(mockPrepStmt).executeUpdate();
    boolean requestSuccess = jdbcConn.updateDomainModTimestamp("my-domain");
    assertFalse(requestSuccess);
    Mockito.verify(mockPrepStmt, times(1)).setString(1, "my-domain");
    jdbcConn.close();
}

Example 69 with JDBCConnection

use of com.yahoo.athenz.zms.store.jdbc.JDBCConnection in project athenz by yahoo.

the class JDBCConnectionTest method testCountPoliciesInvalidDomain.

@Test
public void testCountPoliciesInvalidDomain() throws Exception {
    JDBCConnection jdbcConn = new JDBCConnection(mockConn, true);
    Mockito.when(mockResultSet.next()).thenReturn(// this one is for domain id
    false);
    try {
        jdbcConn.countPolicies("my-domain");
    } catch (Exception ex) {
        assertTrue(true);
    }
    jdbcConn.close();
}

Example 70 with JDBCConnection

use of com.yahoo.athenz.zms.store.jdbc.JDBCConnection in project athenz by yahoo.

the class JDBCConnectionTest method testInsertServiceHostInvalidService.

@Test
public void testInsertServiceHostInvalidService() throws Exception {
    JDBCConnection jdbcConn = new JDBCConnection(mockConn, true);
    Mockito.when(mockResultSet.getInt(1)).thenReturn(// domain id
    5);
    Mockito.when(mockResultSet.next()).thenReturn(// this one is for domain id
    true).thenReturn(// this one is for service id
    false);
    try {
        jdbcConn.insertServiceHost("my-domain", "service1", "host1");
        fail();
    } catch (Exception ex) {
        assertTrue(true);
    }
    jdbcConn.close();
}