[ci] Sanitize each test use case (Db2ConnectorIT)

debezium · Feb 10, 2024 · ef7ee32 · ef7ee32
1 parent 287fb43
commit ef7ee32
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/test/java/io/debezium/connector/db2/Db2ConnectorIT.java b/src/test/java/io/debezium/connector/db2/Db2ConnectorIT.java
@@ -58,6 +58,8 @@ public class Db2ConnectorIT extends AbstractConnectorTest {
 
     @Before
     public void before() throws SQLException {
+        TestHelper.dropAllTables();
+
         connection = TestHelper.testConnection();
         connection.execute("DELETE FROM ASNCDC.IBMSNAP_REGISTER");
         connection.execute(

diff --git a/src/test/java/io/debezium/connector/db2/util/TestHelper.java b/src/test/java/io/debezium/connector/db2/util/TestHelper.java
@@ -272,4 +272,19 @@ public static void waitForCDC() {
 
         }
     }
+
+    public static void dropAllTables() throws SQLException {
+        try (Db2Connection connection = testConnection()) {
+            LOGGER.info("Attempting to drop all tables (if exists)");
+            connection.query("SELECT TABNAME FROM syscat.tables WHERE TABSCHEMA = 'DB2INST1'", rs -> {
+                while (rs.next()) {
+                    final String tableName = rs.getString(1);
+                    LOGGER.info("Disabling CDC for table {}", tableName);
+                    disableTableCdc(connection, "DB2INST1", tableName);
+                    LOGGER.warn("Dropping table {}", tableName);
+                    connection.execute("DROP TABLE IF EXISTS " + tableName);
+                }
+            });
+        }
+    }
 }