PROTON-2780 Remove leftover thread sleep from test

Remove an old thread sleep and add in test expecation for connection drop to fully validate client close on request.
apache · Dec 7, 2023 · f2e6298 · f2e6298
1 parent 052d7ef
commit f2e6298
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...-test-driver/src/test/java/org/apache/qpid/protonj2/test/driver/ProtonTestClientTest.java b/...-test-driver/src/test/java/org/apache/qpid/protonj2/test/driver/ProtonTestClientTest.java
@@ -44,6 +44,7 @@ class ProtonTestClientTest extends TestPeerTestsBase {
     public void testClientCanConnectAndExchangeAMQPHeaders() throws Exception {
         try (ProtonTestServer peer = new ProtonTestServer()) {
             peer.expectAMQPHeader().respondWithAMQPHeader();
+            peer.expectConnectionToDrop();
             peer.start();
 
             URI remoteURI = peer.getServerURI();
@@ -58,8 +59,6 @@ public void testClientCanConnectAndExchangeAMQPHeaders() throws Exception {
 
             LOG.info("Test started, peer listening on: {}", remoteURI);
 
-            Thread.sleep(100);
-
             peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
         }
     }