def test_get_premium_message_with_pub_pvt_vul(self):
"""Test Get Premium Message With Public and Private Vulnerabilities."""
obj = CABatchResponseBuilder(self.eco)
obj.public_vul = 1
obj.pvt_vul = 1
obj.package = "django"
obj.version = "1.1"
msg = obj.get_premium_message(1)
ideal_msg = "django - 1.1 has 1 known security vulnerability " \
"and 1 security advisory with 1 exploitable " \
"vulnerabilities. No recommended version."
self.assertEqual(msg, ideal_msg)
def test_get_premium_message_with_0_exploit_vul(self):
"""Test Get Premium Message With Public Vulnerabilities."""
obj = CABatchResponseBuilder(self.eco)
obj.public_vul = 1
obj.pvt_vul = 1
obj.package = "django"
obj.version = "1.1"
obj.severity = ['high']
msg = obj.get_premium_message(0)
ideal_msg = "django - 1.1 has 1 known security vulnerability " \
"and 1 security advisory with high severity. No recommended version."
self.assertEqual(msg, ideal_msg)
def test_get_premium_response(self, _mock1, _mock2):
"""Test Get Premium Response."""
_mock1.return_value = 1
_mock2.return_value = "https://snyk.io/vuln/pypi:django"
obj = CABatchResponseBuilder(self.eco)
obj.severity = ["high", "high"]
obj.package = "django"
obj.version = "1.1"
response = obj.get_premium_response()
ideal_response = {'package': 'django',
'version': '1.1',
'recommended_versions': '',
'package_unknown': False,
'vendor_package_link': 'https://snyk.io/vuln/pypi:django',
'vulnerability': [],
'message': 'django - 1.1 has ',
'highest_severity': 'high',
'known_security_vulnerability_count': 0,
'security_advisory_count': 0,
'exploitable_vulnerabilities_count': 1
}
self.assertEqual(response, ideal_response)