Changes DefaultFilter to implement AdvancedFilter

src/main/java/com/hubspot/jinjava/lib/filter/DefaultFilter.java

@@ -15,13 +15,17 @@
  **********************************************************************/
 package com.hubspot.jinjava.lib.filter;
 
+import java.util.Map;
+import java.util.Objects;
+
 import org.apache.commons.lang3.BooleanUtils;
 
 import com.hubspot.jinjava.doc.annotations.JinjavaDoc;
 import com.hubspot.jinjava.doc.annotations.JinjavaParam;
 import com.hubspot.jinjava.doc.annotations.JinjavaSnippet;
 import com.hubspot.jinjava.interpret.JinjavaInterpreter;
 import com.hubspot.jinjava.interpret.TemplateSyntaxException;
+import com.hubspot.jinjava.objects.PyWrapper;
 import com.hubspot.jinjava.util.ObjectTruthValue;
 
 @JinjavaDoc(
@@ -39,18 +43,18 @@
             desc = "If you want to use default with variables that evaluate to false you have to set the second parameter to true",
             code = "{{ ''|default('the string was empty', true) }}")
     })
-public class DefaultFilter implements Filter {
+public class DefaultFilter implements AdvancedFilter {
 
   @Override
-  public Object filter(Object object, JinjavaInterpreter interpreter, String... args) {
+  public Object filter(Object object, JinjavaInterpreter interpreter, Object[] args, Map<String, Object> kwargs) {
     boolean truthy = false;
 
     if (args.length < 1) {
       throw new TemplateSyntaxException(interpreter, getName(), "requires either 1 (default value to use) or 2 (default value to use, default with variables that evaluate to false) arguments");
     }
 
     if (args.length > 1) {
-      truthy = BooleanUtils.toBoolean(args[1]);
+      truthy = BooleanUtils.toBoolean(Objects.toString(args[1]));
     }
 
     if (truthy) {
@@ -61,7 +65,7 @@ public Object filter(Object object, JinjavaInterpreter interpreter, String... ar
       return object;
     }
 
-    return args[0];
+    return args[0] instanceof PyWrapper ? args[0] : Objects.toString(args[0]);
   }
 
   @Override

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

@@ -0,0 +1,42 @@
+package com.hubspot.jinjava.lib.filter;
+
+import java.util.HashMap;
+
+import com.hubspot.jinjava.Jinjava;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+/**
+ * Created by manishdevgan on 25/06/19.
+ */
+
+public class DefaultFilterTest {
+  Jinjava jinjava;
+
+  @Before
+  public void setup() {
+    jinjava = new Jinjava();
+    jinjava.getGlobalContext().registerClasses(DefaultFilter.class);
+  }
+
+  @Test
+  public void itSetsDefaultStringValues() {
+    assertThat(jinjava.render("{% set d=d | default(\"some random value\") %}{{ d }}", new HashMap<>())).isEqualTo("some random value");    System.out.println(jinjava.render("{% set d=d | default(\"some random value\") %}{{ d }}", new HashMap<>()));
+  }
+
+  @Test
+  public void itSetsDefaultObjectValue() {
+    assertThat(jinjava.render("{% set d=d | default({\"key\": \"value\"}) %}Value = {{ d.key }}", new HashMap<>())).isEqualTo("Value = value");    System.out.println(jinjava.render("{% set d=d | default({\"key\": \"value\"}) %}Value = {{ d.key }}", new HashMap<>()));
+  }
+
+  @Test
+  public void itChecksForType() {
+    assertThat(jinjava.render("{% set d=d | default({\"key\": \"value\"}) %}Type = {{ type(d.key) }}", new HashMap<>()))
+              .isEqualTo("Type = str");
+    assertThat(jinjava.render("{% set d=d | default(\"some random value\") %}{{ type(d) }}", new HashMap<>()))
+              .isEqualTo("str");
+    System.out.println(jinjava.render("{% set d=d | default(\"some random value\") %}{{ type(d) }}", new HashMap<>()));
+  }
+}