add keys function to dictionary

chrisirhc · chrisirhc · commit e1bac10fc258 · 2022-10-26T16:43:02.000+08:00
diff --git a/src/main/java/com/hubspot/jinjava/objects/collections/PyMap.java b/src/main/java/com/hubspot/jinjava/objects/collections/PyMap.java
@@ -38,6 +38,10 @@ public Set<java.util.Map.Entry<String, Object>> items() {
     return entrySet();
   }
 
+  public Set<String> keys() {
+    return keySet();
+  }
+
   public void update(Map<? extends String, ? extends Object> m) {
     if (m == this) {
       throw new IllegalArgumentException("Can't update map object with itself");
diff --git a/src/test/java/com/hubspot/jinjava/el/ext/AstDictTest.java b/src/test/java/com/hubspot/jinjava/el/ext/AstDictTest.java
@@ -44,6 +44,12 @@ public void itDoesItemsMethodCall() {
       .isInstanceOf(Set.class);
   }
 
+  @Test
+  public void itDoesKeysMethodCall() {
+    interpreter.getContext().put("foo", ImmutableMap.of(TestEnum.BAR, "test"));
+    assertThat(interpreter.resolveELExpression("foo.keys()", -1)).isInstanceOf(Set.class);
+  }
+
   @Test
   public void itHandlesEmptyMaps() {
     interpreter.getContext().put("foo", ImmutableMap.of());
