Merge pull request #343 from ClickHouse/skip-tests-if-not-on-cluster

Skip distributed tests if cluster is not provided
ClickHouse · Aug 21, 2024 · 08ed52f · 08ed52f
2 parents a9042ad + 8a5b96a
commit 08ed52f
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/tests/integration/adapter/incremental/test_schema_change.py b/tests/integration/adapter/incremental/test_schema_change.py
@@ -1,3 +1,4 @@
+import os
 from functools import reduce
 
 import pytest
@@ -44,6 +45,11 @@ def models(self):
 
     @pytest.mark.parametrize("model", ("schema_change_ignore", "schema_change_distributed_ignore"))
     def test_ignore(self, project, model):
+        if (
+            model == "schema_change_distributed_ignore"
+            and os.environ.get('DBT_CH_TEST_CLUSTER', '').strip() == ''
+        ):
+            pytest.skip("Not on a cluster")
         run_dbt(["run", "--select", model])
         result = project.run_sql(f"select * from {model} order by col_1", fetch="all")
         assert len(result) == 3
@@ -54,6 +60,11 @@ def test_ignore(self, project, model):
 
     @pytest.mark.parametrize("model", ("schema_change_fail", "schema_change_distributed_fail"))
     def test_fail(self, project, model):
+        if (
+            model == "schema_change_distributed_fail"
+            and os.environ.get('DBT_CH_TEST_CLUSTER', '').strip() == ''
+        ):
+            pytest.skip("Not on a cluster")
         run_dbt(["run", "--select", model])
         result = project.run_sql(f"select * from {model} order by col_1", fetch="all")
         assert len(result) == 3
@@ -70,6 +81,11 @@ def test_fail(self, project, model):
 
     @pytest.mark.parametrize("model", ("schema_change_append", "schema_change_distributed_append"))
     def test_append(self, project, model):
+        if (
+            model == "schema_change_distributed_append"
+            and os.environ.get('DBT_CH_TEST_CLUSTER', '').strip() == ''
+        ):
+            pytest.skip("Not on a cluster")
         run_dbt(["run", "--select", model])
         result = project.run_sql(f"select * from {model} order by col_1", fetch="all")
         assert len(result) == 3
@@ -131,6 +147,8 @@ def models(self):
         ),
     )
     def test_fail(self, project, model):
+        if "distributed" in model and os.environ.get('DBT_CH_TEST_CLUSTER', '').strip() == '':
+            pytest.skip("Not on a cluster")
         run_dbt(["run", "--select", model])
         result = project.run_sql(f"select * from {model} order by col_1", fetch="all")
         assert len(result) == 3
@@ -149,6 +167,11 @@ def test_fail(self, project, model):
         "model", ("complex_schema_change_sync", "complex_schema_change_distributed_sync")
     )
     def test_sync(self, project, model):
+        if (
+            model == "complex_schema_change_distributed_sync"
+            and os.environ.get('DBT_CH_TEST_CLUSTER', '').strip() == ''
+        ):
+            pytest.skip("Not on a cluster")
         run_dbt(["run", "--select", model])
         result = project.run_sql(f"select * from {model} order by col_1", fetch="all")
         assert len(result) == 3
@@ -196,6 +219,11 @@ def models(self):
 
     @pytest.mark.parametrize("model", ("out_of_order_columns", "out_of_order_columns_distributed"))
     def test_reordering(self, project, model):
+        if (
+            model == "out_of_order_columns_distributed"
+            and os.environ.get('DBT_CH_TEST_CLUSTER', '').strip() == ''
+        ):
+            pytest.skip("Not on a cluster")
         run_dbt(["run", "--select", model])
         result = project.run_sql(f"select * from {model} order by col_1", fetch="all")
         assert result[0][1] == 1