Unwrap OutputTooBigException from ELException when handling

jasmith-hs · jasmith-hs · commit 37a53e41395e · 2023-06-23T19:17:03.000-04:00
diff --git a/src/main/java/com/hubspot/jinjava/el/ExpressionResolver.java b/src/main/java/com/hubspot/jinjava/el/ExpressionResolver.java
@@ -13,6 +13,7 @@
 import com.hubspot.jinjava.interpret.InvalidArgumentException;
 import com.hubspot.jinjava.interpret.InvalidInputException;
 import com.hubspot.jinjava.interpret.JinjavaInterpreter;
+import com.hubspot.jinjava.interpret.OutputTooBigException;
 import com.hubspot.jinjava.interpret.TemplateError;
 import com.hubspot.jinjava.interpret.TemplateError.ErrorItem;
 import com.hubspot.jinjava.interpret.TemplateError.ErrorReason;
@@ -244,6 +245,19 @@ private void handleELException(String expression, ELException e) {
           e
         )
       );
+    } else if (e.getCause() != null && e.getCause() instanceof OutputTooBigException) {
+      interpreter.addError(
+        new TemplateError(
+          ErrorType.FATAL,
+          ErrorReason.OUTPUT_TOO_BIG,
+          ErrorItem.FUNCTION,
+          e.getCause().getMessage(),
+          null,
+          interpreter.getLineNumber(),
+          interpreter.getPosition(),
+          e
+        )
+      );
     } else {
       String originatingException = getRootCauseMessage(e);
       final String combinedMessage = String.format(
