fuzzitdev · yevgenypats · Jan 6, 2020 · Dec 30, 2019
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_crash.py b/tests/test_crash.py
@@ -0,0 +1,17 @@
+import unittest
+import zipfile
+import io
+from unittest.mock import patch
+
+import pythonfuzz
+
+class TestFindCrash(unittest.TestCase):
+    def test_find_crash(self):
+        def fuzz(buf):
+            f = io.BytesIO(buf)
+            z = zipfile.ZipFile(f)
+            z.testzip()
+
+        with patch('logging.Logger.info') as mock:
+            pythonfuzz.fuzzer.Fuzzer(fuzz).start()
+            self.assertTrue(mock.called_once)
diff --git a/tests/test_nocrash.py b/tests/test_nocrash.py
@@ -0,0 +1,15 @@
+import unittest
+import zipfile
+import io
+from unittest.mock import patch
+
+import pythonfuzz
+
+class TestFindCrash(unittest.TestCase):
+    def test_find_crash(self):
+        def fuzz(buf):
+            return True
+
+        with patch('logging.Logger.info') as mock:
+            pythonfuzz.fuzzer.Fuzzer(fuzz, runs=100).start()
+            mock.assert_called_with('did %d runs, stopping now.', 100)