diff --git a/artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/SoleLibrarySystemTest.java b/artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/SoleLibrarySystemTest.java index b80a09386f..c1108249f2 100644 --- a/artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/SoleLibrarySystemTest.java +++ b/artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/SoleLibrarySystemTest.java @@ -219,21 +219,27 @@ public void shouldInitiatingLibraryDisconnectSessionOnLibraryTimeout() testSystem.remove(initiatingLibrary); awaitLibraryDisconnect(initiatingEngine, testSystem); - assertEventuallyTrue("Accepting library did not recognize disconnected session", () -> { - testSystem.poll(); - final List sessions = acceptingLibrary.sessions(); - assertEquals(1, sessions.size()); - final Session session = sessions.get(0); - assertEquals(SessionState.DISCONNECTED, session.state()); - }); - - assertEventuallyTrue("Initiating Engine did not disconnect session", () -> { - final Reply> libraryInfoReply = initiatingEngine.libraries(); - assertTrue(libraryInfoReply.hasCompleted()); - final List libraryInfo = libraryInfoReply.resultIfPresent(); - assertEquals(1, libraryInfo.size()); - final LibraryInfo libInfo = libraryInfo.get(0); - assertEquals(0, libInfo.sessions().size()); - }); + assertEventuallyTrue("Accepting library did not recognize disconnected session", + () -> + { + testSystem.poll(); + final List sessions = acceptingLibrary.sessions(); + assertEquals(1, sessions.size()); + final Session session = sessions.get(0); + assertEquals(SessionState.DISCONNECTED, session.state()); + } + ); + + assertEventuallyTrue("Initiating Engine did not disconnect session", + () -> + { + final Reply> libraryInfoReply = initiatingEngine.libraries(); + assertTrue(libraryInfoReply.hasCompleted()); + final List libraryInfo = libraryInfoReply.resultIfPresent(); + assertEquals(1, libraryInfo.size()); + final LibraryInfo libInfo = libraryInfo.get(0); + assertEquals(0, libInfo.sessions().size()); + } + ); } }