wip: updating tests to follow AIOHTTPWebsockets protocol

graphql-python · Jun 4, 2024 · 9870336 · 9870336
1 parent ef84507
commit 9870336
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 4 deletions.
diff --git a/gql/transport/aiohttp_websockets.py b/gql/transport/aiohttp_websockets.py
@@ -73,7 +73,6 @@ def __init__(
         ping_interval: Optional[Union[int, float]] = None,
         pong_timeout: Optional[Union[int, float]] = None,
         answer_pings: bool = True,
-
     ) -> None:
         self.url: StrOrURL = url
         self.headers: Optional[LooseHeaders] = headers

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -460,7 +460,7 @@ async def client_and_server(server):
     async with Client(transport=sample_transport) as session:
 
         # Yield both client session and server
-        yield session, server@pytest_asyncio.fixture
+        yield session, server
 
 @pytest_asyncio.fixture
 async def aiohttp_client_and_server(server):

diff --git a/tests/test_aiohttp_websocket_exceptions.py b/tests/test_aiohttp_websocket_exceptions.py
@@ -149,7 +149,7 @@ async def test_aiohttp_websocket_sending_invalid_data(
 
     invalid_data = "QSDF"
     print(f">>> {invalid_data}")
-    await session.transport.websocket.send(invalid_data)
+    await session.transport.websocket.send_str(invalid_data)
 
     await asyncio.sleep(2 * MS)
 
@@ -313,7 +313,7 @@ async def test_aiohttp_websocket_server_closing_after_ack(
 
     query = gql("query { hello }")
 
-    with pytest.raises(websockets.exceptions.ConnectionClosed):
+    with pytest.raises(ConnectionResetError):
         await session.execute(query)
 
     await session.transport.wait_closed()

diff --git a/tests/test_aiohttp_websocket_online.py b/tests/test_aiohttp_websocket_online.py