def test_cve_key_checker(self): # avoid pragma: no cover assert cve_key_checker(mock.Mock(data={'cve': None})) is None assert cve_key_checker(mock.Mock(data={'cve': True})) assert cve_key_checker(mock.Mock(data={})) is None
def test_without_cve_key(self): myresult = cve_key_checker(Vuln3) self.assertEqual(myresult, None)
def test_cve_key_with_value(self): myresult = cve_key_checker(Vuln1) self.assertEqual(myresult, True)