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
57
58
59
60
61
62
63
64
65
66
67
68
|
"""The Test file for The New Parser (Grammar Classes)."""
from sqlfluff.core import FluffConfig
from sqlfluff.core.errors import SQLParseError
from sqlfluff.core.linter.linter import Linter
from sqlfluff.core.parser import Anything, BaseSegment, KeywordSegment, StringParser
from sqlfluff.core.parser.context import ParseContext
BarKeyword = StringParser("bar", KeywordSegment)
class BasicSegment(BaseSegment):
"""A basic segment for testing parse and expand."""
type = "basic"
match_grammar = Anything()
def test__parser__parse_match(test_segments):
"""Test match method on a real segment."""
ctx = ParseContext(dialect=None)
# This should match and have consumed everything, which should
# now be part of a BasicSegment.
match = BasicSegment.match(test_segments, 0, parse_context=ctx)
assert match
matched = match.apply(test_segments)
assert len(matched) == 1
assert isinstance(matched[0], BasicSegment)
assert matched[0].segments[0].type == "raw"
def test__parser__parse_error():
"""Test that SQLParseError is raised for unparsable section."""
in_str = "SELECT ;"
lnt = Linter(dialect="ansi")
parsed = lnt.parse_string(in_str)
assert len(parsed.violations) == 1
violation = parsed.violations[0]
assert isinstance(violation, SQLParseError)
assert violation.desc() == "Line 1, Position 1: Found unparsable section: 'SELECT'"
# Check that the expected labels work for logging.
# TODO: This is more specific that in previous iterations, but we could
# definitely make this easier to read.
assert (
'Expected: "<Delimited: '
"[<Ref: 'SelectClauseElementSegment'>]> "
"after <WordSegment: ([L: 1, P: 1]) 'SELECT'>. "
"Found nothing."
) in parsed.tree.stringify()
def test_parse_jinja_macro_exclude():
"""Test parsing when excluding macros with unknown tags.
This test case has a file which defines the unknown tag `materialization` which
would cause a templating error if not excluded. By ignoring that folder we can
ensure there are no errors.
"""
config_path = "test/fixtures/templater/jinja_exclude_macro_path/.sqlfluff"
config = FluffConfig.from_path(config_path)
linter = Linter(config=config)
sql_file_path = "test/fixtures/templater/jinja_exclude_macro_path/jinja.sql"
parsed = linter.parse_path(sql_file_path)
for parse in parsed:
assert parse.violations == []
|
