[db2] [tests] Fix unstable Db2ConnectorITCase due to cleanup failure

This closes apache#2345.
GOODBOY008 · Aug 3, 2023 · 61b9a2a · 61b9a2a
1 parent 1330cc4
commit 61b9a2a
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/...ctor-db2-cdc/src/test/java/com/ververica/cdc/connectors/db2/table/Db2ConnectorITCase.java b/...ctor-db2-cdc/src/test/java/com/ververica/cdc/connectors/db2/table/Db2ConnectorITCase.java
@@ -62,6 +62,15 @@ public void before() {
         env.setParallelism(1);
     }
 
+    private void cancelJobIfRunning(TableResult result)
+            throws InterruptedException, ExecutionException {
+        try {
+            result.getJobClient().get().cancel().get();
+        } catch (IllegalStateException ignored) {
+            // job isn't running, ignore it
+        }
+    }
+
     @Test
     public void testConsumingAllEvents()
             throws SQLException, InterruptedException, ExecutionException {
@@ -163,7 +172,7 @@ public void testConsumingAllEvents()
         List<String> actual = TestValuesTableFactory.getResults("sink");
         assertThat(actual, containsInAnyOrder(expected));
 
-        result.getJobClient().get().cancel().get();
+        cancelJobIfRunning(result);
     }
 
     @Test
@@ -254,7 +263,7 @@ public void testAllTypes() throws Exception {
         List<String> actual = TestValuesTableFactory.getRawResults("sink");
         assertEquals(expected, actual);
 
-        result.getJobClient().get().cancel().get();
+        cancelJobIfRunning(result);
     }
 
     @Test
@@ -322,7 +331,7 @@ public void testStartupFromLatestOffset() throws Exception {
         List<String> actual = TestValuesTableFactory.getResults("sink");
         assertThat(actual, containsInAnyOrder(expected));
 
-        result.getJobClient().get().cancel().get();
+        cancelJobIfRunning(result);
     }
 
     @Test
@@ -419,7 +428,7 @@ public void testMetadataColumns() throws Throwable {
         Collections.sort(expected);
         Collections.sort(actual);
         assertEquals(expected, actual);
-        result.getJobClient().get().cancel().get();
+        cancelJobIfRunning(result);
     }
 
     private static void waitForSnapshotStarted(String sinkName) throws InterruptedException {