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)