Examples with ExecutionAttemptID org.apache.flink.runtime.executiongraph.ExecutionAttemptID used on opensource projects

Example 46 with ExecutionAttemptID

use of org.apache.flink.runtime.executiongraph.ExecutionAttemptID in project flink by apache.

the class StackTraceSampleCoordinatorTest method testTriggerStackTraceSampleNotRunningTasks.

/** Tests triggering for non-running tasks fails the future. */
@Test
public void testTriggerStackTraceSampleNotRunningTasks() throws Exception {
    ExecutionVertex[] vertices = new ExecutionVertex[] { mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.RUNNING, true), mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.DEPLOYING, true) };
    Future<StackTraceSample> sampleFuture = coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0);
    assertTrue(sampleFuture.isDone());
    try {
        sampleFuture.get();
        fail("Expected exception.");
    } catch (ExecutionException e) {
        assertTrue(e.getCause() instanceof IllegalStateException);
    }
}

Example 47 with ExecutionAttemptID

use of org.apache.flink.runtime.executiongraph.ExecutionAttemptID in project flink by apache.

the class StackTraceSampleCoordinatorTest method testCollectStackTraceForCanceledSample.

/** Tests that collecting for a cancelled sample throws no Exception. */
@Test
public void testCollectStackTraceForCanceledSample() throws Exception {
    ExecutionVertex[] vertices = new ExecutionVertex[] { mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.RUNNING, true) };
    Future<StackTraceSample> sampleFuture = coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0);
    assertFalse(sampleFuture.isDone());
    coord.cancelStackTraceSample(0, null);
    assertTrue(sampleFuture.isDone());
    // Verify no error on late collect
    ExecutionAttemptID executionId = vertices[0].getCurrentExecutionAttempt().getAttemptId();
    coord.collectStackTraces(0, executionId, new ArrayList<StackTraceElement[]>());
}

Example 48 with ExecutionAttemptID

use of org.apache.flink.runtime.executiongraph.ExecutionAttemptID in project flink by apache.

the class StackTraceSampleCoordinatorTest method testTriggerStackTraceSampleResetRunningTasks.

/** Tests triggering for reset tasks fails the future. */
@Test(timeout = 1000L)
public void testTriggerStackTraceSampleResetRunningTasks() throws Exception {
    ExecutionVertex[] vertices = new ExecutionVertex[] { mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.RUNNING, true), // Fails to send the message to the execution (happens when execution is reset)
    mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.RUNNING, false) };
    Future<StackTraceSample> sampleFuture = coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0);
    try {
        sampleFuture.get();
        fail("Expected exception.");
    } catch (ExecutionException e) {
        assertTrue(e.getCause() instanceof RuntimeException);
    }
}

Example 49 with ExecutionAttemptID

use of org.apache.flink.runtime.executiongraph.ExecutionAttemptID in project flink by apache.

the class StackTraceSampleCoordinatorTest method testShutDown.

/** Tests that shut down fails all pending samples and future sample triggers. */
@Test
public void testShutDown() throws Exception {
    ExecutionVertex[] vertices = new ExecutionVertex[] { mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.RUNNING, true) };
    List<Future<StackTraceSample>> sampleFutures = new ArrayList<>();
    // Trigger
    sampleFutures.add(coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0));
    sampleFutures.add(coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0));
    for (Future<StackTraceSample> future : sampleFutures) {
        assertFalse(future.isDone());
    }
    // Shut down
    coord.shutDown();
    // Verify all completed
    for (Future<StackTraceSample> future : sampleFutures) {
        assertTrue(future.isDone());
    }
    // Verify new trigger returns failed future
    Future<StackTraceSample> future = coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0);
    assertTrue(future.isDone());
    try {
        future.get();
        fail("Expected exception.");
    } catch (ExecutionException e) {
    // we expected an exception here :-)
    }
}

Example 50 with ExecutionAttemptID

use of org.apache.flink.runtime.executiongraph.ExecutionAttemptID in project flink by apache.

the class StackTraceSampleCoordinatorTest method testCollectForDiscardedPendingSample.

/** Tests that collecting for a cancelled sample throws no Exception. */
@Test
public void testCollectForDiscardedPendingSample() throws Exception {
    ExecutionVertex[] vertices = new ExecutionVertex[] { mockExecutionVertex(new ExecutionAttemptID(), ExecutionState.RUNNING, true) };
    Future<StackTraceSample> sampleFuture = coord.triggerStackTraceSample(vertices, 1, Time.milliseconds(100L), 0);
    assertFalse(sampleFuture.isDone());
    coord.cancelStackTraceSample(0, null);
    assertTrue(sampleFuture.isDone());
    // Verify no error on late collect
    ExecutionAttemptID executionId = vertices[0].getCurrentExecutionAttempt().getAttemptId();
    coord.collectStackTraces(0, executionId, new ArrayList<StackTraceElement[]>());
}