Make sure to grab GOAWAY frame

python-hyper · May 16, 2015 · 21f6110 · 21f6110
1 parent 74c1453
commit 21f6110
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_integration.py b/test/test_integration.py
@@ -233,12 +233,14 @@ def socket_handler(listener):
             sock.send(f.serialize())
 
             send_event.wait()
+            sock.recv(65535)
             sock.close()
 
         self._start_server(socket_handler)
         with self.get_connection() as conn:
             conn.connect()
-            send_event.set()
+
+        send_event.set()
 
         # Check that we closed the connection.
         assert conn._sock == None