Fuzzer improvements

- Simplify fuzzer scripts and run more API calls
frnmst · Apr 16, 2024 · b083a3b · b083a3b
1 parent 6600444
commit b083a3b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 8 deletions.
diff --git a/md_toc/tests/fuzzer.py b/md_toc/tests/fuzzer.py
@@ -32,13 +32,7 @@
 def TestBuildToc(data):
     r"""Test the md_toc.api.build_toc function."""
     with tempfile.NamedTemporaryFile() as fp:
-
-        bytez: bytes = b''.join([
-            bytes('#' * (secrets.randbelow(6) + 1), 'UTF-8'),
-            b' ',
-            data,
-        ])
-        fp.write(bytez)
+        fp.write(data)
 
         # Move pointer to the start of the file.
         fp.seek(0)
@@ -48,6 +42,19 @@ def TestBuildToc(data):
                 api.build_toc(filename=fp.name,
                               parser=parser,
                               keep_header_levels=6)
+                api.build_toc(filename=fp.name,
+                              parser=parser,
+                              keep_header_levels=6,
+                              ordered=True)
+                api.build_toc(filename=fp.name,
+                              parser=parser,
+                              keep_header_levels=6,
+                              ordered=True)
+                api.build_toc(filename=fp.name,
+                              parser=parser,
+                              keep_header_levels=6,
+                              skip_lines=secrets.randbelow(100))
+
         except (exceptions.GithubEmptyLinkLabel,
                 exceptions.TocDoesNotRenderAsCoherentList,
                 exceptions.GithubOverflowCharsLinkLabel) as e:

diff --git a/setup.cfg b/setup.cfg
@@ -119,5 +119,8 @@ package = wheel
 wheel_build_env = .pkg
 deps =
     atheris>=2.3,<2.4
+    -r requirements.txt
+    -r requirements-dev.txt
 commands =
-    python3 -m md_toc.tests.fuzzer
+    coverage run -m md_toc.tests.fuzzer -atheris_runs=100000000
+    coverage html