fix tests on Python 3.7

leszekhanusz · Oct 21, 2023 · 1bf3bcc · 1bf3bcc
1 parent cd57ef0
commit 1bf3bcc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_appsync_websockets.py b/tests/test_appsync_websockets.py
@@ -333,7 +333,10 @@ async def receiving_coro():
             print(f"\n            Server: Exception received: {e!s}\n")
         finally:
             print("            Server: waiting for websocket connection to close")
-            await ws.close()
+            try:
+                await asyncio.wait_for(ws.wait_closed(), 1000 * MS)
+            except TimeoutError:
+                pass
             print("            Server: connection closed")
 
     return realtime_appsync_server_template