Search in sources :

Example 1 with Return

use of io.kestra.core.tasks.debugs.Return in project kestra by kestra-io.

the class ElasticSearchExecutionRepositoryTest method builder.

public static Execution.ExecutionBuilder builder(State.Type state, String flowId) {
    State finalState = randomDuration(state);
    Execution.ExecutionBuilder execution = Execution.builder().id(FriendlyId.createFriendlyId()).namespace(NAMESPACE).flowId(flowId == null ? FLOW : flowId).flowRevision(1).state(finalState);
    List<TaskRun> taskRuns = Arrays.asList(TaskRun.of(execution.build(), ResolvedTask.of(Return.builder().id("first").type(Return.class.getName()).format("test").build())).withState(State.Type.SUCCESS), spyTaskRun(TaskRun.of(execution.build(), ResolvedTask.of(Return.builder().id("second").type(Return.class.getName()).format("test").build())).withState(state), state), TaskRun.of(execution.build(), ResolvedTask.of(Return.builder().id("third").type(Return.class.getName()).format("test").build())).withState(state));
    if (flowId == null) {
        return execution.taskRunList(List.of(taskRuns.get(0), taskRuns.get(1), taskRuns.get(2)));
    }
    return execution.taskRunList(List.of(taskRuns.get(0), taskRuns.get(1)));
}
Also used : TaskRun(io.kestra.core.models.executions.TaskRun) Execution(io.kestra.core.models.executions.Execution) Return(io.kestra.core.tasks.debugs.Return) Mockito.doReturn(org.mockito.Mockito.doReturn) State(io.kestra.core.models.flows.State)

Example 2 with Return

use of io.kestra.core.tasks.debugs.Return in project kestra by kestra-io.

the class TemplateControllerTest method createTemplate.

private Template createTemplate() {
    Task t1 = Return.builder().id("task-1").type(Return.class.getName()).format("test").build();
    Task t2 = Return.builder().id("task-2").type(Return.class.getName()).format("test").build();
    return Template.builder().id(IdUtils.create()).namespace("kestra.test").tasks(Arrays.asList(t1, t2)).build();
}
Also used : Task(io.kestra.core.models.tasks.Task) Return(io.kestra.core.tasks.debugs.Return)

Aggregations

Return (io.kestra.core.tasks.debugs.Return)2 Execution (io.kestra.core.models.executions.Execution)1 TaskRun (io.kestra.core.models.executions.TaskRun)1 State (io.kestra.core.models.flows.State)1 Task (io.kestra.core.models.tasks.Task)1 Mockito.doReturn (org.mockito.Mockito.doReturn)1