chore: update runContext deprecated methods

kestra-io · May 13, 2024 · 6fe82c7 · 6fe82c7
1 parent 1aa96ac
commit 6fe82c7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/test/java/io/kestra/plugin/aws/eventbridge/PutEventsTest.java b/src/test/java/io/kestra/plugin/aws/eventbridge/PutEventsTest.java
@@ -45,7 +45,7 @@ private static List<PutEvents.OutputEntry> getOutputEntries(PutEvents put, RunCo
         if (!from.getScheme().equals("kestra")) {
             throw new IllegalArgumentException("Invalid entries parameter, must be a Kestra internal storage URI, or a list of entry.");
         }
-        try (BufferedReader inputStream = new BufferedReader(new InputStreamReader(runContext.uriToInputStream(from)))) {
+        try (BufferedReader inputStream = new BufferedReader(new InputStreamReader(runContext.storage().getFile(from)))) {
             outputEntries = Flux.create(FileSerde.reader(inputStream, PutEvents.OutputEntry.class), FluxSink.OverflowStrategy.BUFFER).collectList().block();
         }
         return outputEntries;
@@ -151,7 +151,7 @@ void runStorage() throws Exception {
             .region(localstack.getRegion())
             .accessKeyId(localstack.getAccessKey())
             .secretKeyId(localstack.getSecretKey())
-            .entries(runContext.putTempFile(tempFile).toString())
+            .entries(runContext.storage().putFile(tempFile).toString())
             .build();
 
 

diff --git a/src/test/java/io/kestra/plugin/aws/kinesis/PutRecordsTest.java b/src/test/java/io/kestra/plugin/aws/kinesis/PutRecordsTest.java
@@ -84,7 +84,7 @@ private static List<PutRecords.OutputEntry> getOutputEntries(PutRecords put, Run
         if (!from.getScheme().equals("kestra")) {
             throw new IllegalArgumentException("Invalid entries parameter, must be a Kestra internal storage URI, or a list of entry.");
         }
-        try (BufferedReader inputStream = new BufferedReader(new InputStreamReader(runContext.uriToInputStream(from)))) {
+        try (BufferedReader inputStream = new BufferedReader(new InputStreamReader(runContext.storage().getFile(from)))) {
             outputEntries = Flux.create(FileSerde.reader(inputStream, PutRecords.OutputEntry.class), FluxSink.OverflowStrategy.BUFFER).collectList().block();
         }
         return outputEntries;
@@ -177,7 +177,7 @@ void runStorage() throws Exception {
             .region(localstack.getRegion())
             .accessKeyId(localstack.getAccessKey())
             .secretKeyId(localstack.getSecretKey())
-            .records(runContext.putTempFile(tempFile).toString())
+            .records(runContext.storage().putFile(tempFile).toString())
             .streamName("streamName")
             .build();
 
@@ -230,7 +230,7 @@ void runStorageUpperCase() throws Exception {
             .region(localstack.getRegion())
             .accessKeyId(localstack.getAccessKey())
             .secretKeyId(localstack.getSecretKey())
-            .records(runContext.putTempFile(tempFile).toString())
+            .records(runContext.storage().putFile(tempFile).toString())
             .streamName("streamName")
             .build();