File: std_test.py

package info (click to toggle)
sqlfluff 3.5.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 34,000 kB
  • sloc: python: 106,131; sql: 34,188; makefile: 52; sh: 8
file content (107 lines) | stat: -rw-r--r-- 3,670 bytes parent folder | download | duplicates (2)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
"""Tests for the standard set of rules."""

import pytest

from sqlfluff.core.config import FluffConfig
from sqlfluff.core.rules import get_ruleset
from sqlfluff.utils.testing.rules import assert_rule_raises_violations_in_file


@pytest.mark.parametrize(
    "rule,path,violations",
    [
        ("LT01", "indentation_errors.sql", [(4, 24)]),
        (
            "LT02",
            "indentation_errors.sql",
            [(2, 1), (3, 1), (4, 1), (5, 1)],
        ),
        # Check we get comma whitespace errors
        ("LT01", "whitespace_errors.sql", [(2, 9), (3, 12)]),
        # Check we get operator whitespace errors and it works with brackets
        (
            "LT01",
            "operator_errors.sql",
            [(3, 8), (4, 10), (7, 6), (7, 7), (7, 9), (7, 10), (7, 12), (7, 13)],
        ),
        ("LT03", "operator_errors.sql", [(5, 9)]),
        (
            "LT01",
            "operator_errors_negative.sql",
            [(2, 6), (2, 9), (5, 6), (5, 7)],
        ),
        # Hard indentation errors
        (
            "LT02",
            "indentation_error_hard.sql",
            [
                (2, 1),
                (6, 1),
                (9, 1),
                (11, 15),
                (12, 1),
                (12, 33),
                (13, 15),
                (14, 1),
                (14, 36),
                (18, 1),
                (19, 1),
                (20, 1),
            ],
        ),
        # Check bracket handling with closing brackets and contained indents works.
        ("LT02", "indentation_error_contained.sql", []),
        # Check we handle block comments as expect. Github #236
        (
            "LT05",
            "block_comment_errors.sql",
            # Errors should flag on the first element of the line.
            [(1, 1), (2, 5), (4, 5)],
        ),
        ("LT05", "block_comment_errors_2.sql", [(1, 1), (2, 1)]),
        # Column references
        ("RF02", "column_references.sql", [(1, 8)]),
        ("RF02", "column_references_bare_function.sql", []),
        ("RF01", "column_references.sql", [(1, 11)]),
        ("AL05", "column_references.sql", [(2, 11)]),
        # Distinct and Group by
        ("AM01", "select_distinct_group_by.sql", [(1, 8)]),
        # Make sure that ignoring works as expected
        ("LT01", "operator_errors_ignore.sql", [(10, 8), (10, 9)]),
        (
            "JJ01",
            "heavy_templating.sql",
            [(12, 13), (12, 25)],
        ),
    ],
)
def test__rules__std_file(rule, path, violations):
    """Test the linter finds the given errors in (and only in) the right places."""
    assert_rule_raises_violations_in_file(
        rule=rule,
        fpath="test/fixtures/linter/" + path,
        violations=violations,
        fluff_config=FluffConfig(overrides=dict(rules=rule, dialect="ansi")),
    )


@pytest.mark.parametrize(
    "rule_config_dict",
    [
        {"allow_scalar": "blah"},
        {"single_table_references": "blah"},
        {"unquoted_identifiers_policy": "blah"},
        {"capitalisation.keywords": {"capitalisation_policy": "blah"}},
        {"aliasing.table": {"aliasing": "blah"}},
        {"aliasing.column": {"aliasing": "blah"}},
        {"capitalisation.identifiers": {"extended_capitalisation_policy": "blah"}},
        {"capitalisation.functions": {"capitalisation_policy": "blah"}},
    ],
)
def test_improper_configs_are_rejected(rule_config_dict):
    """Ensure that unsupported configs raise a ValueError."""
    config = FluffConfig(
        configs={"rules": rule_config_dict}, overrides={"dialect": "ansi"}
    )
    with pytest.raises(ValueError):
        get_ruleset().get_rulepack(config)