Use CountDownLatch for the test

reactor · Mar 13, 2019 · 42f0b52 · 42f0b52
1 parent 027e220
commit 42f0b52
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/test/java/reactor/netty/tcp/TcpClientTests.java b/src/test/java/reactor/netty/tcp/TcpClientTests.java
@@ -836,15 +836,15 @@ private void doTestIssue600(boolean withLoop) {
 	}
 
 	@Test
-	public void testRetryOnDifferentAddress() {
+	public void testRetryOnDifferentAddress() throws Exception {
 		DisposableServer server =
 				TcpServer.create()
 				         .port(0)
 				         .wiretap(true)
 				         .handle((req, res) -> res.sendString(Mono.just("test")))
 				         .bindNow();
 
-		final AtomicBoolean connected = new AtomicBoolean(false);
+		final CountDownLatch latch = new CountDownLatch(1);
 
 		Supplier<SocketAddress> addressSupplier = new Supplier<SocketAddress>() {
 			int i = 2;
@@ -858,7 +858,7 @@ public SocketAddress get() {
 		Connection  conn =
 				TcpClient.create()
 				         .addressSupplier(addressSupplier)
-				         .doOnConnected(connection -> connected.set(true))
+				         .doOnConnected(connection -> latch.countDown())
 				         .option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 100)
 				         .handle((in, out) -> Mono.never())
 				         .wiretap(true)
@@ -867,7 +867,7 @@ public SocketAddress get() {
 				         .block(Duration.ofSeconds(30));
 		assertNotNull(conn);
 
-		assertTrue(connected.get());
+		assertTrue(latch.await(30, TimeUnit.SECONDS));
 
 		conn.disposeNow();
 		server.disposeNow();