Java ExecutionGraph示例

示例#1

文件： FixedDelayRestartStrategyTest.java 项目： kl0u/flink

  @Test
  public void testFixedDelayRestartStrategy() {
    int numberRestarts = 10;
    long restartDelay = 10;

    FixedDelayRestartStrategy fixedDelayRestartStrategy =
        new FixedDelayRestartStrategy(numberRestarts, restartDelay);

    ExecutionGraph executionGraph = mock(ExecutionGraph.class);
    when(executionGraph.getFutureExecutor())
        .thenReturn(ExecutionContext$.MODULE$.fromExecutor(MoreExecutors.directExecutor()));

    while (fixedDelayRestartStrategy.canRestart()) {
      fixedDelayRestartStrategy.restart(executionGraph);
    }

    Mockito.verify(executionGraph, Mockito.times(numberRestarts)).restart();
  }