test(test_ws.py): fix one async race potentially yielding unexpected …

…result
falconry · Oct 14, 2023 · fbac12a · fbac12a
1 parent 937ca9f
commit fbac12a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/asgi/test_ws.py b/tests/asgi/test_ws.py
@@ -370,6 +370,13 @@ async def on_websocket(self, req, ws):
                     await resource.data_received.wait()
                     assert resource.data == sample_data
 
+                # NOTE(vytas): When testing the case where the server
+                #   explicitly closes the connection, try to receive some data
+                #   before closing from the client side (and potentially
+                #   winning the async race of which side closes first).
+                if explicit_close_server:
+                    await ws.receive_data()
+
                 if explicit_close_client:
                     await ws.close(4042)