use of io.kestra.core.runners.RunContext in project kestra by kestra-io.
the class ConditionServiceTest method valid.
@Test
void valid() {
Flow flow = TestsUtils.mockFlow();
Execution execution = TestsUtils.mockExecution(flow, ImmutableMap.of());
RunContext runContext = runContextFactory.of(flow, execution);
ConditionContext conditionContext = conditionService.conditionContext(runContext, flow, execution);
List<Condition> conditions = Arrays.asList(ExecutionFlowCondition.builder().namespace(flow.getNamespace()).flowId(flow.getId()).build(), ExecutionNamespaceCondition.builder().namespace(flow.getNamespace()).build());
boolean valid = conditionService.valid(flow, conditions, conditionContext);
assertThat(valid, is(true));
}
use of io.kestra.core.runners.RunContext in project kestra by kestra-io.
the class ConditionServiceTest method exception.
@Test
void exception() throws InterruptedException {
List<LogEntry> logs = new ArrayList<>();
logQueue.receive(logs::add);
Flow flow = TestsUtils.mockFlow();
Schedule schedule = Schedule.builder().id("unit").type(Schedule.class.getName()).cron("0 0 1 * *").build();
RunContext runContext = runContextFactory.of(flow, schedule);
ConditionContext conditionContext = conditionService.conditionContext(runContext, flow, null);
List<Condition> conditions = Collections.singletonList(ExecutionFlowCondition.builder().namespace(flow.getNamespace()).flowId(flow.getId()).build());
conditionService.valid(flow, conditions, conditionContext);
Thread.sleep(250);
assertThat(logs.stream().filter(logEntry -> logEntry.getNamespace().equals("io.kestra.core.services.ConditionServiceTest")).count(), greaterThan(0L));
assertThat(logs.stream().filter(logEntry -> logEntry.getFlowId().equals("exception")).count(), greaterThan(0L));
}
use of io.kestra.core.runners.RunContext in project kestra by kestra-io.
the class AbstractBashTest method preventRelativeFile.
@Test
void preventRelativeFile() throws Exception {
URL resource = AbstractBashTest.class.getClassLoader().getResource("application.yml");
URI put = storageInterface.put(new URI("/file/storage/get.yml"), new FileInputStream(Objects.requireNonNull(resource).getFile()));
assertThrows(IllegalArgumentException.class, () -> {
Bash bash = configure(Bash.builder().commands(new String[] { "echo 1" }).inputFiles(Map.of("{{ inputs.vars }}", put.toString()))).build();
RunContext runContext = TestsUtils.mockRunContext(runContextFactory, bash, ImmutableMap.of("vars", "../../test.txt"));
bash.run(runContext);
});
assertThrows(IllegalArgumentException.class, () -> {
Bash bash = configure(Bash.builder().commands(new String[] { "echo 1" }).inputFiles(Map.of("{{ inputs.vars }}", put.toString()))).build();
RunContext runContext = TestsUtils.mockRunContext(runContextFactory, bash, ImmutableMap.of("vars", "../../test.txt"));
bash.run(runContext);
});
// we allow dot file starting with a .
Bash bash = configure(Bash.builder().commands(new String[] { "echo 1" }).inputFiles(Map.of("{{ inputs.vars }}", put.toString()))).build();
RunContext runContext = TestsUtils.mockRunContext(runContextFactory, bash, ImmutableMap.of("vars", ".test.txt"));
ScriptOutput run = bash.run(runContext);
assertThat(run.getExitCode(), is(0));
}
use of io.kestra.core.runners.RunContext in project kestra by kestra-io.
the class AbstractBashTest method stopOnFirstFailed.
@Test
@DisabledIfEnvironmentVariable(named = "GITHUB_WORKFLOW", matches = ".*")
void stopOnFirstFailed() {
Bash bash = configure(Bash.builder().commands(new String[] { "unknown", "echo 1" })).build();
RunContext runContext = TestsUtils.mockRunContext(runContextFactory, bash, ImmutableMap.of());
Bash.BashException bashException = assertThrows(Bash.BashException.class, () -> {
bash.run(runContext);
});
assertThat(bashException.getExitCode(), is(127));
assertThat(bashException.getStdOutSize(), is(0));
assertThat(bashException.getStdErrSize(), is(1));
}
use of io.kestra.core.runners.RunContext in project kestra by kestra-io.
the class AbstractBashTest method files.
@Test
void files() throws Exception {
Bash bash = configure(Bash.builder().outputFiles(Arrays.asList("xml", "csv")).inputFiles(ImmutableMap.of("files/in/in.txt", "I'm here")).commands(new String[] { "echo '::{\"outputs\": {\"extract\":\"'$(cat files/in/in.txt)'\"}}::'", "echo 1 >> {{ outputFiles.xml }}", "echo 2 >> {{ outputFiles.csv }}", "echo 3 >> {{ outputFiles.xml }}" })).build();
RunContext runContext = TestsUtils.mockRunContext(runContextFactory, bash, ImmutableMap.of());
ScriptOutput run = bash.run(runContext);
assertThat(run.getExitCode(), is(0));
assertThat(run.getStdErrLineCount(), is(0));
assertThat(run.getStdOutLineCount(), is(1));
assertThat(run.getVars().get("extract"), is("I'm here"));
InputStream get = storageInterface.get(run.getOutputFiles().get("xml"));
assertThat(CharStreams.toString(new InputStreamReader(get)), is("1\n3\n"));
get = storageInterface.get(run.getOutputFiles().get("csv"));
assertThat(CharStreams.toString(new InputStreamReader(get)), is("2\n"));
}
Aggregations