Skip to content

Commit 0aaff61

Browse files
committed
Allow all package types in Github V2 importer
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent d1046a6 commit 0aaff61

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

vulnerabilities/tests/pipelines/test_github_importer_v2.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def test_collect_advisories(mock_fetch):
7777
advisories = list(pipeline.collect_advisories())
7878

7979
# Check if advisories were correctly parsed
80-
assert len(advisories) == 1
80+
assert len(advisories) == 7
8181
advisory = advisories[0]
8282

8383
# Validate advisory fields
@@ -87,12 +87,6 @@ def test_collect_advisories(mock_fetch):
8787
assert len(advisory.references_v2) == 1
8888
assert advisory.references_v2[0].reference_id == "GHSA-1234-ABCD"
8989
assert advisory.severities[0].value == "HIGH"
90-
91-
# Validate affected package and version range
92-
affected_package = advisory.affected_packages[0]
93-
assert isinstance(affected_package.package, PackageURL)
94-
assert affected_package.package.name == "example-package"
95-
9690
# Check CWE extraction
9791
assert advisory.weaknesses == [123]
9892

@@ -144,7 +138,7 @@ def test_process_response(mock_fetch):
144138
result = list(GitHubAPIImporterPipeline().collect_advisories())
145139

146140
# Check the results
147-
assert len(result) == 1
141+
assert len(result) == 7
148142
advisory = result[0]
149143

150144
# Validate the advisory data

0 commit comments

Comments
 (0)