Added per stream capable state to MysqlDataTypeAccuracy

airbytehq · Sep 21, 2023 · ac7269d · ac7269d
1 parent 9aa88a6
commit ac7269d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/.../airbyte/integrations/io/airbyte/integration_tests/sources/MySqlDatatypeAccuracyTest.java b/.../airbyte/integrations/io/airbyte/integration_tests/sources/MySqlDatatypeAccuracyTest.java
@@ -7,6 +7,7 @@
 import com.fasterxml.jackson.databind.JsonNode;
 import com.google.common.collect.ImmutableMap;
 import com.mysql.cj.MysqlType;
+import io.airbyte.commons.features.EnvVariableFeatureFlags;
 import io.airbyte.commons.json.Jsons;
 import io.airbyte.db.Database;
 import io.airbyte.db.factory.DSLContextFactory;
@@ -21,9 +22,16 @@
 import java.util.Map;
 import java.util.Map.Entry;
 import org.jooq.SQLDialect;
+import org.junit.jupiter.api.extension.ExtendWith;
 import org.testcontainers.containers.MySQLContainer;
+import uk.org.webcompere.systemstubs.environment.EnvironmentVariables;
+import uk.org.webcompere.systemstubs.jupiter.SystemStub;
+import uk.org.webcompere.systemstubs.jupiter.SystemStubsExtension;
 
+@ExtendWith(SystemStubsExtension.class)
 public class MySqlDatatypeAccuracyTest extends AbstractMySqlSourceDatatypeTest {
+  @SystemStub
+  private EnvironmentVariables environmentVariables;
 
   @Override
   protected void tearDown(final TestDestinationEnv testEnv) {
@@ -39,6 +47,7 @@ protected void tearDown(final TestDestinationEnv testEnv) {
 
   @Override
   protected Database setupDatabase() throws Exception {
+    environmentVariables.set(EnvVariableFeatureFlags.USE_STREAM_CAPABLE_STATE, "true");
     container = new MySQLContainer<>("mysql:8.0");
     container.start();
     final JsonNode replicationMethod = Jsons.jsonNode(ImmutableMap.builder()