Skip to content

Commit d6d3955

Browse files
committed
Fix: Snowflake integration tests
1 parent eb17e24 commit d6d3955

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

tests/core/engine_adapter/integration/test_integration.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import pytz
1616
from sqlglot import exp, parse_one
1717
from sqlglot.optimizer.normalize_identifiers import normalize_identifiers
18+
from sqlglot.optimizer.qualify_columns import quote_identifiers
1819

1920
from sqlmesh import Config, Context
2021
from sqlmesh.cli.project_init import init_example_project
@@ -1914,11 +1915,17 @@ def test_sushi(ctx: TestContext, tmp_path_factory: pytest.TempPathFactory):
19141915
],
19151916
personal_paths=[pathlib.Path("~/.sqlmesh/config.yaml").expanduser()],
19161917
)
1917-
config.before_all = [
1918+
before_all = [
19181919
f"CREATE SCHEMA IF NOT EXISTS {raw_test_schema}",
19191920
f"DROP VIEW IF EXISTS {raw_test_schema}.demographics",
19201921
f"CREATE VIEW {raw_test_schema}.demographics AS (SELECT 1 AS customer_id, '00000' AS zip)",
19211922
]
1923+
config.before_all = [
1924+
quote_identifiers(parse_one(e, dialect=ctx.dialect), dialect=ctx.dialect).sql(
1925+
dialect=ctx.dialect
1926+
)
1927+
for e in before_all
1928+
]
19221929

19231930
# To enable parallelism in integration tests
19241931
config.gateways = {ctx.gateway: config.gateways[ctx.gateway]}
@@ -2847,16 +2854,12 @@ def _run_plan(sqlmesh_context: Context, environment: str = None) -> PlanResults:
28472854
assert plan_1.snapshot_for(model_a).model.view_name in plan_1.schema_metadata.views
28482855
assert plan_1.snapshot_for(model_b).model.view_name in plan_1.schema_metadata.views
28492856

2850-
assert len(plan_1.internal_schema_metadata.tables) == 3
2857+
assert len(plan_1.internal_schema_metadata.tables) == 1
28512858

28522859
assert plan_1.table_name_for(model_a) in plan_1.internal_schema_metadata.tables
2853-
assert plan_1.dev_table_name_for(model_a) in plan_1.internal_schema_metadata.tables
28542860
assert (
28552861
plan_1.table_name_for(model_b) not in plan_1.internal_schema_metadata.tables
28562862
) # because its a managed table
2857-
assert (
2858-
plan_1.dev_table_name_for(model_b) in plan_1.internal_schema_metadata.tables
2859-
) # its dev table is a normal table however
28602863

28612864
assert len(plan_1.internal_schema_metadata.managed_tables) == 1
28622865
assert plan_1.table_name_for(model_b) in plan_1.internal_schema_metadata.managed_tables

0 commit comments

Comments
 (0)