more test fixes

graphql-python · Jun 10, 2024 · 0452346 · 0452346
1 parent fc73ba1
commit 0452346
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/tests/test_aiohttp_websocket_exceptions.py b/tests/test_aiohttp_websocket_exceptions.py
@@ -310,11 +310,6 @@ async def test_aiohttp_websocket_server_closing_after_ack(
 
     query = gql("query { hello }")
 
-    with pytest.raises(ConnectionResetError):
-        await session.execute(query)
-
-    await session.transport.wait_closed()
-
     with pytest.raises(TransportClosed):
         await session.execute(query)
 

diff --git a/tests/test_aiohttp_websocket_query.py b/tests/test_aiohttp_websocket_query.py
@@ -245,7 +245,7 @@ async def test_aiohttp_websocket_server_closing_after_first_query(
 
     # Now the server is closed but we don't know it yet, we have to send a query
     # to notice it and to receive the exception
-    with pytest.raises(ConnectionResetError):
+    with pytest.raises(TransportClosed):
         await session.execute(query)