[hotfix][runtime] Throw exception from TestingSchedulingPipelinedRegi…

…on#getVertex() for unknown vertices
bbejeck · Apr 24, 2020 · fd13c2c · fd13c2c
1 parent 8514200
commit fd13c2c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...st/java/org/apache/flink/runtime/scheduler/strategy/TestingSchedulingPipelinedRegion.java b/...st/java/org/apache/flink/runtime/scheduler/strategy/TestingSchedulingPipelinedRegion.java
@@ -52,7 +52,11 @@ public Iterable<TestingSchedulingExecutionVertex> getVertices() {
 
  @Override
  public TestingSchedulingExecutionVertex getVertex(ExecutionVertexID vertexId) {
- return regionVertices.get(vertexId);
+ final TestingSchedulingExecutionVertex executionVertex = regionVertices.get(vertexId);
+ if (executionVertex == null) {
+ throw new IllegalArgumentException(String.format("Execution vertex %s not found in pipelined region", vertexId));
+ }
+ return executionVertex;
  }
 
  @Override