Search in sources :

Example 26 with TestingLogicalSlotBuilder

use of org.apache.flink.runtime.jobmaster.TestingLogicalSlotBuilder in project flink by apache.

the class RegionToRestartInBatchJobBenchmark method setup.

@Override
public void setup(JobConfiguration jobConfiguration) throws Exception {
    super.setup(jobConfiguration);
    final JobVertex source = jobVertices.get(0);
    final JobVertex sink = jobVertices.get(1);
    final TestingLogicalSlotBuilder slotBuilder = new TestingLogicalSlotBuilder();
    deployTasks(executionGraph, source.getID(), slotBuilder, true);
    transitionTaskStatus(executionGraph, source.getID(), ExecutionState.FINISHED);
    deployTasks(executionGraph, sink.getID(), slotBuilder, true);
}
Also used : JobVertex(org.apache.flink.runtime.jobgraph.JobVertex) TestingLogicalSlotBuilder(org.apache.flink.runtime.jobmaster.TestingLogicalSlotBuilder)

Aggregations

TestingLogicalSlotBuilder (org.apache.flink.runtime.jobmaster.TestingLogicalSlotBuilder)26 Test (org.junit.Test)19 LogicalSlot (org.apache.flink.runtime.jobmaster.LogicalSlot)16 ExecutionGraphTestUtils.getExecutionVertex (org.apache.flink.runtime.executiongraph.ExecutionGraphTestUtils.getExecutionVertex)10 ExecutionVertex (org.apache.flink.runtime.executiongraph.ExecutionVertex)7 TestingLogicalSlot (org.apache.flink.runtime.jobmaster.TestingLogicalSlot)7 JobVertex (org.apache.flink.runtime.jobgraph.JobVertex)6 IOException (java.io.IOException)5 SimpleAckingTaskManagerGateway (org.apache.flink.runtime.executiongraph.utils.SimpleAckingTaskManagerGateway)4 JobGraph (org.apache.flink.runtime.jobgraph.JobGraph)4 JobVertexID (org.apache.flink.runtime.jobgraph.JobVertexID)4 ArrayList (java.util.ArrayList)3 JobID (org.apache.flink.api.common.JobID)3 ExecutionGraph (org.apache.flink.runtime.executiongraph.ExecutionGraph)3 TaskManagerLocation (org.apache.flink.runtime.taskmanager.TaskManagerLocation)3 List (java.util.List)2 ArrayBlockingQueue (java.util.concurrent.ArrayBlockingQueue)2 CompletableFuture (java.util.concurrent.CompletableFuture)2 Configuration (org.apache.flink.configuration.Configuration)2 JobException (org.apache.flink.runtime.JobException)2