python-jsonpath-rfc9535/tests/test_node.py at d92bf4f9a0b0b000463f186080a1d9d13d15c93a · jg-rp/python-jsonpath-rfc9535 · GitHub

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
from jsonpath_rfc9535 import find_one


def test_parent() -> None:
    data = {"a": {"b": {"c": 1}}}
    query = "$.a.b.c"
    node = find_one(query, data)
    assert node is not None
    assert node.value == 1
    assert node.parent is not None
    assert node.parent.value == data["a"]["b"]


def test_parent_of_root() -> None:
    data = {"a": {"b": {"c": 1}}}
    query = "$"
    node = find_one(query, data)
    assert node is not None
    assert node.value == data
    assert node.parent is None


def test_set_dict_value() -> None:
    data = {"a": {"b": {"c": 1}}}
    query = "$.a.b.c"
    node = find_one(query, data)
    assert node is not None

    new_value = 99
    node.value = new_value
    assert node.value == new_value
    assert data["a"]["b"]["c"] == new_value


def test_set_list_value() -> None:
    data = {"a": {"b": [1, 2, 3]}}
    query = "$.a.b[1]"
    node = find_one(query, data)
    assert node is not None

    new_value = 99
    node.value = new_value
    assert node.value == new_value
    assert data["a"]["b"][1] == new_value


def test_set_root_value() -> None:
    data = {"a": {"b": {"c": 1}}}
    query = "$"
    node = find_one(query, data)
    assert node is not None

    new_value = 99
    node.value = new_value
    assert node.value == new_value
    assert data == {"a": {"b": {"c": 1}}}