Fix tools/afl/harness.py to not break tests

vfaronov · Mar 24, 2017 · 156afa7 · 156afa7
1 parent 7fd0dfc
commit 156afa7
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/tools/afl/harness.py b/tools/afl/harness.py
@@ -5,11 +5,10 @@
 
 import httpolice.cli
 
-args = httpolice.cli.parse_args(sys.argv)
-
-import afl
-while afl.loop(100):
-    httpolice.cli.run_cli(args, sys.stdout, sys.stderr)
-
-# As suggestd by python-afl docs.
-os._exit(0)         # pylint: disable=protected-access
+if __name__ == '__main__':
+    args = httpolice.cli.parse_args(sys.argv)
+    import afl
+    while afl.loop(100):
+        httpolice.cli.run_cli(args, sys.stdout, sys.stderr)
+    # As suggested by python-afl docs.
+    os._exit(0)         # pylint: disable=protected-access