Pass table mapping to _demote_snapshot render args

Author: treysp

sqlmesh/core/plan/evaluator.py

@@ -389,6 +389,12 @@ def _demote_snapshots(
             target_snapshots,
             environment_naming_info,
             snapshots=snapshots,
+            table_mapping=to_view_mapping(
+                snapshots.values(),
+                environment_naming_info,
+                default_catalog=self.default_catalog,
+                dialect=self.snapshot_evaluator.adapter.dialect,
+            ),
             deployability_index=deployability_index,
             on_complete=on_complete,
         )

sqlmesh/core/snapshot/evaluator.py

@@ -278,6 +278,7 @@ def demote(
         target_snapshots: t.Iterable[Snapshot],
         environment_naming_info: EnvironmentNamingInfo,
         snapshots: t.Optional[t.Dict[SnapshotId, Snapshot]] = None,
+        table_mapping: t.Optional[t.Dict[str, str]] = None,
         deployability_index: t.Optional[DeployabilityIndex] = None,
         on_complete: t.Optional[t.Callable[[SnapshotInfoLike], None]] = None,
     ) -> None:
@@ -292,7 +293,12 @@ def demote(
             concurrent_apply_to_snapshots(
                 target_snapshots,
                 lambda s: self._demote_snapshot(
-                    s, environment_naming_info, snapshots, deployability_index, on_complete
+                    s,
+                    environment_naming_info,
+                    snapshots,
+                    table_mapping,
+                    deployability_index,
+                    on_complete,
                 ),
                 self.ddl_concurrent_tasks,
             )
@@ -1018,6 +1024,7 @@ def _demote_snapshot(
         snapshot: Snapshot,
         environment_naming_info: EnvironmentNamingInfo,
         snapshots: t.Optional[t.Dict[SnapshotId, Snapshot]],
+        table_mapping: t.Optional[t.Dict[str, str]],
         deployability_index: t.Optional[DeployabilityIndex],
         on_complete: t.Optional[t.Callable[[SnapshotInfoLike], None]],
     ) -> None:
@@ -1035,8 +1042,8 @@ def _demote_snapshot(
                 adapter.session(
                     snapshot.model.render_session_properties(
                         engine_adapter=adapter,
-                        snapshots=snapshots,
                         deployability_index=deployability_index,
+                        table_mapping=table_mapping,
                         runtime_stage=RuntimeStage.DEMOTING,
                     )
                 ),