Python parser to behave like c parser when no headers in message

Author: quantmind

http_parser/pyparser.py

@@ -324,6 +324,11 @@ def _parse_request_line(self, line):
                 "SERVER_PROTOCOL": bits[2]})
 
     def _parse_headers(self, data):
+        if data == b'\r\n':
+            self.__on_headers_complete = True
+            self.__on_message_begin = True
+            self._buf = []
+            return 0
         idx = data.find(b("\r\n\r\n"))
         if idx < 0: # we don't have all headers
             if self._status_code == 204 and data == b("\r\n"):

testing/test_client.py

@@ -0,0 +1,39 @@
+import pytest
+import io
+
+from http_parser.http import HttpStream
+from http_parser.parser import HttpParser
+from http_parser.pyparser import HttpParser as PyHttpParser
+
+def _test_no_headers(parser):
+    data = b'HTTP/1.1 200 Connection established\r\n\r\n'
+    assert parser.execute(data, len(data)) == len(data)
+    assert parser.is_headers_complete()
+    assert parser.is_message_begin()
+    assert not parser.is_partial_body()
+    assert not parser.is_message_complete()
+
+def _test_headers(parser):
+    data = (b'HTTP/1.1 200 OK\r\n'
+            b'Connection: Keep-Alive\r\n'
+            b'Content-Length: 4\r\n'
+            b'Content-type: text/plain\r\n\r\n'
+            b'ciao')
+    assert parser.execute(data, len(data)) == len(data)
+    assert parser.is_headers_complete()
+    assert parser.is_message_begin()
+    assert parser.is_partial_body()
+    assert parser.is_message_complete()
+
+def test_no_headers():
+    _test_no_headers(HttpParser())
+
+def test_no_headers_py():
+    _test_no_headers(PyHttpParser())
+
+def test_headers():
+    _test_headers(HttpParser())
+
+def test_headers_py():
+    _test_headers(PyHttpParser())
+