def test_invalid_input(self):
input_str = (
'{"Mags": [{"Key": "tag1", "Value": "value1"}, {"Key": "tag2", "Value": "value2"}]}'
)
field = TagsField(optional=False)
input_data = json.loads(input_str)
with self.assertRaises(TagsFieldMissingTagsKeyException):
field.parse(data=input_data, context={})
def test_optional(self):
input_str = '{"NoTagsHere": []}'
field = TagsField(optional=True)
input_data = json.loads(input_str)
link_collection = field.parse(data=input_data, context={})
self.assertCountEqual(link_collection, LinkCollection())
def test_optional(self):
input_str = '{"NoTagsHere": []}'
field = TagsField(optional=True)
expected_output_data = []
input_data = json.loads(input_str)
links = field.parse(data=input_data, context={})
output_data = [link.to_dict() for link in links]
self.assertCountEqual(output_data, expected_output_data)
def test_valid_input(self):
input_str = (
'{"Tags": [{"Key": "tag1", "Value": "value1"}, {"Key": "tag2", "Value": "value2"}]}'
)
field = TagsField()
expected_output_data = [
{"pred": "tag1", "obj": "value1", "type": "tag"},
{"pred": "tag2", "obj": "value2", "type": "tag"},
]
input_data = json.loads(input_str)
links = field.parse(data=input_data, context={})
output_data = [link.to_dict() for link in links]
self.assertCountEqual(output_data, expected_output_data)
def test_valid_input(self):
input_str = (
'{"Tags": [{"Key": "tag1", "Value": "value1"}, {"Key": "tag2", "Value": "value2"}]}'
)
field = TagsField()
expected_output_data = [
{
"pred": "tag1",
"obj": "value1",
"type": "tag"
},
{
"pred": "tag2",
"obj": "value2",
"type": "tag"
},
]
input_data = json.loads(input_str)
link_collection = field.parse(data=input_data, context={})
expected_link_collection = LinkCollection(tag_links=(TagLink(
pred="tag1", obj="value1"), TagLink(pred="tag2", obj="value2")), )
self.assertEqual(link_collection, expected_link_collection)