Fix flaky execution vertex test, which was susceptible to RPC races

apache · Jan 5, 2015 · ad250bc · ad250bc
1 parent 35291ea
commit ad250bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...time/src/test/java/org/apache/flink/runtime/executiongraph/ExecutionVertexCancelTest.java b/...time/src/test/java/org/apache/flink/runtime/executiongraph/ExecutionVertexCancelTest.java
@@ -479,7 +479,7 @@ public void testSendCancelAndReceiveFail() {
 
 					vertex.cancel();
 
-					assertEquals(ExecutionState.CANCELING, vertex.getExecutionState());
+					assertTrue(vertex.getExecutionState() == ExecutionState.CANCELING || vertex.getExecutionState() == ExecutionState.FAILED);
 
 					vertex.getCurrentExecutionAttempt().markFailed(new Throwable("test"));