fix the test

Signed-off-by: Ziad <[email protected]>

Author: ziadhany

vulnerabilities/importers/osv.py

@@ -57,10 +57,14 @@ def parse_advisory_data(raw_data: dict, supported_ecosystem) -> Optional[Advisor
     for affected_pkg in raw_data.get("affected") or []:
         purl = get_affected_purl(affected_pkg, raw_id)
         if purl.type != supported_ecosystem:
-            logger.error(f"un supported ecosystem package found in the advisories: {purl} - from: {raw_id !r}")
+            logger.error(
+                f"un supported ecosystem package found in the advisories: {purl} - from: {raw_id !r}"
+            )
             continue
 
-        affected_version_range = get_affected_version_range(affected_pkg, raw_id, supported_ecosystem)
+        affected_version_range = get_affected_version_range(
+            affected_pkg, raw_id, supported_ecosystem
+        )
         for fixed_range in affected_pkg.get("ranges", []):
             fixed_version = get_fixed_version(fixed_range, raw_id)
 

vulnerabilities/importers/pypa.py

@@ -27,7 +27,7 @@ class PyPaImporter(Importer):
 
     def advisory_data(self) -> Iterable[AdvisoryData]:
         for file in fork_and_get_files(self.url):
-            yield parse_advisory_data(file, supported_ecosystem='pypi')
+            yield parse_advisory_data(file, supported_ecosystem="pypi")
 
 
 class ForkError(Exception):

vulnerabilities/importers/pysec.py

@@ -41,4 +41,4 @@ def advisory_data(self) -> Iterable[AdvisoryData]:
                 else:
                     with zip_file.open(file_name) as f:
                         vul_info = json.load(f)
-                        yield parse_advisory_data(vul_info, supported_ecosystem='pypi')
+                        yield parse_advisory_data(vul_info, supported_ecosystem="pypi")

vulnerabilities/tests/test_osv.py

@@ -223,7 +223,7 @@ def test_get_affected_version_range(self):
             "1.6.2",
         ]
         assert get_affected_version_range(
-            {"versions": aff_version_range}, "GHSA-j3f7-7rmc-6wqj"
+            {"versions": aff_version_range}, "GHSA-j3f7-7rmc-6wqj", "pypi"
         ) == (
             PypiVersionRange(
                 constraints=[

vulnerabilities/tests/test_pypa.py

@@ -23,6 +23,6 @@ def test_to_advisories_with_summary(self):
         with open(os.path.join(TEST_DATA, "pypa_test.yaml")) as f:
             mock_response = saneyaml.load(f)
         expected_file = os.path.join(TEST_DATA, f"pypa-expected.json")
-        imported_data = parse_advisory_data(mock_response)
+        imported_data = parse_advisory_data(mock_response, "pypi")
         result = imported_data.to_dict()
         util_tests.check_results_against_json(result, expected_file)

vulnerabilities/tests/test_pysec.py

@@ -22,14 +22,14 @@ def test_to_advisories_with_summary(self):
         with open(os.path.join(TEST_DATA, "pysec_test_1.json")) as f:
             mock_response = json.load(f)
         expected_file = os.path.join(TEST_DATA, f"pysec-expected-1.json")
-        imported_data = parse_advisory_data(mock_response)
+        imported_data = parse_advisory_data(mock_response, "pypi")
         result = imported_data.to_dict()
         util_tests.check_results_against_json(result, expected_file)
 
     def test_to_advisories_without_summary(self):
         with open(os.path.join(TEST_DATA, "pysec_test_2.json")) as f:
             mock_response = json.load(f)
         expected_file = os.path.join(TEST_DATA, f"pysec-expected-2.json")
-        imported_data = parse_advisory_data(mock_response)
+        imported_data = parse_advisory_data(mock_response, "pypi")
         result = imported_data.to_dict()
         util_tests.check_results_against_json(result, expected_file)