Fix additional MySQL integration tests (airbytehq#20119)

* Fix test : attempt 1 * Remove setting cloud
siddhant3030 · Dec 6, 2022 · 9ae92e2 · 9ae92e2
1 parent e93f8aa
commit 9ae92e2
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/...te/integrations/io/airbyte/integration_tests/sources/SshKeyMySqlSourceAcceptanceTest.java b/...te/integrations/io/airbyte/integration_tests/sources/SshKeyMySqlSourceAcceptanceTest.java
@@ -4,10 +4,26 @@
 
 package io.airbyte.integrations.io.airbyte.integration_tests.sources;
 
+import io.airbyte.commons.features.EnvVariableFeatureFlags;
+import io.airbyte.integrations.standardtest.source.TestDestinationEnv;
 import java.nio.file.Path;
+import org.junit.jupiter.api.extension.ExtendWith;
+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 SshKeyMySqlSourceAcceptanceTest extends AbstractSshMySqlSourceAcceptanceTest {
 
+  @SystemStub
+  private EnvironmentVariables environmentVariables;
+
+  @Override
+  protected void setupEnvironment(final TestDestinationEnv environment) throws Exception {
+    environmentVariables.set(EnvVariableFeatureFlags.USE_STREAM_CAPABLE_STATE, "true");
+    super.setupEnvironment(environment);
+  }
+
   @Override
   public Path getConfigFilePath() {
     return Path.of("secrets/ssh-key-repl-config.json");

diff --git a/...tegrations/io/airbyte/integration_tests/sources/SshPasswordMySqlSourceAcceptanceTest.java b/...tegrations/io/airbyte/integration_tests/sources/SshPasswordMySqlSourceAcceptanceTest.java
@@ -4,10 +4,26 @@
 
 package io.airbyte.integrations.io.airbyte.integration_tests.sources;
 
+import io.airbyte.commons.features.EnvVariableFeatureFlags;
+import io.airbyte.integrations.standardtest.source.TestDestinationEnv;
 import java.nio.file.Path;
+import org.junit.jupiter.api.extension.ExtendWith;
+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 SshPasswordMySqlSourceAcceptanceTest extends AbstractSshMySqlSourceAcceptanceTest {
 
+  @SystemStub
+  private EnvironmentVariables environmentVariables;
+
+  @Override
+  protected void setupEnvironment(final TestDestinationEnv environment) throws Exception {
+    environmentVariables.set(EnvVariableFeatureFlags.USE_STREAM_CAPABLE_STATE, "true");
+    super.setupEnvironment(environment);
+  }
+
   @Override
   public Path getConfigFilePath() {
     return Path.of("secrets/ssh-pwd-repl-config.json");