Merge pull request #1083 from HubSpot/update-lte-test

Update IsLessThanOrEqualToSymbolExpTest

Author: julia-uy

src/main/java/com/hubspot/jinjava/lib/exptest/IsLessThanOrEqualToSymbolExpTest.java

@@ -3,7 +3,7 @@
 import com.hubspot.jinjava.doc.annotations.JinjavaDoc;
 
 @JinjavaDoc(value = "", aliasOf = "le")
-public class IsLessThanOrEqualToSymbolExpTest extends IsLtTest {
+public class IsLessThanOrEqualToSymbolExpTest extends IsLeTest {
 
   @Override
   public String getName() {

src/test/java/com/hubspot/jinjava/lib/filter/SelectAttrFilterTest.java

@@ -3,6 +3,7 @@
 import static org.assertj.core.api.Assertions.assertThat;
 
 import com.fasterxml.jackson.annotation.JsonValue;
+import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.Lists;
 import com.hubspot.jinjava.BaseJinjavaTest;
 import com.hubspot.jinjava.objects.serialization.PyishSerializable;
@@ -25,6 +26,17 @@ public void setup() {
           new User(2, false, null, new Option(2, "option2"))
         )
       );
+    jinjava
+      .getGlobalContext()
+      .put(
+        "numbers",
+        Lists.newArrayList(
+          ImmutableMap.of("number", 1),
+          ImmutableMap.of("number", 2),
+          ImmutableMap.of("number", 3),
+          ImmutableMap.of("number", 4)
+        )
+      );
   }
 
   @Test
@@ -101,6 +113,50 @@ public void selectAttrWithNestedProperty() {
       .isEqualTo("[2]");
   }
 
+  @Test
+  public void selectAttrWithSymbolicLtExp() {
+    assertThat(
+        jinjava.render(
+          "{{ numbers|selectattr('number', '<', '3')|map('number') }}",
+          new HashMap<String, Object>()
+        )
+      )
+      .isEqualTo("[1, 2]");
+  }
+
+  @Test
+  public void selectAttrWithSymbolicLeExp() {
+    assertThat(
+        jinjava.render(
+          "{{ numbers|selectattr('number', '<=', '3')|map('number') }}",
+          new HashMap<String, Object>()
+        )
+      )
+      .isEqualTo("[1, 2, 3]");
+  }
+
+  @Test
+  public void selectAttrWithSymbolicGtExp() {
+    assertThat(
+        jinjava.render(
+          "{{ numbers|selectattr('number', '>', '3')|map('number') }}",
+          new HashMap<String, Object>()
+        )
+      )
+      .isEqualTo("[4]");
+  }
+
+  @Test
+  public void selectAttrWithSymbolicGeExp() {
+    assertThat(
+        jinjava.render(
+          "{{ numbers|selectattr('number', '>=', '3')|map('number') }}",
+          new HashMap<String, Object>()
+        )
+      )
+      .isEqualTo("[3, 4]");
+  }
+
   public static class User implements PyishSerializable {
     private long num;
     private boolean isActive;