python/nested-lookup
1
0
Fork 0
Code Issues 6 Pull requests Projects Releases Packages Wiki Activity Actions

Add test for recursive use of nested_alter

Browse source
This commit is contained in:
Daniel Hjertholm 2021-03-01 14:56:10 +01:00 • committed by Russell Ballestrini
parent 81bda0dfab
commit 5c5a65452a
1 changed files with 17 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
test_lookup_api.py
View file

@ -585,17 +585,31 @@ class TestNestedAlter(BaseLookUpApi):
self.assertEqual(out[1]["salsa"][0]["burrito"]["taco"], 79) self.assertEqual(out[1]["salsa"][0]["burrito"]["taco"], 79)
def test_nested_alter_work_with_right_order(self): def test_nested_alter_work_with_right_order(self):
document = {"taco": 42, "salsa": [{"burrito":{"key":20}}], "key":50} document = {"taco": 42, "salsa": [{"burrito": {"key": 20}}], "key": 50}
def callback(data): def callback(data):
return data + 100 return data + 100
altered_document = nested_alter(document, "key", callback, in_place=True) altered_document = nested_alter(document, "key", callback, in_place=True)
self.assertEqual(altered_document["salsa"][0]["burrito"]["key"], 120) self.assertEqual(altered_document["salsa"][0]["burrito"]["key"], 120)
self.assertEqual(altered_document["key"], 150) self.assertEqual(altered_document["key"], 150)
def test_nested_alter_recursive(self):
document = {"key": {"rename me": {"key": {"rename me": 1}}}}
def callback(data):
try:
data["renamed"] = data.pop("rename me")
except KeyError:
pass
return data
altered_document = nested_alter(document, "key", callback, in_place=True)
self.assertIn("renamed", altered_document["key"].keys())
self.assertIn("renamed", altered_document["key"]["renamed"]["key"].keys())
def test_sample_data4(self): def test_sample_data4(self):
result = { result = {