File: test_lexer.py

package info (click to toggle)
jsonpath-ng 1.7.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 372 kB
  • sloc: python: 2,172; makefile: 4
file content (56 lines) | stat: -rw-r--r-- 1,762 bytes parent folder | download
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
import pytest

from jsonpath_ng.lexer import JsonPathLexer, JsonPathLexerError

token_test_cases = (
    ("$", (("$", "$"),)),
    ('"hello"', (("hello", "ID"),)),
    ("'goodbye'", (("goodbye", "ID"),)),
    ("'doublequote\"'", (('doublequote"', "ID"),)),
    (r'"doublequote\""', (('doublequote"', "ID"),)),
    (r"'singlequote\''", (("singlequote'", "ID"),)),
    ('"singlequote\'"', (("singlequote'", "ID"),)),
    ("fuzz", (("fuzz", "ID"),)),
    ("1", ((1, "NUMBER"),)),
    ("45", ((45, "NUMBER"),)),
    ("-1", ((-1, "NUMBER"),)),
    (" -13 ", ((-13, "NUMBER"),)),
    ('"fuzz.bang"', (("fuzz.bang", "ID"),)),
    ("fuzz.bang", (("fuzz", "ID"), (".", "."), ("bang", "ID"))),
    ("fuzz.*", (("fuzz", "ID"), (".", "."), ("*", "*"))),
    ("fuzz..bang", (("fuzz", "ID"), ("..", "DOUBLEDOT"), ("bang", "ID"))),
    ("&", (("&", "&"),)),
    ("@", (("@", "ID"),)),
    ("`this`", (("this", "NAMED_OPERATOR"),)),
    ("|", (("|", "|"),)),
    ("where", (("where", "WHERE"),)),
    ("wherenot", (("wherenot", "WHERENOT"),)),
)


@pytest.mark.parametrize("string, expected_token_info", token_test_cases)
def test_lexer(string, expected_token_info):
    lexer = JsonPathLexer(debug=True)
    tokens = list(lexer.tokenize(string))
    assert len(tokens) == len(expected_token_info)
    for token, (expected_value, expected_type) in zip(tokens, expected_token_info):
        assert token.type == expected_type
        assert token.value == expected_value


invalid_token_test_cases = (
    "'\"",
    "\"'",
    '`"',
    "`'",
    '"`',
    "'`",
    "?",
    "$.foo.bar.#",
)


@pytest.mark.parametrize("string", invalid_token_test_cases)
def test_lexer_errors(string):
    with pytest.raises(JsonPathLexerError):
        list(JsonPathLexer().tokenize(string))