def test_get_premium_message_with_pvt_vul(self): """Test Get Premium Message With Private Vulnerabilities.""" obj = CABatchResponseBuilder(self.eco) obj.public_vul = 0 obj.pvt_vul = 1 obj.package = "django" obj.version = "1.1" msg = obj.get_premium_message(1) ideal_msg = "django - 1.1 has 1 security advisory with 1 exploitable vulnerabilities. " self.assertEqual(msg, ideal_msg)
def test_get_premium_message_with_public_vul(self): """Test Get Premium Message With Public Vulnerabilities.""" obj = CABatchResponseBuilder(self.eco) obj.public_vul = 1 obj.pvt_vul = 0 obj.package = "django" obj.version = "1.1" msg = obj.get_premium_message(1) ideal_msg = "django - 1.1 has 1 known security vulnerability 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)