check that saas mode is indeed set

duckdb · Sep 26, 2024 · 273b308 · 273b308
1 parent 86aeece
commit 273b308
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/functional/plugins/motherduck/test_motherduck_saas_mode.py b/tests/functional/plugins/motherduck/test_motherduck_saas_mode.py
@@ -5,6 +5,8 @@
 )
 from dbt.artifacts.schemas.results import RunStatus
 
+from dbt.adapters.duckdb.environments.motherduck import MOTHERDUCK_SAAS_MODE_QUERY
+
 random_logs_sql = """
 {{ config(materialized='table', meta=dict(temp_schema_name='dbt_temp_test')) }}
 
@@ -98,6 +100,8 @@ def run_dbt_scope(self, project, database_name):
         project.run_sql(f"DROP TABLE {database_name}.plugin_table")
 
     def test_motherduck(self, project):
+        (motherduck_saas_mode,) = project.run_sql(MOTHERDUCK_SAAS_MODE_QUERY, fetch="one")
+        assert motherduck_saas_mode == "1"
         result = run_dbt(expect_pass=False)
         expected_msg = "Python models are disabled when MotherDuck SaaS Mode is on."
         assert [_res for _res in result.results if _res.status != RunStatus.Success][0].message == expected_msg