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
108
109
110
111
|
import pytest
from sqlfmt.api import format_string
from sqlfmt.mode import Mode
from tests.util import check_formatting, read_test_data
@pytest.mark.parametrize(
"p",
[
"preformatted/001_select_1.sql",
"preformatted/002_select_from_where.sql",
"preformatted/003_literals.sql",
"preformatted/004_with_select.sql",
"preformatted/005_fmt_off.sql",
"preformatted/006_fmt_off_447.sql",
"preformatted/007_fmt_off_comments.sql",
"preformatted/008_reserved_names.sql",
"preformatted/009_empty.sql",
"preformatted/010_comment_only.sql",
"preformatted/301_multiline_jinjafmt.sql",
"preformatted/302_jinjafmt_multiline_str.sql",
"preformatted/303_jinjafmt_more_mutliline_str.sql",
"preformatted/400_create_table.sql",
"preformatted/401_create_row_access_policy.sql",
"preformatted/402_alter_table.sql",
"unformatted/100_select_case.sql",
"unformatted/101_multiline.sql",
"unformatted/102_lots_of_comments.sql",
"unformatted/103_window_functions.sql",
"unformatted/104_joins.sql",
"unformatted/106_leading_commas.sql",
"unformatted/107_jinja_blocks.sql",
"unformatted/108_test_block.sql",
"unformatted/109_lateral_flatten.sql",
"unformatted/110_other_identifiers.sql",
"unformatted/111_chained_boolean_between.sql",
"unformatted/112_semicolons.sql",
"unformatted/113_utils_group_by.sql",
"unformatted/114_unions.sql",
"unformatted/115_select_star_except.sql",
"unformatted/116_chained_booleans.sql",
"unformatted/117_whitespace_in_tokens.sql",
"unformatted/118_within_group.sql",
"unformatted/119_psycopg_placeholders.sql",
"unformatted/120_array_literals.sql",
"unformatted/121_stubborn_merge_edge_cases.sql",
"unformatted/122_values.sql",
"unformatted/123_spark_keywords.sql",
"unformatted/124_bq_compound_types.sql",
"unformatted/125_numeric_literals.sql",
"unformatted/126_blank_lines.sql",
"unformatted/127_more_comments.sql",
"unformatted/128_double_slash_comments.sql",
"unformatted/129_duckdb_joins.sql",
"unformatted/130_athena_data_types.sql",
"unformatted/131_assignment_statement.sql",
"unformatted/132_spark_number_literals.sql",
"unformatted/133_for_else.sql",
"unformatted/134_databricks_type_hints.sql",
"unformatted/135_star_columns.sql",
"unformatted/136_databricks_variant.sql",
"unformatted/200_base_model.sql",
"unformatted/201_basic_snapshot.sql",
"unformatted/202_unpivot_macro.sql",
"unformatted/203_gitlab_email_domain_type.sql",
"unformatted/204_gitlab_tag_validation.sql",
"unformatted/205_rittman_hubspot_deals.sql",
"unformatted/206_gitlab_prep_geozone.sql",
"unformatted/207_rittman_int_journals.sql",
"unformatted/208_rittman_int_plan_breakout_metrics.sql",
"unformatted/209_rittman_int_web_events_sessionized.sql",
"unformatted/210_gitlab_gdpr_delete.sql",
"unformatted/211_http_2019_cdn_17_20.sql",
"unformatted/212_http_2019_cms_14_02.sql",
"unformatted/213_gitlab_fct_sales_funnel_target.sql",
"unformatted/214_get_unique_attributes.sql",
"unformatted/215_gitlab_get_backup_table_command.sql",
"unformatted/216_gitlab_zuora_revenue_revenue_contract_line_source.sql",
"unformatted/217_dbt_unit_testing_csv.sql",
"unformatted/218_multiple_c_comments.sql",
"unformatted/219_any_all_agg.sql",
"unformatted/220_clickhouse_joins.sql",
"unformatted/300_jinjafmt.sql",
"unformatted/400_create_fn_and_select.sql",
"unformatted/401_explain_select.sql",
"unformatted/402_delete_from_using.sql",
"unformatted/403_grant_revoke.sql",
"unformatted/404_create_function_pg_examples.sql",
"unformatted/405_create_function_snowflake_examples.sql",
"unformatted/406_create_function_bq_examples.sql",
"unformatted/407_alter_function_pg_examples.sql",
"unformatted/408_alter_function_snowflake_examples.sql",
"unformatted/409_create_external_function.sql",
"unformatted/410_create_warehouse.sql",
"unformatted/411_create_clone.sql",
"unformatted/412_pragma.sql",
"unformatted/900_create_view.sql",
"unformatted/999_unsupported_ddl.sql",
],
)
def test_formatting(p: str) -> None:
mode = Mode()
source, expected = read_test_data(p)
actual = format_string(source, mode)
check_formatting(expected, actual, ctx=p)
second_pass = format_string(actual, mode)
check_formatting(expected, second_pass, ctx=f"2nd-{p}")
|
