From 0b0a6bc60b04e9b838db0e63ae17d0d4885888ae Mon Sep 17 00:00:00 2001 From: godfrey he Date: Thu, 11 Jun 2020 20:23:27 +0800 Subject: [PATCH] [FLINK-18247][table-planner-blink] Fix unstable test: TableITCase.testCollectWithClose This closes #12595 --- .../org/apache/flink/table/api/TableITCase.scala | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/flink-table/flink-table-planner-blink/src/test/scala/org/apache/flink/table/api/TableITCase.scala b/flink-table/flink-table-planner-blink/src/test/scala/org/apache/flink/table/api/TableITCase.scala index 354fd5fbc6422..d4b5eb4040b95 100644 --- a/flink-table/flink-table-planner-blink/src/test/scala/org/apache/flink/table/api/TableITCase.scala +++ b/flink-table/flink-table-planner-blink/src/test/scala/org/apache/flink/table/api/TableITCase.scala @@ -28,7 +28,7 @@ import org.apache.flink.util.TestLogger import org.apache.flink.shaded.guava18.com.google.common.collect.Lists -import org.junit.Assert.{assertEquals, assertTrue} +import org.junit.Assert.{assertEquals, assertNotEquals, assertTrue} import org.junit.rules.{ExpectedException, TemporaryFolder} import org.junit.runner.RunWith import org.junit.runners.Parameterized @@ -119,7 +119,16 @@ class TableITCase(tableEnvName: String, isStreaming: Boolean) extends TestLogger assertEquals(ResultKind.SUCCESS_WITH_CONTENT, tableResult.getResultKind) val it = tableResult.collect() it.close() - assertEquals(JobStatus.CANCELED, tableResult.getJobClient.get().getJobStatus().get()) + val jobStatus = try { + Some(tableResult.getJobClient.get().getJobStatus.get()) + } catch { + // ignore the exception, + // because the MiniCluster maybe already been shut down when getting job status + case _: Throwable => None + } + if (jobStatus.isDefined) { + assertNotEquals(JobStatus.RUNNING, jobStatus.get) + } } @Test