def test_duplicate_mappings(self):
m = JSONMapper(mapping=[
Mapping(source='foo', destination='bar', transform=None),
Mapping(source='foz', destination='bar', transform=None),
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foz': 1}), {'bar': 1})
self.assertDictEqual(m.map({'foo': 2}), {'bar': 2})
self.assertDictEqual(m.map({'foo': 1, 'foz': 2}), {'bar': 2})
def test_big_example(self):
data = {
"store": {
"book": [{
"category": "reference",
"author": "Nigel Rees",
"title": "Sayings of the Century",
"price": 8.95
}, {
"category": "fiction",
"author": "Evelyn Waugh",
"title": "Sword of Honour",
"price": 12.99
}, {
"category": "fiction",
"author": "Herman Melville",
"title": "Moby Dick",
"isbn": "0-553-21311-3",
"price": 8.99
}, {
"category": "fiction",
"author": "J. R. R. Tolkien",
"title": "The Lord of the Rings",
"isbn": "0-395-19395-8",
"price": 22.99
}],
"bicycle": {
"color": "red",
"price": 19.95
}
},
"expensive": 10
}
m = JSONMapper(mapping=[
Mapping(source='store.book[*].author',
destination='authors',
transform=None),
Mapping(source='store.book[*].title',
destination='titles',
transform=None),
Mapping(source='store.book[*].price',
destination='prices',
transform=lambda x: x * 100),
])
self.assertDictEqual(
m.map(data), {
'authors': [book['author'] for book in data['store']['book']],
'titles': [book['title'] for book in data['store']['book']],
'prices':
[book['price'] * 100 for book in data['store']['book']],
})
def test_mapping_to_object(self):
m = JSONMapper(mapping=[
Mapping(source='foo.bar', destination='a.b.c.d', transform=None)
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': {
'a': 0,
'bar': 1
}}), {'a': {
'b': {
'c': {
'd': 1
}
}
}})
def test_merge_lists(self):
m = JSONMapper(mapping=[
Mapping(source='foo[*].f', destination='bar[*].b', transform=None),
Mapping(
source='foz[*].f', destination='bar[*].baz', transform=None)
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({
'foo': [{
'f': 1
}],
'foz': [{
'f': 2
}]
}), {'bar': [{
'b': 1,
'baz': 2
}]})
def test_simple_values_mapping(self):
m = JSONMapper(
mapping=[Mapping(source='foo', destination='bar', transform=None)])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': 1}), {'bar': 1})
self.assertDictEqual(m.map({'foo': 1, 'bar': 2}), {'bar': 1})
self.assertDictEqual(m.map({'foo': [1, 2]}), {'bar': [1, 2]})
self.assertDictEqual(m.map({'foo': {'prop': 1}}), {'bar': {'prop': 1}})
def test_list_item_to_root(self):
m = JSONMapper(
mapping=[Mapping(source='foo', destination='$', transform=None)])
self.assertDictEqual(m.map({}), {})
self.assertListEqual(m.map({'foo': [0, 1, 2, 3]}), [0, 1, 2, 3])
self.assertListEqual(
m.map({'foo': [{
'a': 1,
'b': 2
}, {
'c': 3,
'p': 0
}]}), [{
'a': 1,
'b': 2
}, {
'c': 3,
'p': 0
}])
def test_two_items_to_root(self):
m = JSONMapper(mapping=[
Mapping(source='foo', destination='$', transform=None),
Mapping(source='bar', destination='$', transform=None),
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': {'a': 1}}), {'a': 1})
self.assertDictEqual(m.map({'bar': {'b': 2}}), {'b': 2})
self.assertDictEqual(m.map({
'foo': {
'a': 1
},
'bar': {
'b': 2
}
}), {
'a': 1,
'b': 2
})
def test_nested_objects_mapping_to_simple_value(self):
m = JSONMapper(mapping=[
Mapping(source='foo.bar', destination='bar', transform=None)
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': 1}), {})
self.assertDictEqual(m.map({'foo': {'bar': 1}}), {'bar': 1})
self.assertDictEqual(m.map({'foo': {'bar': 1}, 'bar': 2}), {'bar': 1})
def test_multiple_mappings(self):
m = JSONMapper(mapping=[
Mapping(source='foo', destination='bar', transform=None),
Mapping(source='foz', destination='baz', transform=None),
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': [0, 1, 2, 3]}),
{'bar': [0, 1, 2, 3]})
self.assertDictEqual(m.map({'foz': [0, 1, 2, 3]}),
{'baz': [0, 1, 2, 3]})
self.assertDictEqual(m.map({
'foo': [0, 1, 2, 3],
'foz': [0, 1, 2, 3]
}), {
'bar': [0, 1, 2, 3],
'baz': [0, 1, 2, 3]
})
def test_map_returns_dict(self):
m = JSONMapper(mapping=[])
self.assertDictEqual(m.map({}), {})
def test_item_property_in_list_to_another_property_in_list(self):
m = JSONMapper(mapping=[
Mapping(source='foo[*].f', destination='bar[*].b', transform=None)
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': [{'f': 1}]}), {'bar': [{'b': 1}]})
def test_list_item_to_simple_value(self):
m = JSONMapper(mapping=[
Mapping(source='foo[1]', destination='bar', transform=None)
])
self.assertDictEqual(m.map({}), {})
self.assertDictEqual(m.map({'foo': [0, 1, 2, 3]}), {'bar': 1})