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)