[hotfix][tests] Fix SessionDispatcherLeaderProcessTest.onRemovedJobGr…

…aph_terminatesRunningJob Properly wait on the future completion for the assertion.
lightClouds917 · Jan 31, 2022 · d874b26 · d874b26
1 parent 0290715
commit d874b26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...t/java/org/apache/flink/runtime/dispatcher/runner/SessionDispatcherLeaderProcessTest.java b/...t/java/org/apache/flink/runtime/dispatcher/runner/SessionDispatcherLeaderProcessTest.java
@@ -433,7 +433,7 @@ public void onRemovedJobGraph_terminatesRunningJob() throws Exception {
             jobGraphStore.removeJobGraph(JOB_GRAPH.getJobID());
             dispatcherLeaderProcess.onRemovedJobGraph(JOB_GRAPH.getJobID());
 
-            assertThat(terminateJobFuture).isCompletedWithValue(JOB_GRAPH.getJobID());
+            assertThat(terminateJobFuture.get()).isEqualTo(JOB_GRAPH.getJobID());
         }
     }