class TestAtlasSource(unittest.TestCase):
def setUp(self):
self.kwargs = {"requested": 5, "value": "test", "type": "country"}
self.source = AtlasSource(**self.kwargs)
def test_setting_type(self):
self.assertRaises(
MalFormattedSource, lambda: setattr(self.source, "type", "testing")
)
def test_clean(self):
self.assertEqual(self.source.clean(), None)
self.source.type = "msm"
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.assertEqual(self.source.build_api_struct(), self.kwargs)
validate(self.source.build_api_struct(), probes_create_schema)
def test_clean(self):
# all ok
kwargs = {"requested": 5, "value": "test", "type": "msm"}
source = AtlasSource(**kwargs)
self.assertEqual(source.clean(), None)
# value missing
source.value = None
self.assertRaises(
MalFormattedSource, lambda: source.clean()
)
# type missing
kwargs = {"requested": 5, "value": "test"}
self.assertRaises(
MalFormattedSource, lambda: AtlasSource(**kwargs).clean()
)
# requested missing
source.value = "test"
source.requested = None
self.assertRaises(
MalFormattedSource, lambda: source.clean()
)
def test_clean(self):
# all ok
kwargs = {"requested": 5, "value": "test", "type": "msm"}
source = AtlasSource(**kwargs)
self.assertEqual(source.clean(), None)
# value missing
source.value = None
self.assertRaises(
MalFormattedSource, lambda: source.clean()
)
# type missing
kwargs = {"requested": 5, "value": "test"}
self.assertRaises(
MalFormattedSource, lambda: AtlasSource(**kwargs).clean()
)
# requested missing
source.value = "test"
source.requested = None
self.assertRaises(
MalFormattedSource, lambda: source.clean()
)
def test_set_tags(self):
# all ok
kwargs = {
"requested": 5, "value": "test", "type": "msm",
"tags": {"include": ["one", "two"], "exclude": ["one", "two"]}}
source = AtlasSource(**kwargs)
self.assertEqual(source.clean(), None)
self.assertEqual(
source.tags,
{"include": ["one", "two"], "exclude": ["one", "two"]}
)
# include missing
kwargs = {
"requested": 5, "value": "test", "type": "msm",
"tags": {"exclude": ["one", "two"]}
}
source = AtlasSource(**kwargs)
self.assertEqual(source.clean(), None)
# exclude missing
kwargs = {
"requested": 5, "value": "test", "type": "msm",
"tags": {"include": ["one", "two"]}
}
source = AtlasSource(**kwargs)
self.assertEqual(source.clean(), None)
# invalid tag type
kwargs = {
"requested": 5, "value": "test", "type": "msm",
"tags": {"include": ["one", 2], "exclude": ["one", "two"]}
}
self.assertRaises(MalFormattedSource, lambda: AtlasSource(**kwargs))
# unknown element
kwargs = {
"requested": 5, "value": "test", "type": "msm",
"tags": {
"include": ["one", 2], "exclude": ["one", "two"], "unknown": "?"
}
}
self.assertRaises(MalFormattedSource, lambda: AtlasSource(**kwargs))
