Skip to content

Commit 6a004bb

Browse files
committed
tests: remove unused fixtures and local symbols
1 parent 52ce346 commit 6a004bb

File tree

1 file changed

+18
-24
lines changed

1 file changed

+18
-24
lines changed

tests/test_connection.py

Lines changed: 18 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ async def test_serve(nursery):
194194
# The server nursery begins with one task (server.listen).
195195
assert len(nursery.child_tasks) == 1
196196
no_clients_nursery_count = len(task.child_nurseries)
197-
async with open_websocket(HOST, port, RESOURCE, use_ssl=False) as conn:
197+
async with open_websocket(HOST, port, RESOURCE, use_ssl=False):
198198
# The server nursery has the same number of tasks, but there is now
199199
# one additional nested nursery.
200200
assert len(nursery.child_tasks) == 1
@@ -220,25 +220,22 @@ async def test_serve_ssl(nursery):
220220

221221

222222
async def test_serve_handler_nursery(nursery):
223-
task = current_task()
224223
async with trio.open_nursery() as handler_nursery:
225224
serve_with_nursery = partial(serve_websocket, echo_request_handler,
226225
HOST, 0, None, handler_nursery=handler_nursery)
227226
server = await nursery.start(serve_with_nursery)
228227
port = server.port
229228
# The server nursery begins with one task (server.listen).
230229
assert len(nursery.child_tasks) == 1
231-
no_clients_nursery_count = len(task.child_nurseries)
232-
async with open_websocket(HOST, port, RESOURCE, use_ssl=False) as conn:
230+
async with open_websocket(HOST, port, RESOURCE, use_ssl=False):
233231
# The handler nursery should have one task in it
234232
# (conn._reader_task).
235233
assert len(handler_nursery.child_tasks) == 1
236234

237235

238-
async def test_serve_with_zero_listeners(nursery):
239-
task = current_task()
236+
async def test_serve_with_zero_listeners():
240237
with pytest.raises(ValueError):
241-
server = WebSocketServer(echo_request_handler, [])
238+
WebSocketServer(echo_request_handler, [])
242239

243240

244241
async def test_serve_non_tcp_listener(nursery):
@@ -290,7 +287,7 @@ async def test_client_open_url(path, expected_path, echo_server):
290287

291288
async def test_client_open_invalid_url(echo_server):
292289
with pytest.raises(ValueError):
293-
async with open_websocket_url('http://foo.com/bar') as conn:
290+
async with open_websocket_url('http://foo.com/bar'):
294291
pass
295292

296293

@@ -358,11 +355,10 @@ async def handler(request):
358355
assert not request.local.is_ssl
359356
assert str(request.remote.address) == HOST
360357
assert not request.remote.is_ssl
361-
conn = await request.accept()
358+
await request.accept()
362359

363360
server = await nursery.start(serve_websocket, handler, HOST, 0, None)
364-
async with open_websocket(HOST, server.port, RESOURCE, use_ssl=False
365-
) as client_ws:
361+
async with open_websocket(HOST, server.port, RESOURCE, use_ssl=False):
366362
pass
367363

368364

@@ -410,7 +406,7 @@ async def handler(request):
410406
async def test_handshake_server_headers(nursery):
411407
async def handler(request):
412408
headers = [('X-Test-Header', 'My test header')]
413-
server_ws = await request.accept(extra_headers=headers)
409+
await request.accept(extra_headers=headers)
414410

415411
server = await nursery.start(serve_websocket, handler, HOST, 0, None)
416412
async with open_websocket(HOST, server.port, RESOURCE, use_ssl=False
@@ -433,7 +429,7 @@ async def handler(request):
433429
server = await nursery.start(serve_websocket, handler, HOST, 0,
434430
None)
435431
async with open_websocket(HOST, server.port, RESOURCE,
436-
use_ssl=False) as client_ws:
432+
use_ssl=False):
437433
pass
438434

439435

@@ -445,8 +441,7 @@ async def handler(request):
445441

446442
server = await nursery.start(serve_websocket, handler, HOST, 0, None)
447443
with pytest.raises(ConnectionRejected) as exc_info:
448-
async with open_websocket(HOST, server.port, RESOURCE, use_ssl=False,
449-
) as client_ws:
444+
async with open_websocket(HOST, server.port, RESOURCE, use_ssl=False):
450445
pass
451446
exc = exc_info.value
452447
assert exc.body == b'My body'
@@ -467,16 +462,15 @@ async def handler(stream):
467462
port = listeners[0].socket.getsockname()[1]
468463

469464
with pytest.raises(ConnectionRejected) as exc_info:
470-
async with open_websocket(HOST, port, RESOURCE, use_ssl=False,
471-
) as client_ws:
465+
async with open_websocket(HOST, port, RESOURCE, use_ssl=False):
472466
pass
473467
exc = exc_info.value
474468
assert exc.status_code == 100
475469
assert repr(exc) == 'ConnectionRejected<status_code=100>'
476470
assert exc.body is None
477471

478472

479-
async def test_handshake_protocol_error(nursery, echo_server):
473+
async def test_handshake_protocol_error(echo_server):
480474
'''
481475
If a client connects to a trio-websocket server and tries to speak HTTP
482476
instead of WebSocket, the server should reject the connection. (If the
@@ -604,15 +598,15 @@ async def test_client_open_timeout(nursery, autojump_clock):
604598
'''
605599
async def handler(request):
606600
await trio.sleep(FORCE_TIMEOUT)
607-
server_ws = await request.accept()
601+
await request.accept()
608602
pytest.fail('Should not reach this line.')
609603

610604
server = await nursery.start(
611605
partial(serve_websocket, handler, HOST, 0, ssl_context=None))
612606

613607
with pytest.raises(ConnectionTimeout):
614608
async with open_websocket(HOST, server.port, '/', use_ssl=False,
615-
connect_timeout=TIMEOUT) as client_ws:
609+
connect_timeout=TIMEOUT):
616610
pass
617611

618612

@@ -649,7 +643,7 @@ async def test_client_connect_networking_error():
649643
connect_websocket_mock:
650644
connect_websocket_mock.side_effect = OSError()
651645
with pytest.raises(HandshakeError):
652-
async with open_websocket(HOST, 0, '/', use_ssl=False) as client_ws:
646+
async with open_websocket(HOST, 0, '/', use_ssl=False):
653647
pass
654648

655649

@@ -672,7 +666,7 @@ async def handler(request):
672666

673667
old_task_count = len(nursery.child_tasks)
674668
# This stream is not a WebSocket, so it won't send a handshake:
675-
stream = await trio.open_tcp_stream(HOST, server.port)
669+
await trio.open_tcp_stream(HOST, server.port)
676670
# Checkpoint so the server's handler task can spawn:
677671
await trio.sleep(0)
678672
assert len(nursery.child_tasks) == old_task_count + 1, \
@@ -713,7 +707,7 @@ async def handler(request):
713707
# Spawn client inside an inner nursery so that we can cancel it's reader
714708
# so that it won't do a closing handshake.
715709
async with trio.open_nursery() as inner:
716-
ws = await connect_websocket(inner, HOST, server.port, RESOURCE,
710+
await connect_websocket(inner, HOST, server.port, RESOURCE,
717711
use_ssl=False)
718712
# Checkpoint so the server can spawn a handler task:
719713
await trio.sleep(0)
@@ -785,7 +779,7 @@ async def handler(stream):
785779

786780
async def test_server_handler_exit(nursery, autojump_clock):
787781
async def handler(request):
788-
server_ws = await request.accept()
782+
await request.accept()
789783
await trio.sleep(1)
790784

791785
server = await nursery.start(

0 commit comments

Comments
 (0)