tests(): attempt to fix failing tests (#256)

Author: Malaydewangan09

src/main/java/io/kestra/plugin/fs/vfs/Uploads.java

@@ -5,14 +5,18 @@
 import io.kestra.core.models.property.Property;
 import io.kestra.core.models.tasks.RunnableTask;
 import io.kestra.core.runners.RunContext;
+import io.kestra.core.serializers.JacksonMapper;
 import io.swagger.v3.oas.annotations.media.Schema;
 import jakarta.validation.constraints.NotNull;
 import lombok.*;
 import lombok.experimental.SuperBuilder;
 import org.apache.commons.vfs2.impl.StandardFileSystemManager;
 
 import java.net.URI;
+import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
+import java.util.Objects;
 
 import static io.kestra.core.utils.Rethrow.throwFunction;
 
@@ -21,12 +25,13 @@
 @EqualsAndHashCode
 @Getter
 @NoArgsConstructor
-public abstract class Uploads extends AbstractVfsTask implements RunnableTask<Uploads.Output> {
+public abstract class Uploads extends AbstractVfsTask implements RunnableTask<Uploads.Output>, Data.From {
     @Schema(
         title = "The files to upload, must be internal storage URIs, must be a list of URIs or a pebble template that returns a list of URIs",
         anyOf = {
             String.class,
-            String[].class
+            List.class,
+            Map.class
         },
         description = "Must be Kestra internal storage URIs. Can be a single URI string, a list of URI strings, or an internal storage URI pointing to a file containing URIs."
     )
@@ -45,33 +50,43 @@ public Output run(RunContext runContext) throws Exception {
             fsm.setConfiguration(StandardFileSystemManager.class.getResource(KestraStandardFileSystemManager.CONFIG_RESOURCE));
             fsm.init();
 
-            String renderedTo = runContext.render(this.to).as(String.class).orElseThrow();
+            String[] renderedFrom = parseFromProperty(runContext);
 
-            List<Upload.Output> outputs = Data.from(from)
-                .readAs(runContext, String.class, obj -> obj.toString())
-                .map(throwFunction(fromURI -> {
-                    if (!fromURI.startsWith("kestra://")) {
-                        throw new IllegalArgumentException("'from' must be a list of Kestra's internal storage URI");
-                    }
-
-                    return VfsService.upload(
-                        runContext,
-                        fsm,
-                        this.fsOptions(runContext),
-                        URI.create(fromURI),
-                        this.uri(runContext, renderedTo + fromURI.substring(fromURI.lastIndexOf('/') + (renderedTo.endsWith("/") ? 1 : 0)))
-                    );
-                }))
-                .collectList()
-                .block();
+            List<Upload.Output> outputs = Arrays.stream(renderedFrom).map(throwFunction(fromURI -> {
+                var rTo = runContext.render(this.to).as(String.class).orElseThrow();
+                return VfsService.upload(
+                    runContext,
+                    fsm,
+                    this.fsOptions(runContext),
+                    URI.create(fromURI),
+                    this.uri(runContext, rTo + fromURI.substring(fromURI.lastIndexOf('/') + (rTo.endsWith("/") ? 1 : 0)))
+                );
+            })).toList();
 
             return Output.builder()
-                    .files(outputs.stream()
-                            .map(Upload.Output::getTo)
-                            .toList()
-                    )
-                    .build();
+                .files(outputs.stream()
+                    .map(Upload.Output::getTo)
+                    .toList()
+                )
+                .build();
+        }
+    }
+
+    private String[] parseFromProperty(RunContext runContext) throws Exception {
+        if (this.from instanceof String from) {
+            var rFrom = runContext.render(from).trim();
+
+            if (rFrom.startsWith("[") && rFrom.endsWith("]")) {
+                return JacksonMapper.ofJson().readValue(rFrom, String[].class);
+            }
         }
+
+        return Objects.requireNonNull(Data.from(this.from)
+                .readAs(runContext, String.class, Object::toString)
+                .map(throwFunction(runContext::render))
+                .collectList()
+                .block())
+            .toArray(String[]::new);
     }
 
     @Builder