use of org.apache.flink.runtime.executiongraph.ErrorInfo in project flink by apache.
the class StopWithSavepointTest method testFailingOnUpdateTaskExecutionStateWithNoRestart.
@Test
public void testFailingOnUpdateTaskExecutionStateWithNoRestart() throws Exception {
try (MockStopWithSavepointContext ctx = new MockStopWithSavepointContext()) {
StateTrackingMockExecutionGraph executionGraph = new StateTrackingMockExecutionGraph();
StopWithSavepoint sws = createStopWithSavepoint(ctx, executionGraph);
ctx.setStopWithSavepoint(sws);
ctx.setHowToHandleFailure(FailureResult::canNotRestart);
ctx.setExpectFailing(failingArguments -> {
assertThat(failingArguments.getFailureCause(), containsCause(RuntimeException.class));
});
Exception exception = new RuntimeException();
TestingAccessExecution execution = TestingAccessExecution.newBuilder().withExecutionState(ExecutionState.FAILED).withErrorInfo(new ErrorInfo(exception, System.currentTimeMillis())).build();
executionGraph.registerExecution(execution);
TaskExecutionStateTransition taskExecutionStateTransition = ExecutingTest.createFailingStateTransition(execution.getAttemptId(), exception);
assertThat(sws.updateTaskExecutionState(taskExecutionStateTransition), is(true));
}
}
use of org.apache.flink.runtime.executiongraph.ErrorInfo in project flink by apache.
the class ExecutingTest method testFailureReportedViaUpdateTaskExecutionStateCausesFailingOnNoRestart.
@Test
public void testFailureReportedViaUpdateTaskExecutionStateCausesFailingOnNoRestart() throws Exception {
try (MockExecutingContext ctx = new MockExecutingContext()) {
StateTrackingMockExecutionGraph returnsFailedStateExecutionGraph = new StateTrackingMockExecutionGraph();
Executing exec = new ExecutingStateBuilder().setExecutionGraph(returnsFailedStateExecutionGraph).build(ctx);
ctx.setHowToHandleFailure(FailureResult::canNotRestart);
ctx.setExpectFailing(assertNonNull());
Exception exception = new RuntimeException();
TestingAccessExecution execution = TestingAccessExecution.newBuilder().withExecutionState(ExecutionState.FAILED).withErrorInfo(new ErrorInfo(exception, System.currentTimeMillis())).build();
returnsFailedStateExecutionGraph.registerExecution(execution);
TaskExecutionStateTransition taskExecutionStateTransition = createFailingStateTransition(execution.getAttemptId(), exception);
exec.updateTaskExecutionState(taskExecutionStateTransition);
}
}
Aggregations