Test fix

nbauernfeind · Mar 5, 2024 · 2a7b16e · 2a7b16e
1 parent a03abe4
commit 2a7b16e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/engine/context/src/main/java/io/deephaven/engine/context/QueryCompiler.java b/engine/context/src/main/java/io/deephaven/engine/context/QueryCompiler.java
@@ -220,8 +220,14 @@ public Class<?> compile(@NotNull final QueryCompilerRequest request) {
         compile(request, resolver);
         try {
             return resolver.getFuture().get();
-        } catch (InterruptedException | ExecutionException e) {
-            throw new UncheckedDeephavenException("Could not compile class", e);
+        } catch (ExecutionException e) {
+            Throwable cause = e.getCause();
+            if (cause instanceof RuntimeException) {
+                throw (RuntimeException) cause;
+            }
+            throw new UncheckedDeephavenException("Error while compiling class", cause);
+        } catch (InterruptedException e) {
+            throw new UncheckedDeephavenException("Interrupted while compile class", e);
         }
     }