diff --git a/src/main/java/io/kestra/plugin/cloudquery/CloudQueryCLI.java b/src/main/java/io/kestra/plugin/cloudquery/CloudQueryCLI.java index fcfac21..d23d401 100644 --- a/src/main/java/io/kestra/plugin/cloudquery/CloudQueryCLI.java +++ b/src/main/java/io/kestra/plugin/cloudquery/CloudQueryCLI.java @@ -106,6 +106,7 @@ public ScriptOutput run(RunContext runContext) throws Exception { .withBeforeCommands(Property.ofValue(List.of("alias cloudquery='/app/cloudquery'"))) .withCommands(this.commands) .withEnv(runContext.render(this.getEnv()).asMap(String.class, String.class).isEmpty() ? new HashMap<>() : runContext.render(this.getEnv()).asMap(String.class, String.class)) + .withNamespaceFiles(namespaceFiles) .withInputFiles(inputFiles) .withOutputFiles(renderedOutputFiles.isEmpty() ? null : renderedOutputFiles); diff --git a/src/test/java/io/kestra/plugin/cloudquery/CloudQueryCLITest.java b/src/test/java/io/kestra/plugin/cloudquery/CloudQueryCLITest.java index 8fb8ebf..a2f6a84 100644 --- a/src/test/java/io/kestra/plugin/cloudquery/CloudQueryCLITest.java +++ b/src/test/java/io/kestra/plugin/cloudquery/CloudQueryCLITest.java @@ -1,5 +1,7 @@ package io.kestra.plugin.cloudquery; +import java.io.ByteArrayInputStream; +import java.nio.file.Path; import java.util.List; import java.util.Map; @@ -8,6 +10,7 @@ import io.kestra.core.junit.annotations.KestraTest; import io.kestra.core.models.property.Property; +import io.kestra.core.models.tasks.NamespaceFiles; import io.kestra.core.runners.RunContext; import io.kestra.core.runners.RunContextFactory; import io.kestra.core.utils.IdUtils; @@ -55,4 +58,29 @@ void run() throws Exception { assertThat(runOutput.getVars().get("customEnv"), is(envValue)); } -} \ No newline at end of file + + @Test + void namespaceFiles() throws Exception { + CloudQueryCLI execute = CloudQueryCLI.builder() + .id(IdUtils.create()) + .type(CloudQueryCLI.class.getName()) + .namespaceFiles( + NamespaceFiles.builder() + .enabled(Property.ofValue(true)) + .build() + ) + .commands(Property.ofValue(List.of("test -f config.yml"))) + .build(); + + RunContext runContext = TestsUtils.mockRunContext(runContextFactory, execute, Map.of()); + + runContext.storage().namespace().putFile( + Path.of("config.yml"), + new ByteArrayInputStream("kind: source".getBytes()) + ); + + ScriptOutput runOutput = execute.run(runContext); + + assertThat(runOutput.getExitCode(), is(0)); + } +}