Use supported_ecosystem to skip unsupported PURLs

Signed-off-by: Omkar Phansopkar <omkarphansopkar@gmail.com>
aboutcode-org · May 13, 2024 · 107cdaf · 107cdaf
1 parent 23fb8f5
commit 107cdaf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vulntotal/datasources/safetydb.py b/vulntotal/datasources/safetydb.py
@@ -46,7 +46,7 @@ def fetch_advisory(self):
         return response.json()
 
     def datasource_advisory(self, purl) -> Iterable[VendorData]:
-        if purl.type != "pypi":
+        if purl.type not in self.supported_ecosystem():
             return []
         advisory = self.fetch_advisory()
         self._raw_dump.append(advisory)

diff --git a/vulntotal/tests/test_safetydb.py b/vulntotal/tests/test_safetydb.py
@@ -17,7 +17,7 @@
 from vulntotal.datasources import safetydb
 
 
-class TestGithub(testcase.FileBasedTesting):
+class TestSafetydb(testcase.FileBasedTesting):
     test_data_dir = str(Path(__file__).resolve().parent / "test_data" / "safetydb")
 
     def test_parse_advisory(self):