tests.rs

//! Files containing tests for generated code.

use std::fmt;
use std::path::Path;

use console::style;
use prettyplease::unparse;
use similar::{Algorithm, ChangeTag, TextDiffConfig};

use crate::integration::Buffer;

fn build_template(ast: &syn::DeriveInput) -> Result<String, crate::CompileError> {
    let mut buf = Buffer::new();
    crate::build_template(&mut buf, ast)?;
    Ok(buf.into_string())
}

// This function makes it much easier to compare expected code by adding the wrapping around
// the code we want to check.
#[track_caller]
fn compare(jinja: &str, expected: &str, fields: &[(&str, &str)], size_hint: usize) {
    let generated = jinja_to_rust(jinja, fields).unwrap();

    let expected: proc_macro2::TokenStream = expected.parse().unwrap();
    let expected: syn::File = syn::parse_quote! {
        const _: () = {
            extern crate rinja as rinja;

            impl rinja::Template for Foo {
                fn render_into_with_values<RinjaW>(
                    &self,
                    __rinja_writer: &mut RinjaW,
                    __rinja_values: &dyn rinja::Values,
                ) -> rinja::Result<()>
                where
                    RinjaW: rinja::helpers::core::fmt::Write + ?rinja::helpers::core::marker::Sized,
                {
                    #[allow(unused_imports)]
                    use rinja::{
                        filters::{AutoEscape as _, WriteWritable as _},
                        helpers::{ResultConverter as _, core::fmt::Write as _},
                    };
                    #expected
                    rinja::Result::Ok(())
                }
                const SIZE_HINT: rinja::helpers::core::primitive::usize = #size_hint;
            }

            /// Implement the [`format!()`][rinja::helpers::std::format] trait for [`Foo`]
            ///
            /// Please be aware of the rendering performance notice in the [`Template`][rinja::Template] trait.
            impl rinja::helpers::core::fmt::Display for Foo {
                #[inline]
                fn fmt(&self, f: &mut rinja::helpers::core::fmt::Formatter<'_>) -> rinja::helpers::core::fmt::Result {
                    rinja::Template::render_into(self, f).map_err(|_| rinja::helpers::core::fmt::Error)
                }
            }

            impl rinja::filters::FastWritable for Foo {
                #[inline]
                fn write_into<RinjaW>(&self, dest: &mut RinjaW) -> rinja::Result<()>
                where
                    RinjaW: rinja::helpers::core::fmt::Write + ?rinja::helpers::core::marker::Sized,
                {
                    rinja::Template::render_into(self, dest)
                }
            }
        };
    };

    let expected = unparse(&expected);
    let generated = unparse(&generated);
    if expected != generated {
        struct Diff<'a>(&'a str, &'a str);

        impl fmt::Display for Diff<'_> {
            fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
                let diff = TextDiffConfig::default()
                    .algorithm(Algorithm::Patience)
                    .diff_lines(self.0, self.1);
                for change in diff.iter_all_changes() {
                    let (change, line) = match change.tag() {
                        ChangeTag::Equal => (
                            style(" ").dim().bold(),
                            style(change.to_string_lossy()).dim(),
                        ),
                        ChangeTag::Delete => (
                            style("-").red().bold(),
                            style(change.to_string_lossy()).red(),
                        ),
                        ChangeTag::Insert => (
                            style("+").green().bold(),
                            style(change.to_string_lossy()).green(),
                        ),
                    };
                    write!(f, "{change}{line}")?;
                }
                Ok(())
            }
        }

        panic!(
            "\n\
            === Expected ===\n\
            \n\
            {expected}\n\
            \n\
            === Generated ===\n\
            \n\
            {generated}\n\
            \n\
            === Diff ===\n\
            \n\
            {diff}\n\
            \n\
            === FAILURE ===",
            expected = style(&expected).red(),
            generated = style(&generated).green(),
            diff = Diff(&expected, &generated),
        );
    }
}

fn jinja_to_rust(jinja: &str, fields: &[(&str, &str)]) -> syn::Result<syn::File> {
    let jinja = format!(
        r##"#[template(source = {jinja:?}, ext = "txt")]
struct Foo {{ {} }}"##,
        fields
            .iter()
            .map(|(name, type_)| format!("{name}: {type_}"))
            .collect::<Vec<_>>()
            .join(","),
    );

    let generated = build_template(&syn::parse_str::<syn::DeriveInput>(&jinja).unwrap()).unwrap();
    let generated = match generated.parse() {
        Ok(generated) => generated,
        Err(err) => panic!(
            "\n\
            === Invalid code generated ===\n\
            \n\
            {generated}\n\
            \n\
            === Error ===\n\
            \n\
            {err}"
        ),
    };
    syn::parse2(generated)
}

#[test]
fn check_if_let() {
    // In this test, we ensure that `query` never is `self.query`.
    compare(
        "{% if let Some(query) = s && !query.is_empty() %}{{query}}{% endif %}",
        r"if let Some(query,) = &self.s && !rinja::helpers::as_bool(&(query.is_empty())) {
    match (
        &((&&rinja::filters::AutoEscaper::new(&(query), rinja::filters::Text)).rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}",
        &[],
        3,
    );

    // In this test, we ensure that `s` is `self.s` only in the first `if let Some(s) = self.s`
    // condition.
    compare(
        "{% if let Some(s) = s %}{{ s }}{% endif %}",
        r"if let Some(s,) = &self.s {
    match (
        &((&&rinja::filters::AutoEscaper::new(&(s), rinja::filters::Text)).rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}",
        &[],
        3,
    );

    // In this test, we ensure that `s` is `self.s` only in the first `if let Some(s) = self.s`
    // condition.
    compare(
        "{% if let Some(s) = s && !s.is_empty() %}{{s}}{% endif %}",
        r"if let Some(s,) = &self.s && !rinja::helpers::as_bool(&(s.is_empty())) {
    match (
        &((&&rinja::filters::AutoEscaper::new(&(s), rinja::filters::Text)).rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}",
        &[],
        3,
    );
}

// Since this feature is not stable yet, we can't add a "normal" test for it so instead we check
// the generated code.
#[test]
fn check_if_let_chain() {
    // Both `bla` and `blob` variables must exist in this `if`.
    compare(
        "{% if let Some(bla) = y && x && let Some(blob) = y %}{{bla}} {{blob}}{% endif %}",
        r#"if let Some(bla) = &self.y && rinja::helpers::as_bool(&(self.x)) && let Some(blob) = &self.y {
    match (
        &((&&rinja::filters::AutoEscaper::new(&(bla), rinja::filters::Text))
            .rinja_auto_escape()?),
        &((&&rinja::filters::AutoEscaper::new(&(blob), rinja::filters::Text))
            .rinja_auto_escape()?),
    ) {
        (expr0, expr2) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            __rinja_writer.write_str(" ")?;
            (&&&rinja::filters::Writable(expr2)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}"#,
        &[],
        7,
    );

    compare(
        r#"{% if let Some(bla) = y
            && bla == "x"
            && let Some(blob) = z
            && blob == "z" %}{{bla}} {{blob}}{% endif %}"#,
        r#"if let Some(bla) = &self.y && rinja::helpers::as_bool(&(bla == "x"))
             && let Some(blob) = &self.z && rinja::helpers::as_bool(&(blob == "z"))
{
    match (
        &((&&rinja::filters::AutoEscaper::new(&(bla), rinja::filters::Text))
            .rinja_auto_escape()?),
        &((&&rinja::filters::AutoEscaper::new(&(blob), rinja::filters::Text))
            .rinja_auto_escape()?),
    ) {
        (expr0, expr2) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            __rinja_writer.write_str(" ")?;
            (&&&rinja::filters::Writable(expr2)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}"#,
        &[],
        7,
    );

    // Bindings variables with the same name as the bound variable should be declared in the right
    // order.
    compare(
        r#"{% if let Some(y) = y
            && y == "x"
            && w
            && let Some(z) = z
            && z == "z" %}{{y}} {{z}}{% endif %}"#,
        r#"if let Some(y) = &self.y && self.y == "x"
    && rinja::helpers::as_bool(&(self.w)) && let Some(z) = &self.z
    && rinja::helpers::as_bool(&(z == "z"))
{
    match (
        &((&&rinja::filters::AutoEscaper::new(&(y), rinja::filters::Text))
            .rinja_auto_escape()?),
        &((&&rinja::filters::AutoEscaper::new(&(z), rinja::filters::Text))
            .rinja_auto_escape()?),
    ) {
        (expr0, expr2) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            __rinja_writer.write_str(" ")?;
            (&&&rinja::filters::Writable(expr2)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}"#,
        &[],
        7,
    );

    compare(
        r#"{% if w
            && let Some(y) = y
            && y == "x"
            && let Some(z) = z
            && z == "z" %}{{y}} {{z}}{% endif %}"#,
        r#"if rinja::helpers::as_bool(&(self.w)) && let Some(y) = &self.y
    && self.y == "x" && let Some(z) = &self.z
    && rinja::helpers::as_bool(&(z == "z"))
{
    match (
        &((&&rinja::filters::AutoEscaper::new(&(y), rinja::filters::Text))
            .rinja_auto_escape()?),
        &((&&rinja::filters::AutoEscaper::new(&(z), rinja::filters::Text))
            .rinja_auto_escape()?),
    ) {
        (expr0, expr2) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            __rinja_writer.write_str(" ")?;
            (&&&rinja::filters::Writable(expr2)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}"#,
        &[],
        7,
    );
}

#[test]
fn check_includes_only_once() {
    // In this test we make sure that every used template gets referenced exactly once.
    let path = Path::new(env!("CARGO_MANIFEST_DIR")).join("templates");
    let path1 = path.join("include1.html").canonicalize().unwrap();
    let path2 = path.join("include2.html").canonicalize().unwrap();
    let path3 = path.join("include3.html").canonicalize().unwrap();
    compare(
        r#"{% include "include1.html" %}"#,
        &format!(
            r#"const _: &[rinja::helpers::core::primitive::u8] = rinja::helpers::core::include_bytes!({path1:#?});
            const _: &[rinja::helpers::core::primitive::u8] = rinja::helpers::core::include_bytes!({path2:#?});
            const _: &[rinja::helpers::core::primitive::u8] = rinja::helpers::core::include_bytes!({path3:#?});
            __rinja_writer.write_str("3333")?;"#
        ),
        &[],
        4,
    );
}

#[test]
fn check_is_defined() {
    // Checks that it removes conditions if we know at compile-time that they always return false.
    //
    // We're forced to add `bla` otherwise `compare` assert fails in weird ways...
    compare(
        "{% if y is defined %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if x is not defined %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[("x", "u32")],
        3,
    );
    compare(
        "{% if y is defined && x is not defined %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[("x", "u32")],
        3,
    );

    // Same with declared variables.
    compare(
        "{% set y = 12 %}
         {%- if y is not defined %}{{query}}{% endif %}bla",
        r#"let y = 12;
__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% set y = 12 %}
         {%- if y is not defined && x is defined %}{{query}}{% endif %}bla",
        r#"let y = 12;
__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );

    // Checks that if the condition is always `true` at compile-time, then we keep the code but
    // remove the condition.
    compare(
        "{% if y is defined %}bla{% endif %}",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[("y", "u32")],
        3,
    );
    compare(
        "{% if x is not defined %}bla{% endif %}",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    // Same with declared variables.
    compare(
        "{% set y = 12 %}
         {%- if y is defined %}bla{% endif %}",
        r#"let y = 12;
__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );

    // If the always `true` condition is followed by more `else if`/`else`, check that they are
    // removed as well.
    compare(
        "{% if x is defined %}bli
         {%- else if x == 12 %}12{% endif %}bla",
        r#"__rinja_writer.write_str("blibla")?;"#,
        &[("x", "u32")],
        6,
    );
    compare(
        "{% if x is defined %}bli
         {%- else if x == 12 %}12
         {%- else %}nope{% endif %}bla",
        r#"__rinja_writer.write_str("blibla")?;"#,
        &[("x", "u32")],
        6,
    );
    // If it's not the first one.
    compare(
        "{% if x == 12 %}bli
         {%- else if x is defined %}12
         {%- else %}nope{% endif %}",
        r#"if rinja::helpers::as_bool(&(self.x == 12)) {
__rinja_writer.write_str("bli")?;
} else {
__rinja_writer.write_str("12")?;
}"#,
        &[("x", "u32")],
        5,
    );

    // Checking that it doesn't remove the condition if other non-"if (not) defined" checks
    // are present.
    compare(
        "{% if y is defined || x == 12 %}{{x}}{% endif %}",
        r"if rinja::helpers::as_bool(&(self.x == 12)) {
    match (
        &((&&rinja::filters::AutoEscaper::new(&(self.x), rinja::filters::Text)).rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}
",
        &[("x", "u32")],
        3,
    );
    compare(
        "{% if y is defined || x == 12 %}{{x}}{% endif %}",
        r"match (
    &((&&rinja::filters::AutoEscaper::new(&(self.x), rinja::filters::Text)).rinja_auto_escape()?),
) {
    (expr0,) => {
        (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
    }
}
",
        &[("y", "u32"), ("x", "u32")],
        3,
    );
    compare(
        "{% if y is defined && y == 12 %}{{x}}{% endif %}",
        r"",
        &[],
        0,
    );
    compare(
        "{% if y is defined && y == 12 %}{{y}}{% else %}bli{% endif %}",
        r#"__rinja_writer.write_str("bli")?;"#,
        &[],
        3,
    );
    compare(
        "{% if y is defined && y == 12 %}{{y}}{% else %}bli{% endif %}",
        r#"
if rinja::helpers::as_bool(&(self.y == 12)) {
    match (
        &((&&rinja::filters::AutoEscaper::new(
            &(self.y),
            rinja::filters::Text,
        ))
            .rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
} else {
    __rinja_writer.write_str("bli")?;
}
"#,
        &[("y", "u32")],
        6,
    );
    // Since the first `if` is always `true`, the `else` should not be generated.
    compare(
        "{% if y is defined %}{{y}}{% else %}bli{% endif %}",
        r"
match (
    &((&&rinja::filters::AutoEscaper::new(
        &(self.y),
        rinja::filters::Text,
    ))
        .rinja_auto_escape()?),
) {
    (expr0,) => {
        (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
    }
}
",
        &[("y", "u32")],
        3,
    );

    // Checking some funny cases.

    // This one is a bit useless because you can use `is not defined` but I suppose it's possible
    // to encounter cases like that in the wild so better have a check.
    compare(
        "{% if !(y is defined) %}bla{% endif %}",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if !(y is not defined) %}bli{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if !(y is defined) %}bli{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[("y", "u32")],
        3,
    );
    compare(
        "{% if !(y is not defined) %}bla{% endif %}",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[("y", "u32")],
        3,
    );

    // Ensure that the `!` is kept .
    compare(
        "{% if y is defined && !y %}bla{% endif %}",
        r#"if !rinja::helpers::as_bool(&(self.y)) {
    __rinja_writer.write_str("bla")?;
}"#,
        &[("y", "bool")],
        3,
    );
    compare(
        "{% if y is defined && !(y) %}bla{% endif %}",
        r#"if !(rinja::helpers::as_bool(&(self.y))) {
    __rinja_writer.write_str("bla")?;
}"#,
        &[("y", "bool")],
        3,
    );
    compare(
        "{% if y is not defined || !y %}bla{% endif %}",
        r#"if !rinja::helpers::as_bool(&(self.y)) {
    __rinja_writer.write_str("bla")?;
}"#,
        &[("y", "bool")],
        3,
    );
    compare(
        "{% if y is not defined || !(y) %}bla{% endif %}",
        r#"if !(rinja::helpers::as_bool(&(self.y))) {
    __rinja_writer.write_str("bla")?;
}"#,
        &[("y", "bool")],
        3,
    );
}

#[test]
fn check_bool_conditions() {
    // Checks that it removes conditions if we know at compile-time that they always return false.
    //
    // We're forced to add `bla` otherwise `compare` assert fails in weird ways...
    compare(
        "{% if false %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if false && false %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if false && true %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if true && false %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if false || true %}bli{% endif %}bla",
        r#"__rinja_writer.write_str("blibla")?;"#,
        &[],
        6,
    );
    compare(
        "{% if true || false %}bli{% endif %}bla",
        r#"__rinja_writer.write_str("blibla")?;"#,
        &[],
        6,
    );

    compare(
        "{% if true || x == 12 %}{{x}}{% endif %}",
        r"match (
    &((&&rinja::filters::AutoEscaper::new(&(self.x), rinja::filters::Text)).rinja_auto_escape()?),
) {
    (expr0,) => {
        (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
    }
}
",
        &[("x", "u32")],
        3,
    );
    compare(
        "{% if false || x == 12 %}{{x}}{% endif %}",
        r"if rinja::helpers::as_bool(&(self.x == 12)) {
    match (
        &((&&rinja::filters::AutoEscaper::new(
            &(self.x),
            rinja::filters::Text,
        ))
            .rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}
",
        &[("x", "u32")],
        3,
    );

    // Checking that it also works with sub conditions.

    // It's important here that the `(true || x == 12)` part remains since it's not first in the
    // condition.
    compare(
        "{% if y == 3 || (true || x == 12) %}{{x}}{% endif %}",
        r"if rinja::helpers::as_bool(&(self.y == 3)) || (true) {
    match (
        &((&&rinja::filters::AutoEscaper::new(
            &(self.x),
            rinja::filters::Text,
        ))
            .rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}
",
        &[],
        3,
    );
    // However in this case, since `(true || x == 12)` is evaluated to `true`, `y == 3` will never
    // be evaluated so the whole code is removed.
    compare(
        "{% if (true || x == 12) || y == 3 %}{{x}}{% endif %}",
        r"match (
    &((&&rinja::filters::AutoEscaper::new(
        &(self.x),
        rinja::filters::Text,
    ))
        .rinja_auto_escape()?),
) {
    (expr0,) => {
        (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
    }
}
",
        &[],
        3,
    );
    compare(
        "{% if y == 3 || (x == 12 || true) %}{{x}}{% endif %}",
        r"
if rinja::helpers::as_bool(&(self.y == 3))
    || (rinja::helpers::as_bool(&(self.x == 12)) || true)
{
    match (
        &((&&rinja::filters::AutoEscaper::new(
            &(self.x),
            rinja::filters::Text,
        ))
            .rinja_auto_escape()?),
    ) {
        (expr0,) => {
            (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
        }
    }
}
",
        &[],
        3,
    );

    // Some funny cases.
    compare(
        "{% if !(false) %}bla{% endif %}",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );
    compare(
        "{% if !(true) %}{{query}}{% endif %}bla",
        r#"__rinja_writer.write_str("bla")?;"#,
        &[],
        3,
    );

    // Complex condition
    compare(
        "{% if (a || !b) && !(c || !d) %}x{% endif %}",
        r#"
            if (
                rinja::helpers::as_bool(&(self.a))
                || !rinja::helpers::as_bool(&(self.b))
            ) && !(
                rinja::helpers::as_bool(&(self.c))
                || !rinja::helpers::as_bool(&(self.d))
            ) {
                __rinja_writer.write_str("x")?;
            }"#,
        &[("a", "i32"), ("b", "i32"), ("c", "i32"), ("d", "i32")],
        1,
    );
}

#[test]
fn check_escaping_at_compile_time() {
    compare(
        r#"The card is
        {%- match suit %}
            {%- when Suit::Clubs or Suit::Spades -%}
                {{ " black" }}
            {%- when Suit::Diamonds or Suit::Hearts -%}
                {{ " red" }}
        {%- endmatch %}"#,
        r#"__rinja_writer.write_str("The card is")?;
        match &self.suit {
            Suit::Clubs {} | Suit::Spades {} => {
                __rinja_writer.write_str(" black")?;
            }
            Suit::Diamonds {} | Suit::Hearts {} => {
                __rinja_writer.write_str(" red")?;
            }
        }"#,
        &[("suit", "Suit")],
        16,
    );

    compare(
        r#"{{ '\x41' }}{{ '\n' }}{{ '\r' }}{{ '\t' }}{{ '\\' }}{{ '\u{2665}' }}{{ '\'' }}{{ '\"' }}{{ '"' }}
{{ "\x41\n\r\t\\\u{2665}\'\"'" }}"#,
        r#"__rinja_writer.write_str("A
\r	\\♥'\"\"
A
\r	\\♥'\"'")?;"#,
        &[],
        23,
    );

    compare(
        r"{{ 1_2_3_4 }} {{ 4e3 }} {{ false }}",
        r#"__rinja_writer.write_str("1234 4000 false")?;"#,
        &[],
        15,
    );
}

#[cfg(feature = "code-in-doc")]
#[test]
fn test_code_in_comment() {
    let ts = r#"
        #[template(ext = "txt", in_doc = true)]
        /// ```rinja
        /// Hello world!
        /// ```
        struct Tmpl;
    "#;
    let ast = syn::parse_str(ts).unwrap();
    let generated = build_template(&ast).unwrap();
    assert!(generated.contains("Hello world!"));
    assert!(!generated.contains("compile_error"));

    let ts = r#"
        #[template(ext = "txt", in_doc = true)]
        /// ```rinja
        /// Hello
        /// world!
        /// ```
        struct Tmpl;
    "#;
    let ast = syn::parse_str(ts).unwrap();
    let generated = build_template(&ast).unwrap();
    assert!(generated.contains("Hello\nworld!"));
    assert!(!generated.contains("compile_error"));

    let ts = r#"
        /// ```rinja
        /// Hello
        #[template(ext = "txt", in_doc = true)]
        /// world!
        /// ```
        struct Tmpl;
    "#;
    let ast = syn::parse_str(ts).unwrap();
    let generated = build_template(&ast).unwrap();
    assert!(generated.contains("Hello\nworld!"));
    assert!(!generated.contains("compile_error"));

    let ts = r#"
        /// This template greets the whole world
        ///
        /// ```rinja
        /// Hello
        #[template(ext = "txt", in_doc = true)]
        /// world!
        /// ```
        ///
        /// Some more text.
        struct Tmpl;
    "#;
    let ast = syn::parse_str(ts).unwrap();
    let generated = build_template(&ast).unwrap();
    assert!(generated.contains("Hello\nworld!"));
    assert!(!generated.contains("compile_error"));

    let ts = "
        #[template(ext = \"txt\", in_doc = true)]
        #[doc = \"```rinja\nHello\nworld!\n```\"]
        struct Tmpl;
    ";
    let ast = syn::parse_str(ts).unwrap();
    let generated = build_template(&ast).unwrap();
    assert!(generated.contains("Hello\nworld!"));
    assert!(!generated.contains("compile_error"));

    let ts = "
        #[template(ext = \"txt\", in_doc = true)]
        /// `````
        /// ```rinja
        /// {{bla}}
        /// ```
        /// `````
        struct BlockOnBlock;
    ";
    let ast = syn::parse_str(ts).unwrap();
    let err = build_template(&ast).unwrap_err();
    assert_eq!(
        err.to_string(),
        "when using `in_doc` with the value `true`, the struct's documentation needs a `rinja` \
         code block"
    );

    let ts = "
        #[template(ext = \"txt\", in_doc = true)]
        /// ```rinja
        /// `````
        /// {{bla}}
        /// `````
        /// ```
        struct BlockOnBlock;
    ";
    let ast = syn::parse_str(ts).unwrap();
    let generated = build_template(&ast).unwrap();
    assert!(!generated.contains("compile_error"));
}

#[test]
fn test_pluralize() {
    compare(
        r"{{dogs}} dog{{dogs|pluralize}}",
        r#"
        match (
            &((&&rinja::filters::AutoEscaper::new(
                &(self.dogs),
                rinja::filters::Text,
            ))
                .rinja_auto_escape()?),
            &(rinja::filters::pluralize(
                &(self.dogs),
                rinja::helpers::Empty,
                rinja::filters::Safe("s"),
            )?),
        ) {
            (expr0, expr3) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
                __rinja_writer.write_str(" dog")?;
                (&&&rinja::filters::Writable(expr3)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }"#,
        &[("dogs", "i8")],
        10,
    );
    compare(
        r#"{{dogs}} dog{{dogs|pluralize("go")}}"#,
        r#"
        match (
            &((&&rinja::filters::AutoEscaper::new(
                &(self.dogs),
                rinja::filters::Text,
            ))
                .rinja_auto_escape()?),
            &(rinja::filters::pluralize(
                &(self.dogs),
                rinja::filters::Safe("go"),
                rinja::filters::Safe("s"),
            )?),
        ) {
            (expr0, expr3) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
                __rinja_writer.write_str(" dog")?;
                (&&&rinja::filters::Writable(expr3)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }"#,
        &[("dogs", "i8")],
        10,
    );
    compare(
        r#"{{mice}} {{mice|pluralize("mouse", "mice")}}"#,
        r#"
        match (
            &((&&rinja::filters::AutoEscaper::new(
                &(self.mice),
                rinja::filters::Text,
            ))
                .rinja_auto_escape()?),
            &(rinja::filters::pluralize(
                &(self.mice),
                rinja::filters::Safe("mouse"),
                rinja::filters::Safe("mice"),
            )?),
        ) {
            (expr0, expr2) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
                __rinja_writer.write_str(" ")?;
                (&&&rinja::filters::Writable(expr2)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }"#,
        &[("dogs", "i8")],
        7,
    );

    compare(
        r"{{count|pluralize(one, count)}}",
        r"
        match (
            &(rinja::filters::pluralize(
                &(self.count),
                (&&rinja::filters::AutoEscaper::new(
                    &(self.one),
                    rinja::filters::Text,
                ))
                    .rinja_auto_escape()?,
                (&&rinja::filters::AutoEscaper::new(
                    &(self.count),
                    rinja::filters::Text,
                ))
                    .rinja_auto_escape()?,
            )?),
        ) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        ",
        &[("count", "i8"), ("one", "&'static str")],
        3,
    );

    compare(
        r"{{0|pluralize(sg, pl)}}",
        r"
        match (
            &((&&rinja::filters::AutoEscaper::new(&(self.pl), rinja::filters::Text))
                .rinja_auto_escape()?),
        ) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        ",
        &[("sg", "&'static str"), ("pl", "&'static str")],
        3,
    );
    compare(
        r"{{1|pluralize(sg, pl)}}",
        r"
        match (
            &((&&rinja::filters::AutoEscaper::new(&(self.sg), rinja::filters::Text))
                .rinja_auto_escape()?),
        ) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        ",
        &[("sg", "&'static str"), ("pl", "&'static str")],
        3,
    );

    compare(
        r#"{{0|pluralize("sg", "pl")}}"#,
        r#"
        match (&(rinja::filters::Safe("pl")),) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        "#,
        &[],
        3,
    );
    compare(
        r#"{{1|pluralize("sg", "pl")}}"#,
        r#"
        match (&(rinja::filters::Safe("sg")),) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        "#,
        &[],
        3,
    );

    compare(
        r"{{0|pluralize}}",
        r#"
        match (&(rinja::filters::Safe("s")),) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        "#,
        &[],
        3,
    );
    compare(
        r"{{1|pluralize}}",
        r"
        match (&(rinja::helpers::Empty),) {
            (expr0,) => {
                (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
            }
        }
        ",
        &[],
        3,
    );
}

#[test]
fn test_concat() {
    compare(
        r#"{{ "<" ~ a ~ "|" ~ b ~ '>' }}"#,
        r#"
            __rinja_writer.write_str("<")?;
            match (
                &((&&rinja::filters::AutoEscaper::new(&(self.a), rinja::filters::Text))
                    .rinja_auto_escape()?),
                &((&&rinja::filters::AutoEscaper::new(&(self.b), rinja::filters::Text))
                    .rinja_auto_escape()?),
            ) {
                (expr1, expr3) => {
                    (&&&rinja::filters::Writable(expr1)).rinja_write(__rinja_writer, __rinja_values)?;
                    __rinja_writer.write_str("|")?;
                    (&&&rinja::filters::Writable(expr3)).rinja_write(__rinja_writer, __rinja_values)?;
                }
            }
            __rinja_writer.write_str(">")?;
        "#,
        &[("a", "&'static str"), ("b", "u32")],
        9,
    );

    compare(
        r#"{{ ("a=" ~ a ~ " b=" ~ b)|upper }}"#,
        r#"
            match (
                &((&&rinja::filters::AutoEscaper::new(
                    &(rinja::filters::upper(
                        &((rinja::helpers::Concat(
                            &(rinja::helpers::Concat(&("a="), &(self.a))),
                            &(rinja::helpers::Concat(&(" b="), &(self.b))),
                        ))),
                    )?),
                    rinja::filters::Text,
                ))
                    .rinja_auto_escape()?),
            ) {
                (expr0,) => {
                    (&&&rinja::filters::Writable(expr0)).rinja_write(__rinja_writer, __rinja_values)?;
                }
            }
        "#,
        &[("a", "&'static str"), ("b", "u32")],
        3,
    );
}

#[test]
fn extends_with_whitespace_control() {
    const CONTROL: &[&str] = &["", "\t", "-", "+", "~"];

    let expected = jinja_to_rust(r#"front {% extends "a.html" %} back"#, &[]).unwrap();
    let expected = unparse(&expected);
    for front in CONTROL {
        for back in CONTROL {
            let src = format!(r#"front {{%{front} extends "a.html" {back}%}} back"#);
            let actual = jinja_to_rust(&src, &[]).unwrap();
            let actual = unparse(&actual);
            assert_eq!(expected, actual, "source: {:?}", src);
        }
    }
}