Skip to content

Commit 3c98196

Browse files
committed
Fix: Allow other tables in sqlmesh state schema (#2005)
1 parent 152f70b commit 3c98196

File tree

1 file changed

+40
-32
lines changed

1 file changed

+40
-32
lines changed

tests/core/test_integration.py

Lines changed: 40 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -1605,14 +1605,16 @@ def get_default_catalog_and_non_tables(
16051605
assert len(dev_views) == 0
16061606
assert len(user_default_tables) == 23
16071607
assert state_metadata.schemas == ["sqlmesh"]
1608-
assert [x.sql() for x in state_metadata.qualified_tables] == [
1609-
"physical.sqlmesh._environments",
1610-
"physical.sqlmesh._intervals",
1611-
"physical.sqlmesh._plan_dags",
1612-
"physical.sqlmesh._seeds",
1613-
"physical.sqlmesh._snapshots",
1614-
"physical.sqlmesh._versions",
1615-
]
1608+
assert {x.sql() for x in state_metadata.qualified_tables}.issuperset(
1609+
{
1610+
"physical.sqlmesh._environments",
1611+
"physical.sqlmesh._intervals",
1612+
"physical.sqlmesh._plan_dags",
1613+
"physical.sqlmesh._seeds",
1614+
"physical.sqlmesh._snapshots",
1615+
"physical.sqlmesh._versions",
1616+
}
1617+
)
16161618
apply_to_environment(context, "dev")
16171619
prod_views, dev_views = get_prod_dev_views(metadata)
16181620
(
@@ -1624,14 +1626,16 @@ def get_default_catalog_and_non_tables(
16241626
assert len(user_default_tables) == 23
16251627
assert len(non_default_tables) == 0
16261628
assert state_metadata.schemas == ["sqlmesh"]
1627-
assert [x.sql() for x in state_metadata.qualified_tables] == [
1628-
"physical.sqlmesh._environments",
1629-
"physical.sqlmesh._intervals",
1630-
"physical.sqlmesh._plan_dags",
1631-
"physical.sqlmesh._seeds",
1632-
"physical.sqlmesh._snapshots",
1633-
"physical.sqlmesh._versions",
1634-
]
1629+
assert {x.sql() for x in state_metadata.qualified_tables}.issuperset(
1630+
{
1631+
"physical.sqlmesh._environments",
1632+
"physical.sqlmesh._intervals",
1633+
"physical.sqlmesh._plan_dags",
1634+
"physical.sqlmesh._seeds",
1635+
"physical.sqlmesh._snapshots",
1636+
"physical.sqlmesh._versions",
1637+
}
1638+
)
16351639
apply_to_environment(context, "prodnot")
16361640
prod_views, dev_views = get_prod_dev_views(metadata)
16371641
(
@@ -1643,14 +1647,16 @@ def get_default_catalog_and_non_tables(
16431647
assert len(user_default_tables) == 23
16441648
assert len(non_default_tables) == 0
16451649
assert state_metadata.schemas == ["sqlmesh"]
1646-
assert [x.sql() for x in state_metadata.qualified_tables] == [
1647-
"physical.sqlmesh._environments",
1648-
"physical.sqlmesh._intervals",
1649-
"physical.sqlmesh._plan_dags",
1650-
"physical.sqlmesh._seeds",
1651-
"physical.sqlmesh._snapshots",
1652-
"physical.sqlmesh._versions",
1653-
]
1650+
assert {x.sql() for x in state_metadata.qualified_tables}.issuperset(
1651+
{
1652+
"physical.sqlmesh._environments",
1653+
"physical.sqlmesh._intervals",
1654+
"physical.sqlmesh._plan_dags",
1655+
"physical.sqlmesh._seeds",
1656+
"physical.sqlmesh._snapshots",
1657+
"physical.sqlmesh._versions",
1658+
}
1659+
)
16541660
context.invalidate_environment("dev")
16551661
context._run_janitor()
16561662
prod_views, dev_views = get_prod_dev_views(metadata)
@@ -1663,14 +1669,16 @@ def get_default_catalog_and_non_tables(
16631669
assert len(user_default_tables) == 23
16641670
assert len(non_default_tables) == 0
16651671
assert state_metadata.schemas == ["sqlmesh"]
1666-
assert [x.sql() for x in state_metadata.qualified_tables] == [
1667-
"physical.sqlmesh._environments",
1668-
"physical.sqlmesh._intervals",
1669-
"physical.sqlmesh._plan_dags",
1670-
"physical.sqlmesh._seeds",
1671-
"physical.sqlmesh._snapshots",
1672-
"physical.sqlmesh._versions",
1673-
]
1672+
assert {x.sql() for x in state_metadata.qualified_tables}.issuperset(
1673+
{
1674+
"physical.sqlmesh._environments",
1675+
"physical.sqlmesh._intervals",
1676+
"physical.sqlmesh._plan_dags",
1677+
"physical.sqlmesh._seeds",
1678+
"physical.sqlmesh._snapshots",
1679+
"physical.sqlmesh._versions",
1680+
}
1681+
)
16741682

16751683

16761684
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)