def test_get_message_with_pvt_vul_unequal_len(self):
"""Test Message with Private Vulnerability unequal len of severities and vul count."""
response_obj = ComponentAnalysisResponseBuilder("pypi", "django", "1.1")
response_obj.pvt_vul = 2
response_obj.severity = ['high']
message = response_obj.get_message()
ideal_msg = "django - 1.1 has 2 security advisory with 1 having high severity. "
self.assertEqual(message, ideal_msg)
def test_get_message_with_public_vul_equal(self):
"""Test Message with Public Vulnerability equal len of severities and vul count."""
response_obj = ComponentAnalysisResponseBuilder("pypi", "django", "1.1")
response_obj.public_vul = 1
response_obj.nocve_version = "3.1"
response_obj.severity = ['high']
message = response_obj.get_message()
ideal_msg = "django - 1.1 has 1 known security vulnerability " \
"having high severity. Recommendation: use version 3.1."
self.assertEqual(message, ideal_msg)
def test_get_message_with_both_vul(self):
"""Test Message with Both Vulnerability."""
response_obj = ComponentAnalysisResponseBuilder("pypi", "django", "1.1")
response_obj.public_vul = 1
response_obj.pvt_vul = 1
response_obj.severity = ['high']
message = response_obj.get_message()
ideal_msg = "django - 1.1 has 1 known security vulnerability and 1 " \
"security advisory with 1 having high severity. " \
"No recommended version."
self.assertEqual(message, ideal_msg)
