Skip to content

Commit

Permalink
remove unnecessary websockets imports
Browse files Browse the repository at this point in the history
  • Loading branch information
Taylor Lowery committed Jun 5, 2024
1 parent 9870336 commit 6584f54
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tests/test_aiohttp_websocket_exceptions.py
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,6 @@ async def server_closing_directly(ws, path):
@pytest.mark.asyncio
@pytest.mark.parametrize("server", [server_closing_directly], indirect=True)
async def test_aiohttp_websocket_server_closing_directly(event_loop, server):
import websockets

from gql.transport.aiohttp_websockets import AIOHTTPWebsocketsTransport

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

sample_transport = AIOHTTPWebsocketsTransport(url=url)

with pytest.raises(websockets.exceptions.ConnectionClosed):
with pytest.raises(ConnectionResetError):
async with Client(transport=sample_transport):
pass

Expand All @@ -307,8 +306,6 @@ async def test_aiohttp_websocket_server_closing_after_ack(
event_loop, aiohttp_client_and_server
):

import websockets

session, server = aiohttp_client_and_server

query = gql("query { hello }")
Expand Down

0 comments on commit 6584f54

Please sign in to comment.