def test_get_cve_maps_empty(self):
"""Test cve maps with empty cve map, expect empty list []."""
response_obj = ComponentAnalysisResponseBuilder(
self.eco, self.pkg, self.ver)
response_obj._cves = []
cve_maps = response_obj.get_cve_maps()
self.assertListEqual(cve_maps, [])
def test_get_cve_maps_non_empty(self):
"""Test cve maps with vulnerability data, expect to get mock data as response."""
response_obj = ComponentAnalysisResponseBuilder(
self.eco, self.pkg, self.ver)
vul_data = dict(snyk_vuln_id=["SNYK:0101"],
cvss_scores=["9.0"],
snyk_pvt_vulnerability=[True],
snyk_cwes=["CWES-01", "CWES-02"],
snyk_cvss_v3=["4.5"],
severity=["medium"],
title=["Test title for CVE"],
snyk_url=["https://test.com/cve-01"],
snyk_cve_ids=["CVE-01", "CVE-02", "CVE-03"],
fixed_in=[">=0.19.0-beta.1"])
mocked_response = [
dict(vendor_cve_ids=vul_data['snyk_vuln_id'][0],
cvss=vul_data['cvss_scores'][0],
is_private=vul_data['snyk_pvt_vulnerability'][0],
cwes=["CWES-01", "CWES-02"],
cvss_v3="4.5",
severity="medium",
title="Test title for CVE",
url="https://test.com/cve-01",
cve_ids=["CVE-01", "CVE-02", "CVE-03"],
fixed_in=[">=0.19.0-beta.1"])
]
response_obj._cves = [vul_data]
cve_maps = response_obj.get_cve_maps()
self.assertListEqual(cve_maps, mocked_response)
def test_get_cve_maps_default(self):
"""Test cve maps with empty value, expect to get respose with default values."""
response_obj = ComponentAnalysisResponseBuilder(self.eco, self.pkg, self.ver)
vul_data = dict()
mocked_response = [dict(
vendor_cve_ids=None,
cvss='None',
is_private=None,
cwes=[],
cvss_v3=None,
severity=None,
title=None,
url=None,
cve_ids=[],
fixed_in=[])
]
response_obj._cves = [vul_data]
cve_maps = response_obj.get_cve_maps()
self.assertListEqual(cve_maps, mocked_response)
