Update importer mock test #597

Reference: #935

Signed-off-by: John M. Horan <[email protected]>

Author: johnmhoran

vulnerabilities/importers/archlinux.py

@@ -36,7 +36,7 @@ def fetch(self) -> Iterable[Mapping]:
 
     def advisory_data(self) -> Iterable[AdvisoryData]:
         for record in self.fetch():
-            yield self.parse_advisory(record)
+            yield from self.parse_advisory(record)
 
     def parse_advisory(self, record) -> List[AdvisoryData]:
         advisories = []

vulnerabilities/tests/test_archlinux.py

@@ -43,13 +43,11 @@ def test_parse_advisory_single():
     util_tests.check_results_against_json(result, expected_file)
 
 
-@patch("vulnerabilities.importers.archlinux.ArchlinuxImporter.fetch")
+@patch("vulnerabilities.utils.fetch_response")
 def test_archlinux_importer(mock_response):
     with open(os.path.join(TEST_DATA, "archlinux-multi.json")) as f:
         mock_response.return_value = json.load(f)
 
     expected_file = os.path.join(TEST_DATA, f"archlinux-multi-expected.json")
     result = [data.to_dict() for data in list(archlinux.ArchlinuxImporter().advisory_data())]
-    # result = [data.to_dict() for data in archlinux.ArchlinuxImporter().advisory_data()]
-    # result = archlinux.ArchlinuxImporter().advisory_data()
     util_tests.check_results_against_json(result, expected_file)