test keep-alive chunked requests

see #598

dummyserver/handlers.py

@@ -21,9 +21,6 @@
 
 class Response(object):
     def __init__(self, body='', status='200 OK', headers=None):
-        if not isinstance(body, bytes):
-            body = body.encode('utf8')
-
         self.body = body
         self.status = status
         self.headers = headers or [("Content-type", "text/plain")]
@@ -34,7 +31,20 @@ def __call__(self, request_handler):
         for header,value in self.headers:
             request_handler.add_header(header,value)
 
-        request_handler.write(self.body)
+        # chunked
+        if isinstance(self.body, list):
+            for item in self.body:
+                if not isinstance(item, bytes):
+                    item = item.encode('utf8')
+            request_handler.write(item)
+            request_handler.flush()
+        else:
+            body = self.body
+            if not isinstance(body, bytes):
+                body = body.encode('utf8')
+
+            request_handler.write(body)
+
 
 RETRY_TEST_NAMES = collections.defaultdict(int)
 
@@ -208,6 +218,9 @@ def successful_retry(self, request):
         else:
             return Response("need to keep retrying!", status="418 I'm A Teapot")
 
+    def chunked(self, request):
+        return Response(['123'] * 4)
+
     def shutdown(self, request):
         sys.exit()
 

test/with_dummyserver/test_connectionpool.py

@@ -618,6 +618,25 @@ def test_source_address_error(self):
             self.assertRaises(ProtocolError,
                     pool.request, 'GET', '/source_address')
 
+    def test_stream_keepalive(self):
+        x = 2
+
+        for _ in range(x):
+            response = self.pool.request(
+                    'GET',
+                    '/chunked',
+                    headers={
+                        'Connection': 'keep-alive',
+                        },
+                    preload_content=False,
+                    retries=0,
+                    )
+            for chunk in response.stream():
+                self.assertEqual(chunk, b'123')
+
+        self.assertEqual(self.pool.num_connections, 1)
+        self.assertEqual(self.pool.num_requests, x)
+
 
 class TestRetry(HTTPDummyServerTestCase):
     def setUp(self):