Implemented json functions and their tests

tests/commands/json/test_arrappend.py

@@ -0,0 +1,53 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+from typing import List
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_arrappend"
+    value: JSONValueT = {"int": 1, "array": [], "object": {"array": [ 1 ]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+
+def test_arrappend_single_element(redis: Redis):
+    key = "json_arrappend"
+    path = "$.array"
+
+    assert redis.json.arrappend(key, path, 1) == [1]
+    assert redis.json.arrappend(key, path, 'new val') == [2]
+    assert redis.json.arrappend(key, path, 1.5) == [3]
+    assert redis.json.arrappend(key, path, True) == [4]
+    assert redis.json.arrappend(key, path, [1]) == [5]
+    assert redis.json.arrappend(key, path, {"key": "value"}) == [6]
+    assert redis.json.get(key, path) == [[1, 'new val', 1.5, True, [1], {"key": "value"} ]]
+
+
+def test_arrappend_multiple_elements(redis: Redis):
+    key = "json_arrappend"
+    path = "$.array"
+    new_values: List[JSONValueT] = [1, 'new val', 1.5, True, [1], {"key": "value"}]
+
+    assert redis.json.arrappend(key, path, *new_values) == [6]
+    assert redis.json.get(key, path) == [[1, 'new val', 1.5, True, [1], {"key": "value"}]]
+
+
+def test_arrappend_nonarray_path(redis: Redis):
+    key = "json_arrappend"
+    path = "$.int"
+    new_value = 9
+
+    assert redis.json.arrappend(key, path, new_value) == [ None ]
+    assert redis.json.get(key, path) == [ 1 ]
+
+
+def test_arrappend_wildcard(redis: Redis):
+    key = "json_arrappend"
+    path = "$..array"
+    new_value = 9
+
+    assert redis.json.arrappend(key, path, new_value) == [2, 1]
+    assert redis.json.get(key, path) == [[1, 9], [9]]

tests/commands/json/test_arrindex.py

@@ -0,0 +1,43 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_arrindex"
+    value: JSONValueT = {"array": [1, 'test', ['a'], 1.5, {'test': 1}], "int": 1, "object": {"array": [ 0, 1 ]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+def test_arrindex_existing_element(redis: Redis):
+    key = "json_arrindex"
+    path = "$.array"
+
+    assert redis.json.arrindex(key, path, 1) == [0]
+    assert redis.json.arrindex(key, path, 'test') == [1]
+    assert redis.json.arrindex(key, path, ['a']) == [2]
+    assert redis.json.arrindex(key, path, 1.5) == [3]
+    assert redis.json.arrindex(key, path, {'test': 1}) == [4]
+
+def test_arrindex_nonexisting_element(redis: Redis):
+    key = "json_arrindex"
+    path = "$.array"
+    value = 4
+
+    assert redis.json.arrindex(key, path, value) == [-1]
+
+def test_arrindex_nonarray_path(redis: Redis):
+    key = "json_arrindex"
+    path = "$.int"
+    value = 4
+
+    assert redis.json.arrindex(key, path, value) == [None]
+
+def test_arrindex_wildcard(redis: Redis):
+    key = "json_arrindex"
+    path = "$..array"
+    value = 1
+
+    assert redis.json.arrindex(key, path, value) == [1, 0]

tests/commands/json/test_arrinsert.py

@@ -0,0 +1,54 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_arrinsert"
+    value: JSONValueT = {"int": 1, "array": [], "object": {"array": [0]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+
+def test_arrinsert_single_element(redis: Redis):
+    key = "json_arrinsert"
+    path = "$.array"
+
+    assert redis.json.arrinsert(key, path,  0,1) == [1]
+    assert redis.json.arrinsert(key, path,  0, 'new val') == [2]
+    assert redis.json.arrinsert(key, path,  0, 1.5) == [3]
+    assert redis.json.arrinsert(key, path,  0, True) == [4]
+    assert redis.json.arrinsert(key, path,  0, [1]) == [5]
+    assert redis.json.arrinsert(key, path,  0, {"key": "value"}) == [6]
+    assert redis.json.get(key, path) == [[ {"key": "value"}, [1],True , 1.5, 'new val', 1 ]]
+
+
+def test_arrinsert_multiple_elements(redis: Redis):
+    key = "json_arrinsert"
+    path = "$.array"
+    new_values = [5, 6, 7]
+
+    assert redis.json.arrinsert(key, path, 0, *new_values) == [3]
+    assert redis.json.get(key, path) == [[5, 6, 7]]
+
+
+def test_arrinsert_nonarray_path(redis: Redis):
+    key = "json_arrinsert"
+    path = "$.int"
+    new_value = 9
+    index = 0
+
+    assert redis.json.arrinsert(key, path, index, new_value) == [None]
+    assert redis.json.get(key, path) == [1]
+
+
+def test_arrinsert_wildcard(redis: Redis):
+    key = "json_arrinsert"
+    path = "$..array"
+    new_value = 9
+    index = 0
+
+    assert redis.json.arrinsert(key, path, index, new_value) == [2, 1]
+    assert redis.json.get(key, path) == [[9, 0], [9]]

tests/commands/json/test_arrlen.py

@@ -0,0 +1,33 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_arrlen"
+    value: JSONValueT = {"array": [1, 2, 3, 4], "int": 1, "object": {"array": [1, 2, 3]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+def test_arrlen_existing_element(redis: Redis):
+    key = "json_arrlen"
+    path = "$.array"
+
+    length = redis.json.arrlen(key, path)
+    assert length == [4]
+
+def test_arrlen_nonarray_path(redis: Redis):
+    key = "json_arrlen"
+    path = "$.int"
+
+    length = redis.json.arrlen(key, path)
+    assert length == [None]
+
+def test_arrlen_wildcard(redis: Redis):
+    key = "json_arrlen"
+    path = "$..array"
+
+    length = redis.json.arrlen(key, path)
+    assert length == [3, 4]

tests/commands/json/test_arrpop.py

@@ -0,0 +1,62 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_arrpop"
+    value: JSONValueT = {"int": 1, "array": [1, 2, 3, 4], "object": {"array": [5, 6, 7]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+
+def test_arrpop_single_element(redis: Redis):
+    key = "json_arrpop"
+    path = "$.array"
+    index = 0
+
+    removed_element = redis.json.arrpop(key, path, index)
+    assert removed_element == [1]
+    assert redis.json.get(key, path) == [[2, 3, 4]]
+
+
+def test_arrpop_negative_index(redis: Redis):
+    key = "json_arrpop"
+    path = "$.array"
+    index = -2
+
+    removed_element = redis.json.arrpop(key, path, index)
+    assert removed_element == [3]
+    assert redis.json.get(key, path) == [[1, 2, 4]]
+
+
+def test_arrpop_out_of_range_index(redis: Redis):
+    key = "json_arrpop"
+    path = "$.array"
+    index = 4
+
+    removed_element = redis.json.arrpop(key, path, index)
+    assert removed_element == [4]
+    assert redis.json.get(key, path) == [[1, 2, 3]]
+
+
+def test_arrpop_nonarray_path(redis: Redis):
+    key = "json_arrpop"
+    path = "$.int"
+    index = 0
+
+    removed_element = redis.json.arrpop(key, path, index)
+    assert removed_element == [None]
+    assert redis.json.get(key, path) == [1]
+
+
+def test_arrpop_wildcard(redis: Redis):
+    key = "json_arrpop"
+    path = "$..array"
+    index = 1
+
+    removed_element = redis.json.arrpop(key, path, index)
+    assert removed_element == [6, 2]
+    assert redis.json.get(key, path) == [[5, 7], [1, 3, 4]]

tests/commands/json/test_arrtrim.py

@@ -0,0 +1,78 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_arrtrim"
+    value: JSONValueT = {"int": 1, "array": [1, 2, 3, 4], "object": {"array": [5, 6, 7]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+
+def test_arrtrim_default(redis: Redis):
+    key = "json_arrtrim"
+    path = "$.array"
+    start = 1
+    stop = 2
+
+    newlen = redis.json.arrtrim(key, path, start, stop)
+    assert newlen == [2]
+    assert redis.json.get(key, path) == [[2, 3]]
+
+
+def test_arrtrim_negative_index(redis: Redis):
+    key = "json_arrtrim"
+    path = "$.array"
+    start = -2
+    stop = -1
+
+    newlen = redis.json.arrtrim(key, path, start, stop)
+    assert newlen == [2]
+    assert redis.json.get(key, path) == [[3, 4]]
+
+
+def test_arrtrim_out_of_range_index(redis: Redis):
+    key = "json_arrtrim"
+    path = "$.array"
+    start = 4
+    stop = 5
+
+    newlen = redis.json.arrtrim(key, path, start, stop)
+    assert newlen == [0]
+    assert redis.json.get(key, path) == [[]]
+
+
+def test_arrtrim_start_greater_than_stop(redis: Redis):
+    key = "json_arrtrim"
+    path = "$.array"
+    start = 1
+    stop = 0
+
+    newlen = redis.json.arrtrim(key, path, start, stop)
+    assert newlen == [0]
+    assert redis.json.get(key, path) == [[]]
+
+
+def test_arrtrim_nonarray_path(redis: Redis):
+    key = "json_arrtrim"
+    path = "$.int"
+    start = 0
+    stop = 0
+
+    newlen = redis.json.arrtrim(key, path, start, stop)
+    assert newlen == [None]
+    assert redis.json.get(key, path) == [1]
+
+
+def test_arrtrim_wildcard(redis: Redis):
+    key = "json_arrtrim"
+    path = "$..array"
+    start = 0
+    stop = 1
+
+    newlen = redis.json.arrtrim(key, path, start, stop)
+    assert newlen == [2, 2]
+    assert redis.json.get(key, path) == [[5, 6], [1, 2]]

tests/commands/json/test_clear.py

@@ -0,0 +1,30 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_clear"
+    value: JSONValueT = {"int": 1, "array": [1, 2, 3, 4], "object": {"array": [1, 2, 3]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+
+def test_clear_single_element(redis: Redis):
+    key = "json_clear"
+    path = "$.array"
+
+    removed_element = redis.json.clear(key, path)
+    assert removed_element == 1
+    assert redis.json.get(key) == [{"int": 1, "array": [], "object": {"array": [1, 2, 3]}}]
+
+
+def test_clear_wildcard(redis: Redis):
+    key = "json_clear"
+    path = "$..array"
+
+    removed_element = redis.json.clear(key, path)
+    assert removed_element == 2
+    assert redis.json.get(key) == [{"int": 1, "object": {"array": []}, "array": []}]

tests/commands/json/test_delete_json.py

@@ -0,0 +1,30 @@
+import pytest
+from upstash_redis import Redis
+from upstash_redis.typing import JSONValueT
+
+
+@pytest.fixture(autouse=True)
+def setup_json(redis: Redis):
+    json_key = "json_delete"
+    value: JSONValueT = {"int": 1, "array": [1, 2, 3, 4], "object": {"array": [1, 2, 3]}}
+    redis.json.set(json_key, "$", value)
+    yield
+    redis.delete(json_key)
+
+
+def test_delete_single_element(redis: Redis):
+    key = "json_delete"
+    path = "$.array"
+
+    removed_element = redis.json.delete(key, path)
+    assert removed_element == 1
+    assert redis.json.get(key) == [{"int": 1, "object": {"array": [1, 2, 3]}}]
+
+
+def test_delete_wildcard(redis: Redis):
+    key = "json_delete"
+    path = "$..array"
+
+    removed_element = redis.json.delete(key, path)
+    assert removed_element == 2
+    assert redis.json.get(key) == [{"int": 1, "object": {}}]