[hotfix] Remove mocking from ExecutionGraphRestartTest

laubersder · Oct 11, 2018 · 0b0c261 · 0b0c261
1 parent 8ebedc6
commit 0b0c261
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/...time/src/test/java/org/apache/flink/runtime/executiongraph/ExecutionGraphRestartTest.java b/...time/src/test/java/org/apache/flink/runtime/executiongraph/ExecutionGraphRestartTest.java
@@ -90,7 +90,6 @@
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
-import static org.mockito.Mockito.spy;
 
 /**
  * Tests the restart behaviour of the {@link ExecutionGraph}.
@@ -851,14 +850,6 @@ public void run() {
  }
 
  private static Tuple2<ExecutionGraph, Instance> createExecutionGraph(RestartStrategy restartStrategy) throws Exception {
- return createExecutionGraph(restartStrategy, false);
- }
-
- private static Tuple2<ExecutionGraph, Instance> createSpyExecutionGraph(RestartStrategy restartStrategy) throws Exception {
- return createExecutionGraph(restartStrategy, true);
- }
-
- private static Tuple2<ExecutionGraph, Instance> createExecutionGraph(RestartStrategy restartStrategy, boolean isSpy) throws Exception {
  Instance instance = ExecutionGraphTestUtils.getInstance(
  new ActorTaskManagerGateway(
  new SimpleActorGateway(TestingUtils.directExecutionContext())),
@@ -872,9 +863,6 @@ private static Tuple2<ExecutionGraph, Instance> createExecutionGraph(RestartStra
  JobGraph jobGraph = new JobGraph("Pointwise job", sender);
 
  ExecutionGraph eg = newExecutionGraph(restartStrategy, scheduler);
- if (isSpy) {
- eg = spy(eg);
- }
  eg.attachJobGraph(jobGraph.getVerticesSortedTopologicallyFromSources());
 
  assertEquals(JobStatus.CREATED, eg.getState());