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)