def test_get_vulnerabilities_count_non_zero_public(self):
"""Test Vulnerabilities count Public."""
response_obj = ComponentAnalysisResponseBuilder(self.eco, self.pkg, self.ver)
response_obj._cves = [dict(snyk_pvt_vulnerability=[False])]
pub_vul, pvt_vul = response_obj.get_vulnerabilities_count()
self.assertEquals(pub_vul, 1)
self.assertEquals(pvt_vul, 0)
def test_get_vulnerabilities_count_zero_exception(self):
"""Test Vulnerabilities count Exception."""
response_obj = ComponentAnalysisResponseBuilder(self.eco, self.pkg, self.ver)
response_obj._cves = [dict()]
pub_vul, pvt_vul = response_obj.get_vulnerabilities_count()
self.assertEquals(pub_vul, 0)
self.assertEquals(pvt_vul, 0)
