summaryrefslogtreecommitdiffstats
path: root/tests/test_jsonpath.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-01-31 05:44:37 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-01-31 05:44:37 +0000
commit5f8be2e0852f3c925fb873a48946caee3050899f (patch)
tree1f31666277e226f47180321c08be7ebbedc2780e /tests/test_jsonpath.py
parentAdding upstream version 20.9.0. (diff)
downloadsqlglot-5f8be2e0852f3c925fb873a48946caee3050899f.tar.xz
sqlglot-5f8be2e0852f3c925fb873a48946caee3050899f.zip
Adding upstream version 20.11.0.upstream/20.11.0
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/test_jsonpath.py')
-rw-r--r--tests/test_jsonpath.py140
1 files changed, 140 insertions, 0 deletions
diff --git a/tests/test_jsonpath.py b/tests/test_jsonpath.py
new file mode 100644
index 0000000..01cd899
--- /dev/null
+++ b/tests/test_jsonpath.py
@@ -0,0 +1,140 @@
+import json
+import os
+import unittest
+
+from sqlglot import jsonpath
+from sqlglot.errors import ParseError, TokenError
+from tests.helpers import FIXTURES_DIR
+
+
+class TestJsonpath(unittest.TestCase):
+ maxDiff = None
+
+ def test_jsonpath(self):
+ self.assertEqual(
+ jsonpath.parse("$.*.a[0]['x'][*, 'y', 1].z[?(@.a == 'b'), 1:][1:5][1,?@.a][(@.x)]"),
+ [
+ {"kind": "root"},
+ {"kind": "child", "value": "*"},
+ {"kind": "child", "value": "a"},
+ {"kind": "subscript", "value": 0},
+ {"kind": "key", "value": "x"},
+ {"kind": "union", "value": [{"kind": "wildcard"}, "y", 1]},
+ {"kind": "child", "value": "z"},
+ {"kind": "selector", "value": {"kind": "filter", "value": "(@.a == 'b'), 1:"}},
+ {
+ "kind": "subscript",
+ "value": {"end": 5, "kind": "slice", "start": 1, "step": None},
+ },
+ {"kind": "union", "value": [1, {"kind": "filter", "value": "@.a"}]},
+ {"kind": "selector", "value": {"kind": "script", "value": "@.x)"}},
+ ],
+ )
+
+ def test_identity(self):
+ for selector, expected in (
+ ("$.select", "$.select"),
+ ("$[(@.length-1)]", "$[(@.length-1)]"),
+ ("$[((@.length-1))]", "$[((@.length-1))]"),
+ ):
+ with self.subTest(f"{selector} -> {expected}"):
+ self.assertEqual(jsonpath.generate(jsonpath.parse(selector)), expected)
+
+ def test_cts_file(self):
+ with open(os.path.join(FIXTURES_DIR, "jsonpath", "cts.json")) as file:
+ tests = json.load(file)["tests"]
+
+ # sqlglot json path generator rewrites to a normal form
+ overrides = {
+ """$['a',1]""": """$["a",1]""",
+ """$[*,'a']""": """$[*,"a"]""",
+ """$..['a','d']""": """$..["a","d"]""",
+ """$[1, ?@.a=='b', 1:]""": """$[1,?@.a=='b', 1:]""",
+ """$["a"]""": """$.a""",
+ """$["c"]""": """$.c""",
+ """$['a']""": """$.a""",
+ """$['c']""": """$.c""",
+ """$[' ']""": """$[" "]""",
+ """$['\\'']""": """$["\'"]""",
+ """$['\\\\']""": """$["\\\\"]""",
+ """$['\\/']""": """$["\\/"]""",
+ """$['\\b']""": """$["\\b"]""",
+ """$['\\f']""": """$["\\f"]""",
+ """$['\\n']""": """$["\\n"]""",
+ """$['\\r']""": """$["\\r"]""",
+ """$['\\t']""": """$["\\t"]""",
+ """$['\\u263A']""": """$["\\u263A"]""",
+ """$['\\u263a']""": """$["\\u263a"]""",
+ """$['\\uD834\\uDD1E']""": """$["\\uD834\\uDD1E"]""",
+ """$['\\uD83D\\uDE00']""": """$["\\uD83D\\uDE00"]""",
+ """$['']""": """$[""]""",
+ """$[? @.a]""": """$[?@.a]""",
+ """$[?\n@.a]""": """$[?@.a]""",
+ """$[?\t@.a]""": """$[?@.a]""",
+ """$[?\r@.a]""": """$[?@.a]""",
+ """$[? (@.a)]""": """$[?(@.a)]""",
+ """$[?\n(@.a)]""": """$[?(@.a)]""",
+ """$[?\t(@.a)]""": """$[?(@.a)]""",
+ """$[?\r(@.a)]""": """$[?(@.a)]""",
+ """$[ ?@.a]""": """$[?@.a]""",
+ """$[\n?@.a]""": """$[?@.a]""",
+ """$[\t?@.a]""": """$[?@.a]""",
+ """$[\r?@.a]""": """$[?@.a]""",
+ """$ ['a']""": """$.a""",
+ """$\n['a']""": """$.a""",
+ """$\t['a']""": """$.a""",
+ """$\r['a']""": """$.a""",
+ """$['a'] ['b']""": """$.a.b""",
+ """$['a'] \n['b']""": """$.a.b""",
+ """$['a'] \t['b']""": """$.a.b""",
+ """$['a'] \r['b']""": """$.a.b""",
+ """$ .a""": """$.a""",
+ """$\n.a""": """$.a""",
+ """$\t.a""": """$.a""",
+ """$\r.a""": """$.a""",
+ """$[ 'a']""": """$.a""",
+ """$[\n'a']""": """$.a""",
+ """$[\t'a']""": """$.a""",
+ """$[\r'a']""": """$.a""",
+ """$['a' ]""": """$.a""",
+ """$['a'\n]""": """$.a""",
+ """$['a'\t]""": """$.a""",
+ """$['a'\r]""": """$.a""",
+ """$['a' ,'b']""": """$["a","b"]""",
+ """$['a'\n,'b']""": """$["a","b"]""",
+ """$['a'\t,'b']""": """$["a","b"]""",
+ """$['a'\r,'b']""": """$["a","b"]""",
+ """$['a', 'b']""": """$["a","b"]""",
+ """$['a',\n'b']""": """$["a","b"]""",
+ """$['a',\t'b']""": """$["a","b"]""",
+ """$['a',\r'b']""": """$["a","b"]""",
+ """$[1 :5:2]""": """$[1:5:2]""",
+ """$[1\n:5:2]""": """$[1:5:2]""",
+ """$[1\t:5:2]""": """$[1:5:2]""",
+ """$[1\r:5:2]""": """$[1:5:2]""",
+ """$[1: 5:2]""": """$[1:5:2]""",
+ """$[1:\n5:2]""": """$[1:5:2]""",
+ """$[1:\t5:2]""": """$[1:5:2]""",
+ """$[1:\r5:2]""": """$[1:5:2]""",
+ """$[1:5 :2]""": """$[1:5:2]""",
+ """$[1:5\n:2]""": """$[1:5:2]""",
+ """$[1:5\t:2]""": """$[1:5:2]""",
+ """$[1:5\r:2]""": """$[1:5:2]""",
+ """$[1:5: 2]""": """$[1:5:2]""",
+ """$[1:5:\n2]""": """$[1:5:2]""",
+ """$[1:5:\t2]""": """$[1:5:2]""",
+ """$[1:5:\r2]""": """$[1:5:2]""",
+ }
+
+ for test in tests:
+ selector = test["selector"]
+
+ with self.subTest(f"{selector.strip()} /* {test['name']} */"):
+ if test.get("invalid_selector"):
+ try:
+ jsonpath.parse(selector)
+ except (ParseError, TokenError):
+ pass
+ else:
+ nodes = jsonpath.parse(selector)
+ self.assertEqual(jsonpath.generate(nodes), overrides.get(selector, selector))