[18202] Destination-redshift: fixed tmp test tables removal in integr…

…ation tests (#22495)
airbytehq · Feb 7, 2023 · 3d2a995 · 3d2a995
1 parent 6f0c0be
commit 3d2a995
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/...io/airbyte/integrations/destination/redshift/RedshiftInsertDestinationAcceptanceTest.java b/...io/airbyte/integrations/destination/redshift/RedshiftInsertDestinationAcceptanceTest.java
@@ -78,6 +78,12 @@ void setup() {
             .withDestinationSyncMode(DestinationSyncMode.APPEND)));
   }
 
+  @Override
+  protected void tearDown(final TestDestinationEnv testEnv) throws Exception {
+    getDatabase().query(ctx -> ctx.execute(String.format("DROP SCHEMA IF EXISTS %s CASCADE", DATASET_ID)));
+    super.tearDown(testEnv);
+  }
+
   @Test
   void testIfSuperTmpTableWasCreatedAfterVarcharTmpTable() throws Exception {
     setup();

diff --git a/...e/integrations/destination/redshift/RedshiftS3StagingInsertDestinationAcceptanceTest.java b/...e/integrations/destination/redshift/RedshiftS3StagingInsertDestinationAcceptanceTest.java
@@ -74,6 +74,12 @@ void setup() {
             .withDestinationSyncMode(DestinationSyncMode.APPEND)));
   }
 
+  @Override
+  protected void tearDown(final TestDestinationEnv testEnv) throws Exception {
+    getDatabase().query(ctx -> ctx.execute(String.format("DROP SCHEMA IF EXISTS %s CASCADE", DATASET_ID)));
+    super.tearDown(testEnv);
+  }
+
   @Test
   void testIfSuperTmpTableWasCreatedAfterVarcharTmpTableDuringS3Staging() throws Exception {
     setup();

diff --git a/...airbyte/integrations/destination/redshift/RedshiftStagingS3DestinationAcceptanceTest.java b/...airbyte/integrations/destination/redshift/RedshiftStagingS3DestinationAcceptanceTest.java
@@ -215,7 +215,7 @@ protected void setup(final TestDestinationEnv testEnv) throws Exception {
 
   @Override
   protected void tearDown(final TestDestinationEnv testEnv) throws Exception {
-    final String dropSchemaQuery = String.format("DROP SCHEMA IF EXISTS %s CASCADE", config.get("schema").asText());
+    getDatabase().query(ctx -> ctx.execute(String.format("DROP SCHEMA IF EXISTS %s CASCADE", config.get("schema").asText())));
     getDatabase().query(ctx -> ctx.execute(String.format("drop user if exists %s;", USER_WITHOUT_CREDS)));
   }