fixed unitests

Author: OsamaMahmood

unittests/tools/test_wizcli_dir_parser.py

@@ -42,8 +42,8 @@ def test_multiple_findings(self):
         with (get_unit_tests_scans_path("wizcli_dir") / "wizcli_dir_many_vul.json").open(encoding="utf-8") as testfile:
             parser = WizcliDirParser()
             findings = parser.get_findings(testfile, Test())
-            self.assertEqual(7, len(findings))
-
+            self.assertEqual(204, len(findings))
+            
             # Test first finding
             finding = findings[0]
             self.assertEqual("github.com/golang-jwt/jwt/v4 4.5.1 - CVE-2025-30204", finding.title)

unittests/tools/test_wizcli_iac_parser.py

@@ -1,18 +1,18 @@
 from dojo.models import Test
-from dojo.tools.wizcli_iac.parser import WizcliIaCParser
+from dojo.tools.wizcli_iac.parser import WizcliIacParser
 from unittests.dojo_test_case import DojoTestCase, get_unit_tests_scans_path
 
 
-class TestWizcliIaCParser(DojoTestCase):
+class TestWizcliIacParser(DojoTestCase):
     def test_no_findings(self):
         with (get_unit_tests_scans_path("wizcli_iac") / "wizcli_iac_zero_vul.json").open(encoding="utf-8") as testfile:
-            parser = WizcliIaCParser()
+            parser = WizcliIacParser()
             findings = parser.get_findings(testfile, Test())
             self.assertEqual(len(findings), 0)
 
     def test_one_findings(self):
         with (get_unit_tests_scans_path("wizcli_iac") / "wizcli_iac_one_vul.json").open(encoding="utf-8") as testfile:
-            parser = WizcliIaCParser()
+            parser = WizcliIacParser()
             findings = parser.get_findings(testfile, Test())
             self.assertEqual(1, len(findings))
             finding = findings[0]
@@ -43,7 +43,7 @@ def test_one_findings(self):
 
     def test_multiple_findings(self):
         with (get_unit_tests_scans_path("wizcli_iac") / "wizcli_iac_many_vul.json").open(encoding="utf-8") as testfile:
-            parser = WizcliIaCParser()
+            parser = WizcliIacParser()
             findings = parser.get_findings(testfile, Test())
             self.assertEqual(25, len(findings))
 

unittests/tools/test_wizcli_img_parser.py

@@ -38,8 +38,8 @@ def test_multiple_findings(self):
         with (get_unit_tests_scans_path("wizcli_img") / "wizcli_img_many_vul.json").open(encoding="utf-8") as testfile:
             parser = WizcliImgParser()
             findings = parser.get_findings(testfile, Test())
-            self.assertEqual(9, len(findings))
-
+            self.assertEqual(29, len(findings))
+            
             # Test first finding
             finding = findings[0]
             self.assertEqual("OS Pkg: libcrypto3 3.3.1-r0 - CVE-2024-5535", finding.title)