Merge pull request #729 from GuillaumeGomez/fix-print

Fix print = "code"

Author: GuillaumeGomez

askama_derive/src/lib.rs

@@ -480,10 +480,15 @@ fn build_template_item(
     }
 
     if input.print == Print::Ast || input.print == Print::All {
-        eprintln!("{:?}", templates[&input.path].nodes());
+        eprintln!("== Askama AST ==\n{:?}", templates[&input.path].nodes());
     }
 
     let size_hint = template_to_string(buf, &input, &contexts, heritage.as_ref(), tmpl_kind)?;
+
+    if input.print == Print::Code || input.print == Print::All {
+        eprintln!("== Askama code ==\n{}", buf.to_token_stream());
+    }
+
     Ok(size_hint)
 }
 

testing/tests/print.rs

@@ -0,0 +1,24 @@
+use std::path::Path;
+use std::process::Command;
+
+#[test]
+fn test_print_config() {
+    let Ok(cargo_home) = std::env::var("CARGO_MANIFEST_DIR") else {
+        panic!(">> cannot get `CARGO_MANIFEST_DIR` env variable");
+    };
+    let output = Command::new("cargo")
+        .arg("check")
+        .current_dir(Path::new(&cargo_home).join("tests/print"))
+        .output()
+        .unwrap();
+
+    let content = String::from_utf8_lossy(&output.stderr);
+
+    if !output.status.success() {
+        panic!("Failed to build `tests/print`: {content}");
+    }
+
+    // There should be two of each (one for "all" and one for "ast"/"code").
+    assert_eq!(content.split("== Askama AST ==").count(), 3);
+    assert_eq!(content.split("== Askama code ==").count(), 3);
+}

testing/tests/print/Cargo.toml

@@ -0,0 +1,11 @@
+[package]
+name = "print"
+edition = "2024"
+
+[lib]
+path = "./lib.rs"
+
+[dependencies]
+askama = { path = "../../../askama" }
+
+[workspace]

testing/tests/print/lib.rs

@@ -0,0 +1,16 @@
+// This test (coupled with `testing/tests/print.rs`) ensures that `print = "..."`
+// works as expected.
+
+use askama::Template;
+
+#[derive(Template)]
+#[template(source = r#"{% let x = 12 %}{{x}}"#, ext = "txt", print = "all")]
+struct PrintAll;
+
+#[derive(Template)]
+#[template(source = r#"{% let x = 12 %}{{x}}"#, ext = "txt", print = "ast")]
+struct PrintAst;
+
+#[derive(Template)]
+#[template(source = r#"{% let x = 12 %}{{x}}"#, ext = "txt", print = "code")]
+struct PrintCode;