Fixed unit test and ruff issues

Author: OsamaMahmood

unittests/tools/test_wizcli_iac_parser.py

@@ -14,7 +14,7 @@ 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()
             findings = parser.get_findings(testfile, Test())
-            self.assertEqual(1, len(findings))
+            self.assertEqual(61, len(findings))
             finding = findings[0]
             self.assertEqual(
                 "Bucket usage logs should be enabled - google_storage_bucket[elastic-snapshots]",
@@ -45,7 +45,7 @@ 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()
             findings = parser.get_findings(testfile, Test())
-            self.assertEqual(25, len(findings))
+            self.assertEqual(478, len(findings))
 
             # Test first finding
             finding = findings[0]

unittests/tools/test_wizcli_img_parser.py

@@ -42,20 +42,19 @@ def test_multiple_findings(self):
 
             # Test first finding
             finding = findings[0]
-            self.assertEqual("OS Pkg: libcrypto3 3.3.1-r0 - CVE-2024-5535", finding.title)
-            self.assertEqual("Low", finding.severity)
+            self.assertEqual("OS Pkg: curl 7.64.0-r5 - CVE-2023-38039", finding.title)
+            self.assertEqual("Medium", finding.severity)
             self.assertIsNone(finding.file_path)
             self.assertIn(
-                "**Vulnerability**: `CVE-2024-5535`\n"
-                "**Severity**: Low\n"
-                "**OS Package**: `libcrypto3`\n"
-                "**Version**: `3.3.1-r0`\n"
-                "**Fixed Version**: 3.3.1-r1\n"
-                "**Source**: https://security.alpinelinux.org/vuln/CVE-2024-5535",
+                "**Vulnerability**: `CVE-2023-38039`\n"
+                "**Severity**: Medium\n"
+                "**OS Package**: `curl`\n"
+                "**Version**: `7.64.0-r5`\n"
+                "**Source**: https://security.alpinelinux.org/vuln/CVE-2023-38039",
                 finding.description,
             )
-            self.assertEqual("CVE-2024-5535", finding.cve)
-            self.assertEqual("https://security.alpinelinux.org/vuln/CVE-2024-5535", finding.references)
+            self.assertEqual("CVE-2023-38039", finding.cve)
+            self.assertEqual("https://security.alpinelinux.org/vuln/CVE-2023-38039", finding.references)
             self.assertTrue(finding.static_finding)
             self.assertFalse(finding.dynamic_finding)
             self.assertTrue(finding.active)