feat: multi-tenant

kestra-io · Oct 13, 2023 · 125f541 · 125f541
1 parent 8a8420a
commit 125f541
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/test/java/io/kestra/plugin/nats/NatsTest.java b/src/test/java/io/kestra/plugin/nats/NatsTest.java
@@ -22,7 +22,7 @@ class NatsTest {
     protected StorageInterface storageInterface;
 
     protected List<Map<String, Object>> toMessages(Consume.Output output) throws IOException {
-        BufferedReader inputStream = new BufferedReader(new InputStreamReader(storageInterface.get(output.getUri())));
+        BufferedReader inputStream = new BufferedReader(new InputStreamReader(storageInterface.get(null, output.getUri())));
         List<Map<String, Object>> result = new ArrayList<>();
         FileSerde.reader(inputStream, r -> result.add((Map<String, Object>) r));
         return result;

diff --git a/src/test/java/io/kestra/plugin/nats/ProduceTest.java b/src/test/java/io/kestra/plugin/nats/ProduceTest.java
@@ -141,7 +141,7 @@ void produceMultipleMessagesFromInternalStorage() throws Exception {
         try (OutputStream outputStream = new FileOutputStream(tempFile)) {
             messages.forEach(throwConsumer(message -> FileSerde.write(outputStream, message)));
         }
-        URI uri = storageInterface.put(URI.create("/" + IdUtils.create() + ".ion"), new FileInputStream(tempFile));
+        URI uri = storageInterface.put(null, URI.create("/" + IdUtils.create() + ".ion"), new FileInputStream(tempFile));
 
         Produce.Output produceOutput = Produce.builder()
             .url("localhost:4222")

diff --git a/src/test/java/io/kestra/plugin/nats/TriggerTest.java b/src/test/java/io/kestra/plugin/nats/TriggerTest.java
@@ -64,7 +64,7 @@ void simpleConsumeTrigger() throws Exception {
         Execution execution = triggerFlow();
 
 
-        BufferedReader inputStream = new BufferedReader(new InputStreamReader(storageInterface.get(URI.create((String) execution.getTrigger().getVariables().get("uri")))));
+        BufferedReader inputStream = new BufferedReader(new InputStreamReader(storageInterface.get(null, URI.create((String) execution.getTrigger().getVariables().get("uri")))));
         List<Map<String, Object>> result = new ArrayList<>();
         FileSerde.reader(inputStream, r -> result.add((Map<String, Object>) r));