1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
from moto.dynamodb.models.utilities import find_nested_key
class TestFindDictionaryKeys:
def setup_method(self):
self.item = {
"simplestring": "val",
"nesteddict": {
"level21": {"level3.1": "val", "level3.2": "val"},
"level22": {"level3.1": "val", "level3.2": "val"},
"nestedlist": [
{"ll21": {"ll3.1": "val", "ll3.2": "val"}},
{"ll22": {"ll3.1": "val", "ll3.2": "val"}},
],
},
"rootlist": [
{"ll21": {"ll3.1": "val", "ll3.2": "val"}},
{"ll22": {"ll3.1": "val", "ll3.2": "val"}},
],
}
def test_find_nothing(self):
assert find_nested_key([""], self.item) == {}
def test_find_unknown_key(self):
assert find_nested_key(["unknown"], self.item) == {}
def test_project_single_key_string(self):
assert find_nested_key(["simplestring"], self.item) == {"simplestring": "val"}
def test_project_single_key_dict(self):
assert find_nested_key(["nesteddict"], self.item) == {
"nesteddict": {
"level21": {"level3.1": "val", "level3.2": "val"},
"level22": {"level3.1": "val", "level3.2": "val"},
"nestedlist": [
{"ll21": {"ll3.1": "val", "ll3.2": "val"}},
{"ll22": {"ll3.1": "val", "ll3.2": "val"}},
],
}
}
def test_project_nested_key(self):
assert find_nested_key(["nesteddict", "level21"], self.item) == {
"nesteddict": {"level21": {"level3.1": "val", "level3.2": "val"}}
}
def test_project_multi_level_nested_key(self):
assert find_nested_key(["nesteddict", "level21", "level3.2"], self.item) == {
"nesteddict": {"level21": {"level3.2": "val"}}
}
def test_project_nested_key__partial_fix(self):
assert find_nested_key(["nesteddict", "levelunknown"], self.item) == {}
def test_project_nested_key__partial_fix2(self):
assert find_nested_key(["nesteddict", "unknown", "unknown2"], self.item) == {}
def test_list_index(self):
assert find_nested_key(["rootlist[0]"], self.item) == {
"rootlist": [{"ll21": {"ll3.1": "val", "ll3.2": "val"}}]
}
def test_nested_list_index(self):
assert find_nested_key(["nesteddict", "nestedlist[1]"], self.item) == {
"nesteddict": {"nestedlist": [{"ll22": {"ll3.1": "val", "ll3.2": "val"}}]}
}
def test_nested_obj_in_list(self):
assert find_nested_key(
["nesteddict", "nestedlist[1]", "ll22", "ll3.1"], self.item
) == {"nesteddict": {"nestedlist": [{"ll22": {"ll3.1": "val"}}]}}
def test_list_unknown_indexes(self):
assert find_nested_key(["nesteddict", "nestedlist[25]"], self.item) == {}
|
