Adding upstream version 20.11.0.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
8a068da99c
commit
07f4660f31
GPG key ID:
FBB4F0E80A80222F
141 changed files with 66644 additions and 54334 deletions
140
tests/test_jsonpath.py
Normal file
140
tests/test_jsonpath.py
Normal file
|
|
@ -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))
|
||||
Loading…
Add table
Add a link
Reference in a new issue