@@ -194,7 +194,7 @@ async def test_serve(nursery):
194
194
# The server nursery begins with one task (server.listen).
195
195
assert len (nursery .child_tasks ) == 1
196
196
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 ):
198
198
# The server nursery has the same number of tasks, but there is now
199
199
# one additional nested nursery.
200
200
assert len (nursery .child_tasks ) == 1
@@ -220,25 +220,22 @@ async def test_serve_ssl(nursery):
220
220
221
221
222
222
async def test_serve_handler_nursery (nursery ):
223
- task = current_task ()
224
223
async with trio .open_nursery () as handler_nursery :
225
224
serve_with_nursery = partial (serve_websocket , echo_request_handler ,
226
225
HOST , 0 , None , handler_nursery = handler_nursery )
227
226
server = await nursery .start (serve_with_nursery )
228
227
port = server .port
229
228
# The server nursery begins with one task (server.listen).
230
229
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 ):
233
231
# The handler nursery should have one task in it
234
232
# (conn._reader_task).
235
233
assert len (handler_nursery .child_tasks ) == 1
236
234
237
235
238
- async def test_serve_with_zero_listeners (nursery ):
239
- task = current_task ()
236
+ async def test_serve_with_zero_listeners ():
240
237
with pytest .raises (ValueError ):
241
- server = WebSocketServer (echo_request_handler , [])
238
+ WebSocketServer (echo_request_handler , [])
242
239
243
240
244
241
async def test_serve_non_tcp_listener (nursery ):
@@ -290,7 +287,7 @@ async def test_client_open_url(path, expected_path, echo_server):
290
287
291
288
async def test_client_open_invalid_url (echo_server ):
292
289
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' ):
294
291
pass
295
292
296
293
@@ -358,11 +355,10 @@ async def handler(request):
358
355
assert not request .local .is_ssl
359
356
assert str (request .remote .address ) == HOST
360
357
assert not request .remote .is_ssl
361
- conn = await request .accept ()
358
+ await request .accept ()
362
359
363
360
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 ):
366
362
pass
367
363
368
364
@@ -410,7 +406,7 @@ async def handler(request):
410
406
async def test_handshake_server_headers (nursery ):
411
407
async def handler (request ):
412
408
headers = [('X-Test-Header' , 'My test header' )]
413
- server_ws = await request .accept (extra_headers = headers )
409
+ await request .accept (extra_headers = headers )
414
410
415
411
server = await nursery .start (serve_websocket , handler , HOST , 0 , None )
416
412
async with open_websocket (HOST , server .port , RESOURCE , use_ssl = False
@@ -433,7 +429,7 @@ async def handler(request):
433
429
server = await nursery .start (serve_websocket , handler , HOST , 0 ,
434
430
None )
435
431
async with open_websocket (HOST , server .port , RESOURCE ,
436
- use_ssl = False ) as client_ws :
432
+ use_ssl = False ):
437
433
pass
438
434
439
435
@@ -445,8 +441,7 @@ async def handler(request):
445
441
446
442
server = await nursery .start (serve_websocket , handler , HOST , 0 , None )
447
443
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 ):
450
445
pass
451
446
exc = exc_info .value
452
447
assert exc .body == b'My body'
@@ -467,16 +462,15 @@ async def handler(stream):
467
462
port = listeners [0 ].socket .getsockname ()[1 ]
468
463
469
464
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 ):
472
466
pass
473
467
exc = exc_info .value
474
468
assert exc .status_code == 100
475
469
assert repr (exc ) == 'ConnectionRejected<status_code=100>'
476
470
assert exc .body is None
477
471
478
472
479
- async def test_handshake_protocol_error (nursery , echo_server ):
473
+ async def test_handshake_protocol_error (echo_server ):
480
474
'''
481
475
If a client connects to a trio-websocket server and tries to speak HTTP
482
476
instead of WebSocket, the server should reject the connection. (If the
@@ -604,15 +598,15 @@ async def test_client_open_timeout(nursery, autojump_clock):
604
598
'''
605
599
async def handler (request ):
606
600
await trio .sleep (FORCE_TIMEOUT )
607
- server_ws = await request .accept ()
601
+ await request .accept ()
608
602
pytest .fail ('Should not reach this line.' )
609
603
610
604
server = await nursery .start (
611
605
partial (serve_websocket , handler , HOST , 0 , ssl_context = None ))
612
606
613
607
with pytest .raises (ConnectionTimeout ):
614
608
async with open_websocket (HOST , server .port , '/' , use_ssl = False ,
615
- connect_timeout = TIMEOUT ) as client_ws :
609
+ connect_timeout = TIMEOUT ):
616
610
pass
617
611
618
612
@@ -649,7 +643,7 @@ async def test_client_connect_networking_error():
649
643
connect_websocket_mock :
650
644
connect_websocket_mock .side_effect = OSError ()
651
645
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 ):
653
647
pass
654
648
655
649
@@ -672,7 +666,7 @@ async def handler(request):
672
666
673
667
old_task_count = len (nursery .child_tasks )
674
668
# 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 )
676
670
# Checkpoint so the server's handler task can spawn:
677
671
await trio .sleep (0 )
678
672
assert len (nursery .child_tasks ) == old_task_count + 1 , \
@@ -713,7 +707,7 @@ async def handler(request):
713
707
# Spawn client inside an inner nursery so that we can cancel it's reader
714
708
# so that it won't do a closing handshake.
715
709
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 ,
717
711
use_ssl = False )
718
712
# Checkpoint so the server can spawn a handler task:
719
713
await trio .sleep (0 )
@@ -785,7 +779,7 @@ async def handler(stream):
785
779
786
780
async def test_server_handler_exit (nursery , autojump_clock ):
787
781
async def handler (request ):
788
- server_ws = await request .accept ()
782
+ await request .accept ()
789
783
await trio .sleep (1 )
790
784
791
785
server = await nursery .start (
0 commit comments