more test fixes

Author: mpegman-scwx

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)
 

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)