add additional tests for choice deferred functionality

Nathaniel Pautzke · Nathaniel Pautzke · commit 55efe036cbe2 · 2022-07-29T12:54:50.000-05:00
diff --git a/src/test/java/com/hubspot/jinjava/el/ext/eager/EagerAstChoiceTest.java b/src/test/java/com/hubspot/jinjava/el/ext/eager/EagerAstChoiceTest.java
@@ -1,5 +1,6 @@
 package com.hubspot.jinjava.el.ext.eager;
 
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.fail;
 
 import com.hubspot.jinjava.BaseInterpretingTest;
@@ -90,4 +91,28 @@ public void itDoesNotShortCircuitsChoiceNo() {
       Assertions.assertThat(e.getDeferredEvalResult()).isEqualTo("deferred");
     }
   }
+
+  @Test
+  public void itResolvesChoiceYes() {
+    interpreter.getContext().put("bar", "bar val");
+    interpreter.resolveELExpression(
+      "foo_list[0] == 'val' ? foo_list.add(bar) : deferred",
+      -1
+    );
+    PyList result = (PyList) interpreter.getContext().get("foo_list");
+    assertEquals(result.size(), 2);
+    assertEquals(result.get(1), "bar val");
+  }
+
+  @Test
+  public void itResolvesChoiceNo() {
+    interpreter.getContext().put("bar", "bar val");
+    interpreter.resolveELExpression(
+      "foo_list[0] == 'bar' ? deferred : foo_list.add(bar)",
+      -1
+    );
+    PyList result = (PyList) interpreter.getContext().get("foo_list");
+    assertEquals(result.size(), 2);
+    assertEquals(result.get(1), "bar val");
+  }
 }
