def test_visit_dicts_for_lists(self):
collection = [{
'a': 'cake',
'b': {
'foo': 'bar'
}
}, {
'c': 'fish',
'd': {
'baz': 'qux'
}
}]
visit_dicts(collection, self.traverse)
self.assertEqual(collection, [{
'visited': True,
'a': 'cake',
'b': {
'visited': True,
'foo': 'bar',
},
}, {
'visited': True,
'c': 'fish',
'd': {
'visited': True,
'baz': 'qux',
},
}])
def test_visit_dicts(self):
collection = {
'a': 'cake',
'b': {'foo': 'bar'},
}
visit_dicts(collection, self.traverse)
self.assertEqual(collection, {
'a': 'cake',
'b': {
'visited': True,
'foo': 'bar',
}
})
def test_visit_dicts(self):
collection = {
'a': 'cake',
'b': {
'foo': 'bar'
},
}
visit_dicts(collection, self.traverse)
self.assertEqual(collection, {
'a': 'cake',
'b': {
'visited': True,
'foo': 'bar',
}
})
def test_visit_dicts_for_multiple_dict_nestings(self):
collection = [{
'a': 'cake',
'b': {
'foo': 'bar'
}
}, {
'c': 'fish',
'd': {
'baz': [{
'spam': {
'lerp': 'larp'
},
}]
}
}]
visit_dicts(collection, self.traverse)
self.assertEqual(
collection,
[
{
'visited': True,
'a': 'cake',
'b': {
'visited': True,
'foo': 'bar',
}
},
{
'visited': True,
'c': 'fish',
'd': {
'visited':
True,
'baz': [{
'visited': True,
'spam': {
# pep8 or pyflakes somehow isn't happy with this
# being indented properly
'visited': True,
'lerp': 'larp',
},
}]
}
}
])
def test_visit_dicts_for_multiple_dict_nestings(self):
collection = [{
'a': 'cake',
'b': {'foo': 'bar'}
}, {
'c': 'fish',
'd': {
'baz': [{
'spam': {'lerp': 'larp'},
}]
}
}]
visit_dicts(collection, self.traverse)
self.assertEqual(collection, [{
'visited': True,
'a': 'cake',
'b': {
'visited': True,
'foo': 'bar',
}
}, {
'visited': True,
'c': 'fish',
'd': {
'visited': True,
'baz': [{
'visited': True,
'spam': {
# pep8 or pyflakes somehow isn't happy with this
# being indented properly
'visited': True,
'lerp': 'larp',
},
}]
}
}])
def test_visit_dicts_for_lists(self):
collection = [{
'a': 'cake',
'b': {'foo': 'bar'}
}, {
'c': 'fish',
'd': {'baz': 'qux'}
}]
visit_dicts(collection, self.traverse)
self.assertEqual(collection, [{
'visited': True,
'a': 'cake',
'b': {
'visited': True,
'foo': 'bar',
},
}, {
'visited': True,
'c': 'fish',
'd': {
'visited': True,
'baz': 'qux',
},
}])
