def test_clean(self): # all ok kwargs = {"requested": 5, "value": "test", "type": "msm", "action": "add"} source = AtlasChangeSource(**kwargs) self.assertEqual(source.clean(), None) # value missing source.value = None self.assertRaises( MalFormattedSource, lambda: source.clean() ) # type missing kwargs = {"requested": 5, "value": "test", "action": "add"} self.assertRaises( MalFormattedSource, lambda: AtlasChangeSource(**kwargs).clean() ) # action missing kwargs = {"requested": 5, "value": "test", "type": "probes"} self.assertRaises( MalFormattedSource, lambda: AtlasChangeSource(**kwargs).clean() ) # requested missing source.value = "test" source.requested = None self.assertRaises( MalFormattedSource, lambda: source.clean() )
class TestAtlasChangeSource(unittest.TestCase): def setUp(self): self.kwargs = { "requested": 5, "value": "test", "action": "add" } self.source = AtlasChangeSource(**self.kwargs) def test_setting_type(self): for source_type in ["area", "country", "prefix", "asn", "msm"]: self.assertRaises( MalFormattedSource, lambda: setattr(self.source, "type", source_type) ) self.assertEqual(setattr(self.source, "type", "probes"), None) def test_setting_action(self): for source_action in ["remove", "add"]: self.assertEqual( setattr(self.source, "action", source_action), None ) self.assertRaises( MalFormattedSource, lambda: setattr(self.source, "action", "test") ) def test_clean(self): self.assertEqual(self.source.clean(), None) self.source.value = None self.assertRaises( MalFormattedSource, lambda: self.source.clean() ) self.source.value = "test" self.source.requested = None self.assertRaises( MalFormattedSource, lambda: self.source.clean() ) def test_build_api_struct(self): self.kwargs.update({"type": "probes"}) self.assertEqual(self.source.build_api_struct(), self.kwargs) validate(self.source.build_api_struct(), probes_change_schema)
def test_set_tags(self): # missing action kwargs = { "requested": 5, "value": "test", "type": "msm", "tags": {"include": ["one", "two"], "exclude": ["one", "two"]} } self.assertRaises( MalFormattedSource, lambda: AtlasChangeSource(**kwargs).build_api_struct() ) # action == add kwargs = { "requested": 5, "value": "test", "type": "msm", "action": "add", "tags": {"include": ["one", "two"], "exclude": ["one", "two"]} } source = AtlasChangeSource(**kwargs) self.assertEqual(source.clean(), None) # action == remove kwargs = { "requested": 5, "value": "test", "type": "msm", "action": "remove", "tags": {"include": ["one", "two"], "exclude": ["one", "two"]} } self.assertRaises( MalFormattedSource, lambda: AtlasChangeSource(**kwargs) )