Implement Display for all Templates

2 files changed, 23 insertions, 0 deletions

diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs
index b412c9d..713d9b6 100644
--- a/askama_derive/src/generator.rs
+++ b/askama_derive/src/generator.rs
@@ -47,6 +47,7 @@ pub fn generate(ast: &syn::DeriveInput, path: &str, mut nodes: Vec<Node>) -> Str
     } else {
         gen.impl_template(ast, &content);
     }
+    gen.impl_display(ast);
     gen.result()
 }
 
@@ -490,6 +491,15 @@ impl<'a> Generator<'a> {
         self.writeln("}");
     }
 
+    // Implement `Display` for the given context struct.
+    fn impl_display(&mut self, ast: &syn::DeriveInput) {
+        self.write_header(ast, "::std::fmt::Display");
+        self.writeln("fn fmt(&self, f: &mut ::std::fmt::Formatter) -> Result<(), ::std::fmt::Error> {");
+        self.writeln("self.render_into(f)");
+        self.writeln("}");
+        self.writeln("}");
+    }
+
     // Implement `Deref<Parent>` for an inheriting context struct.
     fn deref_to_parent(&mut self, ast: &syn::DeriveInput, parent_type: &syn::Ty) {
         self.write_header(ast, "::std::ops::Deref");
diff --git a/testing/tests/simple.rs b/testing/tests/simple.rs
index b9b7a7c..896e082 100644
--- a/testing/tests/simple.rs
+++ b/testing/tests/simple.rs
@@ -154,3 +154,16 @@ fn test_json() {
     let t = JsonTemplate { foo: "a", bar: "b" };
     assert_eq!(t.render().unwrap(), "{\"foo\": \"a\", \"bar\": \"b\"}");
 }
+
+
+#[derive(Template)]
+#[template(path = "composition.html")]
+struct CompositionTemplate {
+    foo: IfTemplate,
+}
+
+#[test]
+fn test_composition() {
+    let t = CompositionTemplate { foo: IfTemplate { cond: true } };
+    assert_eq!(t.render().unwrap(), "composed: true");
+}