Prove chunked logic works.

Author: Lukasa

hyper/http20/response.py

@@ -155,8 +155,6 @@ def read_chunked(self, decode_content=True):
         if decode_content and self._decompressobj:
             yield self._decompressobj.flush()
 
-        self.close()
-
         return
 
     def fileno(self):

test/test_integration.py

@@ -8,6 +8,7 @@
 """
 import requests
 import threading
+import time
 import hyper
 import hyper.http11.connection
 import pytest
@@ -555,7 +556,6 @@ def socket_handler(listener):
 
         self.tear_down()
 
-
     def test_resetting_stream_with_frames_in_flight(self):
         """
         Hyper emits only one RST_STREAM frame, despite the other frames in
@@ -650,6 +650,74 @@ def socket_handler(listener):
 
         self.tear_down()
 
+    def test_read_chunked_http2(self):
+        self.set_up()
+
+        recv_event = threading.Event()
+        wait_event = threading.Event()
+
+        def socket_handler(listener):
+            sock = listener.accept()[0]
+
+            # 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 a body.
+            f = build_headers_frame([(':status', '200')])
+            f.stream_id = 1
+            sock.send(f.serialize())
+
+            # Send the first two chunks.
+            f = DataFrame(1)
+            f.data = b'hello'
+            sock.sendall(f.serialize())
+            f = DataFrame(1)
+            f.data = b'there'
+            sock.sendall(f.serialize())
+
+            # Now, delay a bit. We want to wait a half a second before we send
+            # the next frame.
+            wait_event.wait(5)
+            time.sleep(0.5)
+            f = DataFrame(1)
+            f.data = b'world'
+            f.flags.add('END_STREAM')
+            sock.sendall(f.serialize())
+
+            # Wait for the message from the main thread.
+            recv_event.set()
+            sock.close()
+
+        self._start_server(socket_handler)
+        conn = self.get_connection()
+        conn.request('GET', '/')
+        resp = conn.get_response()
+
+        # Confirm the status code.
+        assert resp.status == 200
+
+        # Confirm that we can read this, but it has no body. First two chunks
+        # should be easy, then set the event and read the next one.
+        chunks = resp.read_chunked()
+        first_chunk = next(chunks)
+        second_chunk = next(chunks)
+        wait_event.set()
+        third_chunk = next(chunks)
+
+        with pytest.raises(StopIteration):
+            next(chunks)
+
+        assert first_chunk == b'hello'
+        assert second_chunk == b'there'
+        assert third_chunk == b'world'
+
+        # Awesome, we're done now.
+        recv_event.wait(5)
+
+        self.tear_down()
+
 
 class TestRequestsAdapter(SocketLevelTest):
     # This uses HTTP/2.