Add more fragmented header tests

Author: youknowone

tests/test_parser.py

@@ -454,6 +454,75 @@ def test_parser_request_fragmented(self):
              b'Content-Length': b'10'})
         m.on_message_complete.assert_called_once_with()
 
+    def test_parser_request_fragmented_header(self):
+        m = mock.Mock()
+        headers = {}
+        m.on_header.side_effect = headers.__setitem__
+        p = httptools.HttpRequestParser(m)
+
+        REQUEST = (
+            b'PUT / HTTP/1.1\r\nHost: localhost:1234\r\nContent-',
+            b'Type: text/plain; charset=utf-8\r\n\r\n',
+        )
+
+        p.feed_data(REQUEST[0])
+
+        m.on_message_begin.assert_called_once_with()
+        m.on_url.assert_called_once_with(b'/')
+        self.assertEqual(headers, {b'Host': b'localhost:1234'})
+
+        p.feed_data(REQUEST[1])
+        self.assertEqual(
+            headers,
+            {b'Host': b'localhost:1234',
+             b'Content-Type': b'text/plain; charset=utf-8'})
+
+    def test_parser_request_fragmented_value(self):
+        m = mock.Mock()
+        headers = {}
+        m.on_header.side_effect = headers.__setitem__
+        p = httptools.HttpRequestParser(m)
+
+        REQUEST = (
+            b'PUT / HTTP/1.1\r\nHost: localhost:1234\r\nContent-Type:',
+            b' text/pla',
+            b'in; chars',
+            b'et=utf-8\r\n\r\n',
+        )
+
+        p.feed_data(REQUEST[0])
+
+        m.on_message_begin.assert_called_once_with()
+        m.on_url.assert_called_once_with(b'/')
+        self.assertEqual(headers, {b'Host': b'localhost:1234'})
+
+        p.feed_data(REQUEST[1])
+        p.feed_data(REQUEST[2])
+        p.feed_data(REQUEST[3])
+        self.assertEqual(
+            headers,
+            {b'Host': b'localhost:1234',
+             b'Content-Type': b'text/plain; charset=utf-8'})
+
+    def test_parser_request_fragmented_bytes(self):
+        m = mock.Mock()
+        headers = {}
+        m.on_header.side_effect = headers.__setitem__
+        p = httptools.HttpRequestParser(m)
+
+        REQUEST = \
+            b'PUT / HTTP/1.1\r\nHost: localhost:1234\r\nContent-' \
+            b'Type: text/plain; charset=utf-8\r\n\r\n'
+
+        step = 1
+        for i in range(0, len(REQUEST), step):
+            p.feed_data(REQUEST[i:i+step])
+
+        self.assertEqual(
+            headers,
+            {b'Host': b'localhost:1234',
+             b'Content-Type': b'text/plain; charset=utf-8'})
+
 
 class TestUrlParser(unittest.TestCase):