Ensure fuzz targets work as expected.

Author: aaugustin

fuzzing/fuzz_http11_request_parser.py

@@ -20,15 +20,18 @@ def test_one_input(data):
 
     try:
         next(parser)
-    except StopIteration:
-        pass  # request is available in exc.value
+    except StopIteration as exc:
+        assert isinstance(exc.value, Request)
+        return
     except (
         EOFError,  # connection is closed without a full HTTP request
         SecurityError,  # request exceeds a security limit
         ValueError,  # request isn't well formatted
     ):
         pass
 
+    raise RuntimeError("parsing didn't complete")
+
 
 def main():
     atheris.Setup(sys.argv, test_one_input)

fuzzing/fuzz_http11_response_parser.py

@@ -21,8 +21,9 @@ def test_one_input(data):
     )
     try:
         next(parser)
-    except StopIteration:
-        pass  # response is available in exc.value
+    except StopIteration as exc:
+        assert isinstance(exc.value, Response)
+        return
     except (
         EOFError,  # connection is closed without a full HTTP response
         SecurityError,  # response exceeds a security limit
@@ -31,6 +32,8 @@ def test_one_input(data):
     ):
         pass
 
+    raise RuntimeError("parsing didn't complete")
+
 
 def main():
     atheris.Setup(sys.argv, test_one_input)

fuzzing/fuzz_websocket_parser.py

@@ -28,14 +28,17 @@ def test_one_input(data):
 
     try:
         next(parser)
-    except StopIteration:
-        pass  # response is available in exc.value
+    except StopIteration as exc:
+        assert isinstance(exc.value, Frame)
+        return
     except (
         PayloadTooBig,  # frame's payload size exceeds ``max_size``
         ProtocolError,  # frame contains incorrect values
     ):
         pass
 
+    raise RuntimeError("parsing didn't complete")
+
 
 def main():
     atheris.Setup(sys.argv, test_one_input)