Exception in request handling if the server responds before the body is sent aio-libs#1761

Author: fafhrd91

CHANGES.rst

@@ -9,6 +9,8 @@ Changes
 
 - Disable cleanup closed ssl transports by default.
 
+- Exception in request handling if the server responds before the body is sent #1761
+
 
 2.0.4 (2017-03-27)
 ------------------

aiohttp/web_protocol.py

@@ -452,7 +452,8 @@ def start(self, message, payload, handler):
                         now = loop.time()
                         end_t = now + lingering_time
 
-                        with suppress(asyncio.TimeoutError):
+                        with suppress(
+                                asyncio.TimeoutError, asyncio.CancelledError):
                             while (not payload.is_eof() and now < end_t):
                                 timeout = min(end_t - now, lingering_time)
                                 with CeilTimeout(timeout, loop=loop):

tests/test_web_functional.py

@@ -102,6 +102,25 @@ def handler(request):
     assert '' == txt
 
 
+@asyncio.coroutine
+def test_response_before_complete(loop, test_client):
+
+    @asyncio.coroutine
+    def handler(request):
+        return web.Response(body=b'OK')
+
+    app = web.Application()
+    app.router.add_post('/', handler)
+    client = yield from test_client(app)
+
+    data = b'0' * 1024 * 1024
+
+    resp = yield from client.post('/', data=data)
+    assert 200 == resp.status
+    text = yield from resp.text()
+    assert 'OK' == text
+
+
 @asyncio.coroutine
 def test_post_form(loop, test_client):