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_generate_response(self, _mock_msg, _mock_link, _mock_maps): """Test Response Generator Function.""" response_obj = ComponentAnalysisResponseBuilder(self.eco, self.pkg, self.ver) _mock_msg.return_value = 'You are Superb.' _mock_link.return_value = 'https://xyx.com' _mock_maps.return_value = {} response_obj.nocve_version = 1 response_obj.severity = ['high'] response_obj.public_vul = 2 response_obj.pvt_vul = 1 response = response_obj.generate_response() mocked_response = dict( recommended_versions=response_obj.nocve_version, registration_link=_mock_link.return_value, component_analyses=dict(vulnerability=_mock_maps.return_value), message=_mock_msg.return_value, severity=response_obj.severity[0], known_security_vulnerability_count=response_obj.public_vul, security_advisory_count=response_obj.pvt_vul, ) self.assertDictEqual(response, mocked_response)