Add tests for parsing chunked data split before or in the final CRLF (a…

…io-libs#4630)
JustAnotherArchivist · Mar 22, 2020 · a749d85 · a749d85
1 parent 693ed0c
commit a749d85
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/tests/test_http_parser.py b/tests/test_http_parser.py
@@ -826,6 +826,44 @@ async def test_parse_chunked_payload_size_error(self, stream) -> None:
         assert isinstance(out.exception(),
                           http_exceptions.TransferEncodingError)
 
+    async def test_parse_chunked_payload_split_end(self, protocol) -> None:
+        out = aiohttp.StreamReader(protocol, loop=None)
+        p = HttpPayloadParser(out, chunked=True)
+        p.feed_data(b'4\r\nasdf\r\n0\r\n')
+        p.feed_data(b'\r\n')
+
+        assert out.is_eof()
+        assert b'asdf' == b''.join(out._buffer)
+
+    async def test_parse_chunked_payload_split_end2(self, protocol) -> None:
+        out = aiohttp.StreamReader(protocol, loop=None)
+        p = HttpPayloadParser(out, chunked=True)
+        p.feed_data(b'4\r\nasdf\r\n0\r\n\r')
+        p.feed_data(b'\n')
+
+        assert out.is_eof()
+        assert b'asdf' == b''.join(out._buffer)
+
+    async def test_parse_chunked_payload_split_end_with_trailers(self, protocol) -> None:
+        out = aiohttp.StreamReader(protocol, loop=None)
+        p = HttpPayloadParser(out, chunked=True)
+        p.feed_data(b'4\r\nasdf\r\n0\r\n')
+        p.feed_data(b'Content-MD5: 912ec803b2ce49e4a541068d495ab570\r\n')
+        p.feed_data(b'\r\n')
+
+        assert out.is_eof()
+        assert b'asdf' == b''.join(out._buffer)
+
+    async def test_parse_chunked_payload_split_end_with_trailers2(self, protocol) -> None:
+        out = aiohttp.StreamReader(protocol, loop=None)
+        p = HttpPayloadParser(out, chunked=True)
+        p.feed_data(b'4\r\nasdf\r\n0\r\n')
+        p.feed_data(b'Content-MD5: 912ec803b2ce49e4a541068d495ab570\r\n\r')
+        p.feed_data(b'\n')
+
+        assert out.is_eof()
+        assert b'asdf' == b''.join(out._buffer)
+
     async def test_http_payload_parser_length(self, stream) -> None:
         out = aiohttp.FlowControlDataQueue(stream,
                                            loop=asyncio.get_event_loop())