@@ -91,13 +91,13 @@ def socket_handler(listener):
9191 f = SettingsFrame (0 )
9292 sock .send (f .serialize ())
9393
94- send_event .wait ()
94+ send_event .set ()
9595 sock .close ()
9696
9797 self ._start_server (socket_handler )
9898 conn = self .get_connection ()
9999 conn .connect ()
100- send_event .set ( )
100+ send_event .wait ( 5 )
101101
102102 assert data [0 ] == b'PRI * HTTP/2.0\r \n \r \n SM\r \n \r \n '
103103
@@ -125,13 +125,13 @@ def socket_handler(listener):
125125 f = SettingsFrame (0 )
126126 sock .send (f .serialize ())
127127
128- send_event .wait ()
128+ send_event .set ()
129129 sock .close ()
130130
131131 self ._start_server (socket_handler )
132132 conn = self .get_connection ()
133133 conn .connect ()
134- send_event .set ()
134+ send_event .wait ()
135135
136136 # Get the second chunk of data and decode it into a frame.
137137 data = data [1 ]
@@ -270,7 +270,7 @@ def socket_handler(listener):
270270 sock .send (f .serialize ())
271271
272272 # Wait for the message from the main thread.
273- recv_event .wait ()
273+ recv_event .set ()
274274 sock .close ()
275275
276276 self ._start_server (socket_handler )
@@ -281,7 +281,7 @@ def socket_handler(listener):
281281 # Close the response.
282282 resp .close ()
283283
284- recv_event .set ( )
284+ recv_event .wait ( 5 )
285285
286286 assert not conn .streams
287287
@@ -307,7 +307,7 @@ def socket_handler(listener):
307307 sock .send (f .serialize ())
308308
309309 # Wait for the message from the main thread.
310- recv_event .wait ()
310+ recv_event .set ()
311311 sock .close ()
312312
313313 self ._start_server (socket_handler )
@@ -323,7 +323,7 @@ def socket_handler(listener):
323323 assert resp ._stream ._in_window_manager .document_size == 0
324324
325325 # Awesome, we're done now.
326- recv_event .set ( )
326+ recv_event .wait ( 5 )
327327
328328 self .tear_down ()
329329
@@ -360,7 +360,7 @@ def socket_handler(listener):
360360 sock .send (f .serialize ())
361361
362362 # Wait for the message from the main thread.
363- recv_event .wait ()
363+ recv_event .set ()
364364 sock .close ()
365365
366366 self ._start_server (socket_handler )
@@ -383,7 +383,7 @@ def socket_handler(listener):
383383 assert len (resp .trailers ) == 1
384384
385385 # Awesome, we're done now.
386- recv_event .set ( )
386+ recv_event .wait ( 5 )
387387
388388 self .tear_down ()
389389
@@ -407,7 +407,7 @@ def socket_handler(listener):
407407 sock .send (f .serialize ())
408408
409409 # Wait for the message from the main thread.
410- recv_event .wait ()
410+ recv_event .set ()
411411 sock .close ()
412412
413413 self ._start_server (socket_handler )
@@ -418,7 +418,7 @@ def socket_handler(listener):
418418 assert conn ._sock is None
419419
420420 # Awesome, we're done now.
421- recv_event .set ( )
421+ recv_event .wait ( 5 )
422422
423423 self .tear_down ()
424424
@@ -443,7 +443,7 @@ def socket_handler(listener):
443443
444444 # Wait for the message from the main thread.
445445 sock .close ()
446- recv_event .wait ()
446+ recv_event .set ()
447447
448448 self ._start_server (socket_handler )
449449 conn = self .get_connection ()
@@ -455,7 +455,7 @@ def socket_handler(listener):
455455 assert conn ._sock is None
456456
457457 # Awesome, we're done now.
458- recv_event .set ( )
458+ recv_event .wait ( 5 )
459459
460460 self .tear_down ()
461461
@@ -471,7 +471,7 @@ def socket_handler(listener):
471471 receive_preamble (sock )
472472
473473 data .append (sock .recv (65535 ))
474- send_event .wait ()
474+ send_event .wait (5 )
475475
476476 h = HeadersFrame (1 )
477477 h .data = self .get_encoder ().encode (
@@ -518,7 +518,7 @@ def socket_handler(listener):
518518 receive_preamble (sock )
519519
520520 data .append (sock .recv (65535 ))
521- send_event .wait ()
521+ send_event .wait (5 )
522522
523523 h = HeadersFrame (1 )
524524 h .data = self .get_encoder ().encode (
@@ -581,7 +581,7 @@ def socket_handler(listener):
581581 sock .send (f .serialize ())
582582
583583 # Wait for the message from the main thread.
584- recv_event .wait ()
584+ recv_event .wait (5 )
585585 sock .close ()
586586
587587 self ._start_server (socket_handler )
@@ -626,7 +626,7 @@ def socket_handler(listener):
626626 sock .send (f .serialize ())
627627
628628 # Wait for the message from the main thread.
629- recv_event .wait ()
629+ recv_event .wait (5 )
630630 sock .close ()
631631
632632 self ._start_server (socket_handler )
@@ -688,7 +688,7 @@ def socket_handler(listener):
688688 d .flags .add ('END_STREAM' )
689689 sock .send (d .serialize ())
690690
691- send_event .wait ()
691+ send_event .wait (5 )
692692 sock .close ()
693693
694694 self ._start_server (socket_handler )
@@ -720,7 +720,6 @@ def wrap(*args):
720720 monkeypatch .setattr (hyper .http11 .connection , 'wrap_socket' , wrap )
721721
722722 data = []
723- send_event = threading .Event ()
724723
725724 def socket_handler (listener ):
726725 sock = listener .accept ()[0 ]
@@ -742,7 +741,6 @@ def socket_handler(listener):
742741 d .flags .add ('END_STREAM' )
743742 sock .send (d .serialize ())
744743
745- send_event .set ()
746744 sock .close ()
747745
748746 self ._start_server (socket_handler )
0 commit comments