Skip to content

Commit 2f51da5

Browse files
committed
Chore: Fix engine adapter tests
1 parent da2e403 commit 2f51da5

File tree

2 files changed

+14
-8
lines changed

2 files changed

+14
-8
lines changed

sqlmesh/core/snapshot/evaluator.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1357,6 +1357,7 @@ def create(
13571357
is_snapshot_deployable: bool,
13581358
**render_kwargs: t.Any,
13591359
) -> None:
1360+
super().create(snapshot, name, is_table_deployable, is_snapshot_deployable, **render_kwargs)
13601361
if is_table_deployable:
13611362
# For seeds we insert data at the time of table creation.
13621363
model = t.cast(SeedModel, snapshot.model)
@@ -1371,10 +1372,6 @@ def create(
13711372
except Exception:
13721373
self.adapter.drop_table(name)
13731374
raise
1374-
else:
1375-
super().create(
1376-
snapshot, name, is_table_deployable, is_snapshot_deployable, **render_kwargs
1377-
)
13781375

13791376
def insert(
13801377
self,

tests/core/test_snapshot_evaluator.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1714,10 +1714,19 @@ def test_create_seed(mocker: MockerFixture, adapter_mock, make_snapshot):
17141714
**common_create_kwargs,
17151715
)
17161716

1717-
adapter_mock.create_table.assert_called_once_with(
1718-
f"sqlmesh__db.db__seed__{snapshot.version}__temp",
1719-
column_descriptions=None,
1720-
**common_create_kwargs,
1717+
adapter_mock.create_table.assert_has_calls(
1718+
[
1719+
call(
1720+
f"sqlmesh__db.db__seed__{snapshot.version}__temp",
1721+
column_descriptions=None,
1722+
**common_create_kwargs,
1723+
),
1724+
call(
1725+
f"sqlmesh__db.db__seed__{snapshot.version}",
1726+
column_descriptions={},
1727+
**common_create_kwargs,
1728+
),
1729+
]
17211730
)
17221731

17231732
replace_query_calls = adapter_mock.replace_query.call_args_list

0 commit comments

Comments
 (0)