def test_set_visibility(self):
"""
Test setVisibility(self, visibility)
"""
event_type = 'ROOT'
event_data = ''
module = ''
source_event = ''
evt = SpiderFootEvent(event_type, event_data, module, source_event)
evt.setVisibility('visibility')
def test_set_visibility_should_set_visibility_attribute(self):
"""
Test setVisibility(self, visibility)
"""
event_type = 'ROOT'
event_data = ''
module = ''
source_event = ''
evt = SpiderFootEvent(event_type, event_data, module, source_event)
visibility = 100
evt.setVisibility(visibility)
self.assertEqual(visibility, evt.visibility)
def test_set_visibility_invalid_visibility_should_raise(self):
"""
Test setVisibility(self, visibility)
"""
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.setVisibility(invalid_type)
with self.assertRaises(ValueError) as cm:
evt.setVisibility(-1)
with self.assertRaises(ValueError) as cm:
evt.setVisibility(101)
