add custom chunk size support

Author: fafhrd91

aiohttp/web.py

@@ -392,6 +392,7 @@ def __init__(self, *, status=200, reason=None):
         self._body = None
         self._keep_alive = None
         self._chunked = False
+        self._chunk_size = None
         self._headers = CIMultiDict()
         self._cookies = http.cookies.SimpleCookie()
         self.set_status(status, reason)
@@ -434,9 +435,10 @@ def keep_alive(self):
     def force_close(self):
         self._keep_alive = False
 
-    def enable_chunked_encoding(self):
+    def enable_chunked_encoding(self, chunk_size=None):
         """Enables automatic chunked transfer encoding."""
         self._chunked = True
+        self._chunk_size = chunk_size
 
     @property
     def headers(self):
@@ -568,6 +570,8 @@ def start(self, request):
 
         if self._chunked:
             resp_impl.enable_chunked_encoding()
+            if self._chunk_size:
+                resp_impl.add_chunking_filter(self._chunk_size)
 
         headers = self.headers.items()
         for key, val in headers:

tests/test_web_response.py

@@ -127,6 +127,19 @@ def test_chunked_encoding(self, ResponseImpl):
         msg = resp.start(req)
         self.assertTrue(msg.chunked)
 
+    @mock.patch('aiohttp.web.ResponseImpl')
+    def test_chunk_size(self, ResponseImpl):
+        req = self.make_request('GET', '/')
+        resp = StreamResponse()
+        self.assertFalse(resp.chunked)
+
+        resp.enable_chunked_encoding(chunk_size=8192)
+        self.assertTrue(resp.chunked)
+
+        msg = resp.start(req)
+        self.assertTrue(msg.chunked)
+        self.assertIsNotNone(msg.filter)
+
     def test_write_non_byteish(self):
         resp = StreamResponse()
         resp.start(self.make_request('GET', '/'))