Package org.apache.flink.runtime.protocols

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

39
40
41
42
43
44
45
46
47
48
49
  @Test
  public void testRestartManually() {
    final int NUM_TASKS = 31;
   
    try {
      TaskOperationProtocol tm = getSimpleAcknowledgingTaskmanager();
      Instance instance = getInstance(tm);
     
      Scheduler scheduler = new Scheduler();
      scheduler.newInstanceAvailable(instance);
     
View Full Code Here
84
85
86
87
88
89
90
91
92
93
94
  @Test
  public void testRestartSelf() {
    final int NUM_TASKS = 31;
   
    try {
      TaskOperationProtocol tm = getSimpleAcknowledgingTaskmanager();
      Instance instance = getInstance(tm);
     
      Scheduler scheduler = new Scheduler();
      scheduler.newInstanceAvailable(instance);
     
View Full Code Here
37
38
39
40
41
42
43
44
45
46
47
48
public class TaskManagerLossFailsTasksTest {

  @Test
  public void testTasksFailWhenTaskManagerLost() {
    try {
      TaskOperationProtocol tm1 = getSimpleAcknowledgingTaskmanager();
      TaskOperationProtocol tm2 = getSimpleAcknowledgingTaskmanager();
     
      Instance instance1 = getInstance(tm1, 10);
      Instance instance2 = getInstance(tm2, 10);
     
      Scheduler scheduler = new Scheduler();
View Full Code Here
43
44
45
46
47
48
49
50
51
52
53
  @Test
  public void testSlotReleasedWhenScheduledImmediately() {
   
    try {
      // a slot than cannot be deployed to
      final TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      slot.cancel();
      assertFalse(slot.isReleased());
     
View Full Code Here
76
77
78
79
80
81
82
83
84
85
86
  @Test
  public void testSlotReleasedWhenScheduledQueued() {

    try {
      // a slot than cannot be deployed to
      final TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      slot.cancel();
      assertFalse(slot.isReleased());
     
View Full Code Here
115
116
117
118
119
120
121
122
123
124
125
 
  @Test
  public void testScheduleToDeploy() {
    try {
      // a slot than cannot be deployed to
      final TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
     
      final ExecutionJobVertex ejv = getJobVertexNotExecuting(new JobVertexID());
      final ExecutionVertex vertex = new ExecutionVertex(ejv, 0, new IntermediateResult[0]);
View Full Code Here
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
      }
    };
  }
 
  public static TaskOperationProtocol getSimpleAcknowledgingTaskmanager() throws Exception {
    TaskOperationProtocol top = mock(TaskOperationProtocol.class);
   
    when(top.submitTask(any(TaskDeploymentDescriptor.class))).thenAnswer(new Answer<TaskOperationResult>() {
      @Override
      public TaskOperationResult answer(InvocationOnMock invocation) {
        final TaskDeploymentDescriptor tdd = (TaskDeploymentDescriptor) invocation.getArguments()[0];
        return new TaskOperationResult(tdd.getExecutionId(), true);
      }
    });
   
    when(top.cancelTask(Matchers.any(ExecutionAttemptID.class))).thenAnswer(new Answer<TaskOperationResult>() {
      @Override
      public TaskOperationResult answer(InvocationOnMock invocation) {
        final ExecutionAttemptID id = (ExecutionAttemptID) invocation.getArguments()[0];
        return new TaskOperationResult(id, true);
      }
View Full Code Here
TOP

Related Classes of org.apache.flink.runtime.protocols.TaskOperationProtocol

  • org.apache.flink.runtime.executiongraph.ExecutionGraphDeploymentTest
  • org.apache.flink.runtime.executiongraph.ExecutionGraphRestartTest
  • org.apache.flink.runtime.executiongraph.ExecutionGraphTestUtils
  • org.apache.flink.runtime.executiongraph.ExecutionStateProgressTest
  • org.apache.flink.runtime.executiongraph.ExecutionVertexCancelTest
  • org.apache.flink.runtime.executiongraph.ExecutionVertexDeploymentTest
  • org.apache.flink.runtime.executiongraph.ExecutionVertexSchedulingTest
  • org.apache.flink.runtime.executiongraph.TaskManagerLossFailsTasksTest
  • org.apache.flink.runtime.instance.Instance
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.