Fixed flaky RealWebSocketTest.serverCloseThenClientClose(). (square#5800

) square#5708
yoxjames · Feb 23, 2020 · 2873719 · 2873719
1 parent 8e1f628
commit 2873719
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/okhttp/src/test/java/okhttp3/internal/ws/RealWebSocketTest.java b/okhttp/src/test/java/okhttp3/internal/ws/RealWebSocketTest.java
@@ -130,9 +130,15 @@ public final class RealWebSocketTest {
 
   @Test public void serverCloseThenClientClose() throws IOException {
     server.webSocket.close(1000, "Hello!");
+
     client.processNextFrame();
     client.listener.assertClosing(1000, "Hello!");
     assertThat(client.webSocket.close(1000, "Bye!")).isTrue();
+    client.listener.assertClosed(1000, "Hello!");
+
+    server.processNextFrame();
+    server.listener.assertClosing(1000, "Bye!");
+    server.listener.assertClosed(1000, "Bye!");
   }
 
   @Test public void emptyCloseInitiatesShutdown() throws IOException {