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)
