Enable batch tests

MeltanoLabs · Mar 20, 2024 · c74cd05 · c74cd05
1 parent 2d604c1
commit c74cd05
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/tests/test_target_snowflake.py b/tests/test_target_snowflake.py
@@ -12,6 +12,7 @@
 
 from target_snowflake.target import TargetSnowflake
 
+from .batch import batch_target_tests
 from .core import target_tests
 
 SAMPLE_CONFIG: dict[str, Any] = {
@@ -70,16 +71,16 @@ class TestTargetSnowflake(BaseSnowflakeTargetTests, StandardTargetTests):  # typ
 
 
 # # Custom so I can implement all validate methods
-# BATCH_TEST_CONFIG = copy.deepcopy(SAMPLE_CONFIG)
-# BATCH_TEST_CONFIG["default_target_schema"] = f"TARGET_SNOWFLAKE_{uuid.uuid4().hex[0:6]!s}"
-# BATCH_TEST_CONFIG["add_record_metadata"] = False
-# BatchTargetTests = get_target_test_class(
-#     target_class=TargetSnowflake,
-#     config=BATCH_TEST_CONFIG,
-#     custom_suites=[batch_target_tests],
-#     suite_config=None,
-#     include_target_tests=False,
-# )
+BATCH_TEST_CONFIG = copy.deepcopy(SAMPLE_CONFIG)
+BATCH_TEST_CONFIG["default_target_schema"] = f"TARGET_SNOWFLAKE_{uuid.uuid4().hex[0:6]!s}"
+BATCH_TEST_CONFIG["add_record_metadata"] = False
+BatchTargetTests = get_target_test_class(
+    target_class=TargetSnowflake,
+    config=BATCH_TEST_CONFIG,
+    custom_suites=[batch_target_tests],
+    suite_config=None,
+    include_target_tests=False,
+)
 
 
 # class TestTargetSnowflakeBatch(BaseSnowflakeTargetTests, BatchTargetTests):  # type: ignore[misc, valid-type]