1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
"""Test autocomplete commands."""
import pytest
from sqlfluff.cli.autocomplete import dialect_shell_complete
@pytest.mark.parametrize(
"incomplete,expected",
[
["an", ["ansi"]],
["d", ["databricks", "db2", "doris", "duckdb"]],
["g", ["greenplum"]],
["s", ["snowflake", "soql", "sparksql", "sqlite", "starrocks"]],
["post", ["postgres"]],
],
)
def test_dialect_click_type_shell_complete(incomplete, expected):
"""Check that autocomplete returns dialects as expected."""
completion_items = dialect_shell_complete(
ctx="dummy_not_used", param="dummy_not_used", incomplete=incomplete
)
actual = [c.value for c in completion_items]
assert expected == actual
|
