@@ -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
222222async 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
244241async def test_serve_non_tcp_listener (nursery ):
@@ -290,7 +287,7 @@ async def test_client_open_url(path, expected_path, echo_server):
290287
291288async 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):
410406async 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
786780async 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