[Java] Fix assertion.

real-logic · Dec 20, 2024 · a41ee99 · a41ee99
1 parent 11f0617
commit a41ee99
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/aeron-archive/src/test/java/io/aeron/archive/ArchiveTest.java b/aeron-archive/src/test/java/io/aeron/archive/ArchiveTest.java
@@ -933,10 +933,15 @@ void shouldTimeoutInactiveArchiveClients(final String controlRequestChannel, fin
                 while (client2.controlResponsePoller().subscription().isConnected())
                 {
                     assertNull(client1.pollForErrorResponse());
-                    final String response = client2.pollForErrorResponse();
-                    if (null != response)
+                    try
                     {
-                        assertEquals(AeronArchive.NOT_CONNECTED_MSG, response);
+                        client2.checkForErrorResponse();
+                    }
+                    catch (final ArchiveException ex)
+                    {
+                        assertThat(ex.getMessage(), anyOf(
+                            equalTo("ERROR - client is closed"),
+                            equalTo("ERROR - not connected")));
                         break;
                     }
                 }