Skip to content

Commit 6584f54

Browse files
author
Taylor Lowery
committed
remove unnecessary websockets imports
1 parent 9870336 commit 6584f54

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/test_aiohttp_websocket_exceptions.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,6 @@ async def server_closing_directly(ws, path):
282282
@pytest.mark.asyncio
283283
@pytest.mark.parametrize("server", [server_closing_directly], indirect=True)
284284
async def test_aiohttp_websocket_server_closing_directly(event_loop, server):
285-
import websockets
286285

287286
from gql.transport.aiohttp_websockets import AIOHTTPWebsocketsTransport
288287

@@ -291,7 +290,7 @@ async def test_aiohttp_websocket_server_closing_directly(event_loop, server):
291290

292291
sample_transport = AIOHTTPWebsocketsTransport(url=url)
293292

294-
with pytest.raises(websockets.exceptions.ConnectionClosed):
293+
with pytest.raises(ConnectionResetError):
295294
async with Client(transport=sample_transport):
296295
pass
297296

@@ -307,8 +306,6 @@ async def test_aiohttp_websocket_server_closing_after_ack(
307306
event_loop, aiohttp_client_and_server
308307
):
309308

310-
import websockets
311-
312309
session, server = aiohttp_client_and_server
313310

314311
query = gql("query { hello }")

0 commit comments

Comments
 (0)