Partial code didnt have a proper identity callback.

Sam Pullara · Sam Pullara · commit 366df349686e · 2012-07-03T21:13:27.000-07:00
diff --git a/compiler/src/main/java/com/github/mustachejava/codes/PartialCode.java b/compiler/src/main/java/com/github/mustachejava/codes/PartialCode.java
@@ -7,6 +7,7 @@
 import com.github.mustachejava.TemplateContext;
 import com.github.mustachejava.util.LatchedWriter;
 
+import java.io.IOException;
 import java.io.Writer;
 import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutorService;
@@ -30,6 +31,21 @@ public PartialCode(TemplateContext tc, DefaultMustacheFactory cf, String variabl
     this(tc, cf, null, ">", variable);
   }
 
+  @Override
+  public void identity(Writer writer) {
+    try {
+      if (name != null) {
+        writer.write(tc.startChars());
+        writer.write(type);
+        writer.write(name);
+        writer.write(tc.endChars());
+      }
+      appendText(writer);
+    } catch (IOException e) {
+      throw new MustacheException(e);
+    }
+  }
+
   @Override
   public Code[] getCodes() {
     return partial.getCodes();
diff --git a/compiler/src/test/java/com/github/mustachejava/InterpreterTest.java b/compiler/src/test/java/com/github/mustachejava/InterpreterTest.java
@@ -8,6 +8,7 @@
 import org.codehaus.jackson.JsonNode;
 import org.codehaus.jackson.map.MappingJsonFactory;
 
+import javax.annotation.Nullable;
 import java.io.*;
 import java.util.Arrays;
 import java.util.HashMap;
@@ -274,7 +275,24 @@ public void testPartial() throws MustacheException, IOException {
     assertEquals(getContents(root, "template_partial.txt"), sw.toString());
   }
 
-  public void testComplex() throws MustacheException, IOException {
+    public void testPartialWithTF() throws MustacheException, IOException {
+        MustacheFactory c = init();
+        Mustache m = c.compile("partialintemplatefunction.html");
+        StringWriter sw = new StringWriter();
+        m.execute(sw, new Object() {
+            public TemplateFunction i() {
+                return new TemplateFunction() {
+                    @Override
+                    public String apply(@Nullable String s) {
+                        return s;
+                    }
+                };
+            }
+        });
+        assertEquals("This is not interesting.", sw.toString());
+    }
+
+    public void testComplex() throws MustacheException, IOException {
     StringWriter json = new StringWriter();
     MappingJsonFactory jf = new MappingJsonFactory();
     final JsonGenerator jg = jf.createJsonGenerator(json);
diff --git a/compiler/src/test/resources/partialintemplatefunction.html b/compiler/src/test/resources/partialintemplatefunction.html
@@ -0,0 +1 @@
+{{#i}}This is {{>uninterestingpartial}}{{/i}}
diff --git a/compiler/src/test/resources/uninterestingpartial.html b/compiler/src/test/resources/uninterestingpartial.html
@@ -0,0 +1 @@
+not interesting.

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+{{#i}}This is {{>uninterestingpartial}}{{/i}}`