flaky test fixed

undertow-io · Nov 17, 2024 · 9eef7d9 · 9eef7d9
1 parent b44a1c3
commit 9eef7d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.../test/java/io/undertow/websockets/jsr/test/reconnect/ClientEndpointReconnectTestCase.java b/.../test/java/io/undertow/websockets/jsr/test/reconnect/ClientEndpointReconnectTestCase.java
@@ -113,6 +113,7 @@ public void testAnnotatedClientEndpoint() throws Exception {
         session.getBasicRemote().sendText("close");
         Assert.assertEquals("CLOSE", endpoint.message());
         Assert.assertEquals("OPEN", endpoint.message());
+        Assert.assertTrue(session.isOpen()); // Verify session is open before sending the next message
         session.getBasicRemote().sendText("hi");
         Assert.assertEquals("MESSAGE-ECHO-hi", endpoint.message());
         session.getBasicRemote().sendText("close");