Implement PyMap.get with the optional default.

ccutrer · ccutrer · commit db55adc791c2 · 2025-02-05T08:05:45.000-07: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
@@ -21,6 +21,13 @@ protected Map<String, Object> delegate() {
     return map;
   }
 
+  public Object get(String key, Object defaultValue) {
+    if (!map.containsKey(key)) {
+      return defaultValue;
+    }
+    return map.get(key);
+  }
+
   @Override
   public Object put(String s, Object o) {
     if (o == this) {
diff --git a/src/test/java/com/hubspot/jinjava/objects/collections/PyMapTest.java b/src/test/java/com/hubspot/jinjava/objects/collections/PyMapTest.java
@@ -322,6 +322,18 @@ public void itGetsKeysWithVariableName() {
       .isEqualTo("value1");
   }
 
+  @Test
+  public void itSupportsGetWithOptionalDefault() {
+    assertThat(
+      jinjava.render(
+        "{% set test = {\"key1\": \"value1\"} %}" +
+        "{{ test.get(\"key2\", \"default\") }}",
+        Collections.emptyMap()
+      )
+    )
+      .isEqualTo("default");
+  }
+
   @Test
   public void itFallsBackUnknownVariableNameToString() {
     assertThat(
