Support updated/new microbatch tests provided by dbt-tests-adapter

dbt-labs · Oct 22, 2024 · bb38f3b · bb38f3b
1 parent 2ee4fd0
commit bb38f3b
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/functional/adapter/incremental/test_incremental_microbatch.py b/tests/functional/adapter/incremental/test_incremental_microbatch.py
@@ -4,7 +4,8 @@
 
 from dbt.tests.util import run_dbt_and_capture
 from dbt.tests.adapter.incremental.test_incremental_microbatch import (
-    BaseMicrobatch,
+    BaseTestMicrobatchOn,
+    BaseTestMicrobatchOff,
     patch_microbatch_end_time,
 )
 
@@ -16,7 +17,13 @@
 )
 
 
-class TestBigQueryMicrobatch(BaseMicrobatch):
+class TestBigQueryMicrobatchOn(BaseTestMicrobatchOn):
+    @pytest.fixture(scope="class")
+    def microbatch_model_sql(self) -> str:
+        return microbatch_model_no_unique_id_sql
+
+
+class TestBigQueryMicrobatchOff(BaseTestMicrobatchOff):
     @pytest.fixture(scope="class")
     def microbatch_model_sql(self) -> str:
         return microbatch_model_no_unique_id_sql