fix(core): proper exception for invalid RunContext uri

kestra-io · Sep 15, 2023 · b9b24e6 · b9b24e6
1 parent d5f9b8d
commit b9b24e6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/core/src/main/java/io/kestra/core/runners/RunContext.java b/core/src/main/java/io/kestra/core/runners/RunContext.java
@@ -452,11 +452,19 @@ public org.slf4j.Logger logger() {
     }
 
     public InputStream uriToInputStream(URI uri) throws IOException {
+        if (uri == null) {
+            throw new IllegalArgumentException("Invalid internal storage uri, got null");
+        }
+
+        if (uri.getScheme() == null) {
+            throw new IllegalArgumentException("Invalid internal storage uri, got uri '" + uri + "'");
+        }
+
         if (uri.getScheme().equals("kestra")) {
             return this.storageInterface.get(uri);
         }
 
-        throw new IllegalArgumentException("Invalid scheme for uri '" + uri + "'");
+        throw new IllegalArgumentException("Invalid internal storage scheme, got uri '" + uri + "'");
     }
 
     /**