Fix: Take deployability into account when rendering a model's query w…

…ith the 'render' command
TobikoData · Sep 20, 2024 · ac4b828 · ac4b828
1 parent 25f4760
commit ac4b828
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 2 deletions.
diff --git a/sqlmesh/core/context.py b/sqlmesh/core/context.py
@@ -845,12 +845,16 @@ def render(
             )
             return next(pandas_to_sql(t.cast(pd.DataFrame, df), model.columns_to_types))
 
+        snapshots = self.snapshots
+        deployability_index = DeployabilityIndex.create(snapshots.values())
+
         return model.render_query_or_raise(
             start=start,
             end=end,
             execution_time=execution_time,
-            snapshots=self.snapshots,
+            snapshots=snapshots,
             expand=expand,
+            deployability_index=deployability_index,
             engine_adapter=self.engine_adapter,
             **kwargs,
         )

diff --git a/tests/core/test_context.py b/tests/core/test_context.py
@@ -11,7 +11,7 @@
 import pandas as pd
 from pathlib import Path
 from pytest_mock.plugin import MockerFixture
-from sqlglot import parse_one
+from sqlglot import exp, parse_one
 from sqlglot.errors import SchemaError
 
 import sqlmesh.core.constants
@@ -179,6 +179,50 @@ def test_render_sql_model(sushi_context, assert_exp_eq, copy_to_temp_path: t.Cal
     )
 
 
+@pytest.mark.slow
+def test_render_non_deployable_parent(sushi_context, assert_exp_eq, copy_to_temp_path: t.Callable):
+    model = sushi_context.get_model("sushi.waiter_revenue_by_day")
+    forward_only_kind = model.kind.copy(update={"forward_only": True})
+    model = model.copy(update={"kind": forward_only_kind})
+    sushi_context.upsert_model(model)
+    sushi_context.plan("dev", no_prompts=True, auto_apply=True)
+
+    expected_table_name = parse_one(
+        sushi_context.get_snapshot("sushi.waiter_revenue_by_day").table_name(is_deployable=False),
+        into=exp.Table,
+    ).this.this
+
+    assert_exp_eq(
+        sushi_context.render(
+            "sushi.top_waiters",
+            start=date(2021, 1, 1),
+            end=date(2021, 1, 1),
+        ),
+        f"""
+        WITH "test_macros" AS (
+          SELECT
+            2 AS "lit_two",
+            "waiter_revenue_by_day"."revenue" * 2.0 AS "sql_exp",
+            CAST("waiter_revenue_by_day"."revenue" AS TEXT) AS "sql_lit"
+          FROM "memory"."sqlmesh__sushi"."{expected_table_name}" AS "waiter_revenue_by_day" /* memory.sushi.waiter_revenue_by_day */
+        )
+        SELECT
+          CAST("waiter_revenue_by_day"."waiter_id" AS INT) AS "waiter_id",
+          CAST("waiter_revenue_by_day"."revenue" AS DOUBLE) AS "revenue"
+        FROM "memory"."sqlmesh__sushi"."{expected_table_name}" AS "waiter_revenue_by_day" /* memory.sushi.waiter_revenue_by_day */
+        WHERE
+          "waiter_revenue_by_day"."event_date" = (
+            SELECT
+              MAX("waiter_revenue_by_day"."event_date") AS "_col_0"
+            FROM "memory"."sqlmesh__sushi"."{expected_table_name}" AS "waiter_revenue_by_day" /* memory.sushi.waiter_revenue_by_day */
+          )
+        ORDER BY
+          "revenue" DESC
+        LIMIT 10
+        """,
+    )
+
+
 @pytest.mark.slow
 def test_render_seed_model(sushi_context, assert_exp_eq):
     assert_exp_eq(