@Test public void testDiscardExpiredConnections() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(2, 1, connector); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter1::isSuccess); conn.recycle(2L); assertEquals(1, mgr.size()); assertEquals(0, mgr.removeExpired(1L)); assertEquals(1, mgr.size()); assertEquals(1, mgr.removeExpired(2L)); assertEquals(0, mgr.size()); }
assertEquals(1, mgr.closeCount()); assertEquals(0, mgr.size()); assertEquals(0, mgr.pool.waitersInQueue()); assertEquals(0, mgr.pool.weight()); assertEquals(0, mgr.pool.capacity());
@Test public void testConnectFailure() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(3, 4, connector); AtomicReference<Boolean> holdsLock = new AtomicReference<>(); FakeWaiter waiter = new FakeWaiter() { @Override public synchronized void handleFailure(Throwable failure) { assertNull(Vertx.currentContext()); Pool<FakeConnection> pool = mgr.pool(); holdsLock.set(Thread.holdsLock(pool)); super.handleFailure(failure); } }; mgr.getConnection(waiter); FakeConnection conn = connector.assertRequest(); Throwable failure = new Throwable(); conn.fail(failure); assertWaitUntil(waiter::isComplete); assertEquals(Boolean.FALSE, holdsLock.get()); waiter.assertFailure(failure); }
@Test public void testConnectSuccess() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(3, 4, connector); AtomicReference<Boolean> handleLock = new AtomicReference<>(); FakeWaiter waiter = new FakeWaiter() { @Override public synchronized void handleConnection(FakeConnection conn) { assertSame(conn.context, mgr.context); Pool<FakeConnection> pool = mgr.pool(); handleLock.set(Thread.holdsLock(pool)); super.handleConnection(conn); } }; mgr.getConnection(waiter); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter::isComplete); assertEquals(Boolean.FALSE, handleLock.get()); waiter.assertSuccess(conn); waiter.recycle(); assertEquals(0, mgr.size()); assertWaitUntil(() -> mgr.closed()); }
@Test public void testSynchronousConnectionFailure() { Throwable cause = new Throwable(); ConnectionProvider<FakeConnection> connector = new FakeConnectionProviderBase() { @Override public void connect(ConnectionListener<FakeConnection> listener, ContextInternal context, Handler<AsyncResult<ConnectResult<FakeConnection>>> handler) { handler.handle(Future.failedFuture(cause)); } }; FakeConnectionManager mgr = new FakeConnectionManager(3, 3, connector); for (int i = 0;i < 4;i++) { FakeWaiter waiter = new FakeWaiter(); mgr.getConnection(waiter); waitUntil(waiter::isFailure); waiter.assertFailure(cause); assertEquals(0, mgr.pool().weight()); } assertTrue(mgr.closed()); }
@Test public void testEndpointLifecycle() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(3, 1, connector); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter1::isSuccess); conn.close(); assertWaitUntil(mgr::closed); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); assertEquals(2, mgr.sequence()); }
@Test public void testRecycleWithoutDispose() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(-1, 1, connector); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter1::isSuccess); conn.recycle(false); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); assertWaitUntil(waiter2::isSuccess); waiter2.assertSuccess(conn); conn.recycle(true); assertEquals(0, mgr.size()); }
@Test public void testRecycleFIFO() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(-1, 2, connector, true); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection firstInConnection = connector.assertRequest(); firstInConnection.connect(); assertWaitUntil(waiter1::isSuccess); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); FakeConnection lastInConnection = connector.assertRequest(); lastInConnection.connect(); assertWaitUntil(waiter2::isSuccess); waiter2.assertSuccess(lastInConnection); firstInConnection.recycle(false); lastInConnection.recycle(false); assertEquals(2, mgr.size()); FakeWaiter waiter3 = new FakeWaiter(); mgr.getConnection(waiter3); assertWaitUntil(waiter3::isSuccess); waiter3.assertSuccess(firstInConnection); }
@Test public void testRecycleLIFO() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(-1, 2, connector, false); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection firstInConnection = connector.assertRequest(); firstInConnection.connect(); assertWaitUntil(waiter1::isSuccess); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); FakeConnection lastInConnection = connector.assertRequest(); lastInConnection.connect(); assertWaitUntil(waiter2::isSuccess); waiter2.assertSuccess(lastInConnection); firstInConnection.recycle(false); lastInConnection.recycle(false); assertEquals(2, mgr.size()); FakeWaiter waiter3 = new FakeWaiter(); mgr.getConnection(waiter3); assertWaitUntil(waiter3::isSuccess); waiter3.assertSuccess(lastInConnection); }
@Test public void testDiscardExpiredConnections() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(2, 1, connector); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter1::isSuccess); conn.recycle(2L); assertEquals(1, mgr.size()); assertEquals(0, mgr.removeExpired(1L)); assertEquals(1, mgr.size()); assertEquals(1, mgr.removeExpired(2L)); assertEquals(0, mgr.size()); }
assertEquals(1, mgr.closeCount()); assertEquals(0, mgr.size()); assertEquals(0, mgr.pool.waitersInQueue()); assertEquals(0, mgr.pool.weight()); assertEquals(0, mgr.pool.capacity());
@Test public void testConnectFailure() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(3, 4, connector); AtomicReference<Boolean> holdsLock = new AtomicReference<>(); FakeWaiter waiter = new FakeWaiter() { @Override public synchronized void handleFailure(Throwable failure) { assertNull(Vertx.currentContext()); Pool<FakeConnection> pool = mgr.pool(); holdsLock.set(Thread.holdsLock(pool)); super.handleFailure(failure); } }; mgr.getConnection(waiter); FakeConnection conn = connector.assertRequest(); Throwable failure = new Throwable(); conn.fail(failure); assertWaitUntil(waiter::isComplete); assertEquals(Boolean.FALSE, holdsLock.get()); waiter.assertFailure(failure); }
@Test public void testConnectSuccess() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(3, 4, connector); AtomicReference<Boolean> handleLock = new AtomicReference<>(); FakeWaiter waiter = new FakeWaiter() { @Override public synchronized void handleConnection(FakeConnection conn) { assertSame(conn.context, mgr.context); Pool<FakeConnection> pool = mgr.pool(); handleLock.set(Thread.holdsLock(pool)); super.handleConnection(conn); } }; mgr.getConnection(waiter); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter::isComplete); assertEquals(Boolean.FALSE, handleLock.get()); waiter.assertSuccess(conn); waiter.recycle(); assertEquals(0, mgr.size()); assertWaitUntil(() -> mgr.closed()); }
@Test public void testSynchronousConnectionFailure() { Throwable cause = new Throwable(); ConnectionProvider<FakeConnection> connector = new FakeConnectionProviderBase() { @Override public void connect(ConnectionListener<FakeConnection> listener, ContextInternal context, Handler<AsyncResult<ConnectResult<FakeConnection>>> handler) { handler.handle(Future.failedFuture(cause)); } }; FakeConnectionManager mgr = new FakeConnectionManager(3, 3, connector); for (int i = 0;i < 4;i++) { FakeWaiter waiter = new FakeWaiter(); mgr.getConnection(waiter); waitUntil(waiter::isFailure); waiter.assertFailure(cause); assertEquals(0, mgr.pool().weight()); } assertTrue(mgr.closed()); }
@Test public void testEndpointLifecycle() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(3, 1, connector); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter1::isSuccess); conn.close(); assertWaitUntil(mgr::closed); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); assertEquals(2, mgr.sequence()); }
@Test public void testRecycleWithoutDispose() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(-1, 1, connector); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection conn = connector.assertRequest(); conn.connect(); assertWaitUntil(waiter1::isSuccess); conn.recycle(false); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); assertWaitUntil(waiter2::isSuccess); waiter2.assertSuccess(conn); conn.recycle(true); assertEquals(0, mgr.size()); }
@Test public void testRecycleFIFO() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(-1, 2, connector, true); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection firstInConnection = connector.assertRequest(); firstInConnection.connect(); assertWaitUntil(waiter1::isSuccess); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); FakeConnection lastInConnection = connector.assertRequest(); lastInConnection.connect(); assertWaitUntil(waiter2::isSuccess); waiter2.assertSuccess(lastInConnection); firstInConnection.recycle(false); lastInConnection.recycle(false); assertEquals(2, mgr.size()); FakeWaiter waiter3 = new FakeWaiter(); mgr.getConnection(waiter3); assertWaitUntil(waiter3::isSuccess); waiter3.assertSuccess(firstInConnection); }
@Test public void testRecycleLIFO() { FakeConnectionProvider connector = new FakeConnectionProvider(); FakeConnectionManager mgr = new FakeConnectionManager(-1, 2, connector, false); FakeWaiter waiter1 = new FakeWaiter(); mgr.getConnection(waiter1); FakeConnection firstInConnection = connector.assertRequest(); firstInConnection.connect(); assertWaitUntil(waiter1::isSuccess); FakeWaiter waiter2 = new FakeWaiter(); mgr.getConnection(waiter2); FakeConnection lastInConnection = connector.assertRequest(); lastInConnection.connect(); assertWaitUntil(waiter2::isSuccess); waiter2.assertSuccess(lastInConnection); firstInConnection.recycle(false); lastInConnection.recycle(false); assertEquals(2, mgr.size()); FakeWaiter waiter3 = new FakeWaiter(); mgr.getConnection(waiter3); assertWaitUntil(waiter3::isSuccess); waiter3.assertSuccess(lastInConnection); }