def test_set_risk(self): """ Test setRisk(self, risk) """ event_type = 'ROOT' event_data = '' module = '' source_event = '' evt = SpiderFootEvent(event_type, event_data, module, source_event) evt.setRisk('risk')
def test_set_risk_should_set_risk_attribute(self): """ Test setRisk(self, risk) """ event_type = 'ROOT' event_data = '' module = '' source_event = '' evt = SpiderFootEvent(event_type, event_data, module, source_event) risk = 100 evt.setRisk(risk) self.assertEqual(risk, evt.risk)
def test_set_risk_invalid_risk_should_raise(self): """ Test setRisk(self, risk) """ event_type = 'ROOT' event_data = '' module = '' source_event = '' evt = SpiderFootEvent(event_type, event_data, module, source_event) invalid_types = [None, "", list(), dict()] for invalid_type in invalid_types: with self.subTest(invalid_type=invalid_type): with self.assertRaises(TypeError) as cm: evt.setRisk(invalid_type) with self.assertRaises(ValueError) as cm: evt.setRisk(-1) with self.assertRaises(ValueError) as cm: evt.setRisk(101)