Don't forget to receive header frames

python-hyper · Jun 13, 2015 · b05c817 · b05c817
1 parent 2aad1bb
commit b05c817
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/test_integration.py b/test/test_integration.py
@@ -262,6 +262,7 @@ def socket_handler(listener):
             # We're going to get the two messages for the connection open, then
             # a headers frame.
             receive_preamble(sock)
+            sock.recv(65535)
 
             # Now, send the headers for the response.
             f = build_headers_frame([(':status', '200')])
@@ -297,6 +298,7 @@ def socket_handler(listener):
             # We get two messages for the connection open and then a HEADERS
             # frame.
             receive_preamble(sock)
+            sock.recv(65535)
 
             # Now, send the headers for the response. This response has no body.
             f = build_headers_frame([(':status', '204'), ('content-length', '0')])
@@ -339,6 +341,7 @@ def socket_handler(listener):
             # We get two messages for the connection open and then a HEADERS
             # frame.
             receive_preamble(sock)
+            sock.recv(65535)
 
             # Now, send the headers for the response. This response has no body.
             f = build_headers_frame([(':status', '200'), ('content-length', '0')], e)