From e4dbce8b66fdf96fc4853a3b4f1d072e9926b4c7 Mon Sep 17 00:00:00 2001 From: Micah Pegman Date: Mon, 10 Jun 2024 18:59:10 +0000 Subject: [PATCH] more test fixes Signed-off-by: Micah Pegman --- tests/test_aiohttp_websocket_exceptions.py | 5 ----- tests/test_aiohttp_websocket_query.py | 2 +- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/tests/test_aiohttp_websocket_exceptions.py b/tests/test_aiohttp_websocket_exceptions.py index d50ac887..48312544 100644 --- 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 index 3062e5e7..2899038e 100644 --- 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)