Merge pull request #18 from luminougat/fix-fuzzing-with-parameters-2

dzonerzy · web-flow · commit f777067076f6 · 2019-01-15T21:06:47.000+01:00
Fix fuzzing with parameters in case of lists (Part 2)
diff --git a/.gitignore b/.gitignore
@@ -3,3 +3,4 @@ __pycache__
 *.egg-info/
 build/
 dist/
+*.pyc
diff --git a/pyjfuzz/core/pjf_factory.py b/pyjfuzz/core/pjf_factory.py
@@ -160,7 +160,10 @@ def fuzz_elements(self, element):
                     elif type(key) == list:
                         arr.append(self.fuzz_elements(key))
                     else:
-                        arr.append(self.mutator.fuzz(key))
+                        if len(self.config.parameters) <= 0:
+                            arr.append(self.mutator.fuzz(key))
+                        else:
+                            arr.append(key)
                 element = arr
                 del arr
         except Exception as e:
diff --git a/test/test_pjf_factory.py b/test/test_pjf_factory.py
@@ -71,6 +71,12 @@ def test_object_fuzz(self):
                                                      indent=True, nologo=True)))
         self.assertTrue(json.fuzzed)
 
+    def test_object_parameters(self):
+        json = PJFFactory(PJFConfiguration(Namespace(parameters="d", json={"a": [{"b" : "c"}, "abcd"]}, nologo=True, level=6)))
+        self.assertTrue(json != json.fuzzed)
+        self.assertTrue("abcd" in json.fuzzed)
+
+
 
 def test():
     print("=" * len(__TITLE__))

-Original file line number
+Diff line change
 *.egg-info/
 build/
 dist/
 +*.pyc