Skip to content

Commit 9870336

Browse files
author
Taylor Lowery
committed
wip: updating tests to follow AIOHTTPWebsockets protocol
1 parent ef84507 commit 9870336

File tree

4 files changed

+3
-4
lines changed

4 files changed

+3
-4
lines changed

gql/transport/aiohttp_websockets.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,6 @@ def __init__(
7373
ping_interval: Optional[Union[int, float]] = None,
7474
pong_timeout: Optional[Union[int, float]] = None,
7575
answer_pings: bool = True,
76-
7776
) -> None:
7877
self.url: StrOrURL = url
7978
self.headers: Optional[LooseHeaders] = headers

tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,7 @@ async def client_and_server(server):
460460
async with Client(transport=sample_transport) as session:
461461

462462
# Yield both client session and server
463-
yield session, server@pytest_asyncio.fixture
463+
yield session, server
464464

465465
@pytest_asyncio.fixture
466466
async def aiohttp_client_and_server(server):

tests/test_aiohttp_websocket_exceptions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ async def test_aiohttp_websocket_sending_invalid_data(
149149

150150
invalid_data = "QSDF"
151151
print(f">>> {invalid_data}")
152-
await session.transport.websocket.send(invalid_data)
152+
await session.transport.websocket.send_str(invalid_data)
153153

154154
await asyncio.sleep(2 * MS)
155155

@@ -313,7 +313,7 @@ async def test_aiohttp_websocket_server_closing_after_ack(
313313

314314
query = gql("query { hello }")
315315

316-
with pytest.raises(websockets.exceptions.ConnectionClosed):
316+
with pytest.raises(ConnectionResetError):
317317
await session.execute(query)
318318

319319
await session.transport.wait_closed()

tests/test_aiohttp_websocket_online.py

Whitespace-only changes.

0 commit comments

Comments
 (0)