/**
* Tests that a connection gets to receive a keep-alive.
*
* @throws SQLException
* @throws InterruptedException
*/
@Test
public void testIdleConnectionIsSentKeepAlive() throws SQLException, InterruptedException {
ArrayBlockingQueue<ConnectionHandle> fakeFreeConnections =
new ArrayBlockingQueue<ConnectionHandle>(1);
fakeFreeConnections.add(mockConnection);
config.setIdleConnectionTestPeriod(1);
expect(mockPool.getConfig()).andReturn(config).anyTimes();
expect(mockConnectionPartition.getFreeConnections()).andReturn(fakeFreeConnections).anyTimes();
expect(mockConnectionPartition.getMinConnections()).andReturn(10).once();
expect(mockConnection.isPossiblyBroken()).andReturn(false);
expect(mockConnection.getConnectionLastUsed()).andReturn(0L);
expect(mockPool.isConnectionHandleAlive((ConnectionHandle) anyObject()))
.andReturn(true)
.anyTimes();
mockPool.releaseInAnyFreePartition(
(ConnectionHandle) anyObject(), (ConnectionPartition) anyObject());
// connection should be closed
mockConnection.setConnectionLastReset(anyLong());
replay(mockPool, mockConnection, mockConnectionPartition, mockExecutor);
this.testClass = new ConnectionTesterThread(mockConnectionPartition, mockExecutor, mockPool);
this.testClass.run();
verify(mockPool, mockConnectionPartition, mockExecutor, mockConnection);
}
/**
* Tests various getter/setters.
*
* @throws IllegalArgumentException
* @throws IllegalAccessException
* @throws SecurityException
* @throws NoSuchFieldException
*/
@SuppressWarnings("deprecation")
@Test
public void testSettersGetters()
throws IllegalArgumentException, IllegalAccessException, SecurityException,
NoSuchFieldException {
ConnectionPartition mockPartition = createNiceMock(ConnectionPartition.class);
testClass.setOriginatingPartition(mockPartition);
assertEquals(mockPartition, testClass.getOriginatingPartition());
testClass.setConnectionLastReset(123);
assertEquals(testClass.getConnectionLastReset(), 123);
testClass.setConnectionLastUsed(456);
assertEquals(testClass.getConnectionLastUsed(), 456);
Field field = testClass.getClass().getDeclaredField("possiblyBroken");
field.setAccessible(true);
field.setBoolean(testClass, true);
assertTrue(testClass.isPossiblyBroken());
Object debugHandle = new Object();
testClass.setDebugHandle(debugHandle);
assertEquals(debugHandle, testClass.getDebugHandle());
testClass.setInternalConnection(mockConnection);
assertEquals(mockConnection, testClass.getInternalConnection());
assertEquals(mockConnection, testClass.getRawConnection());
field = testClass.getClass().getDeclaredField("logicallyClosed");
field.setAccessible(true);
field.setBoolean(testClass, true);
assertTrue(testClass.isClosed());
testClass.setLogStatementsEnabled(true);
assertTrue(testClass.isLogStatementsEnabled());
assertEquals(testClass.getPool(), mockPool);
ArrayList<ReplayLog> testLog = new ArrayList<ReplayLog>();
testClass.setReplayLog(testLog);
assertEquals(testClass.getReplayLog(), testLog);
testClass.setInReplayMode(true);
assertTrue(testClass.isInReplayMode());
testClass.setInReplayMode(false);
testClass.threadUsingConnection = Thread.currentThread();
assertEquals(Thread.currentThread(), testClass.getThreadUsingConnection());
}