From 85ad2e6ba3a205e9b648431318aac0e75c027a82 Mon Sep 17 00:00:00 2001 From: René Kijewski Date: Fri, 28 Jan 2022 10:48:46 +0100 Subject: Implement error propagation expression: `?` (#590) This change allows using the operator `?` in askama expressions. It works like the same operator in Rust: if a `Result` is `Ok`, it is unwrapped. If it is an error, then the `render()` method fails with this error value. --- testing/tests/try.rs | 69 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 testing/tests/try.rs (limited to 'testing') diff --git a/testing/tests/try.rs b/testing/tests/try.rs new file mode 100644 index 0000000..e199f9c --- /dev/null +++ b/testing/tests/try.rs @@ -0,0 +1,69 @@ +use askama::Template; + +#[derive(Template)] +#[template(source = "{% let v = self.parse()? %}{{s}}={{v}}", ext = "txt")] +struct IntParserTemplate<'a> { + s: &'a str, +} + +impl IntParserTemplate<'_> { + fn parse(&self) -> Result { + self.s.parse() + } +} + +#[test] +fn test_int_parser() { + let template = IntParserTemplate { s: "💯" }; + assert!(matches!(template.render(), Err(askama::Error::Custom(_)))); + assert_eq!( + format!("{}", &template.render().unwrap_err()), + "invalid digit found in string" + ); + + let template = IntParserTemplate { s: "100" }; + assert_eq!(template.render().unwrap(), "100=100"); +} + +#[derive(Template)] +#[template(source = "{{ value()? }}", ext = "txt")] +struct FailFmt { + value: fn() -> Result<&'static str, std::fmt::Error>, +} + +#[test] +fn fail_fmt() { + let template = FailFmt { + value: || Err(std::fmt::Error), + }; + assert!(matches!(template.render(), Err(askama::Error::Custom(_)))); + assert_eq!( + format!("{}", &template.render().unwrap_err()), + format!("{}", std::fmt::Error) + ); + + let template = FailFmt { + value: || Ok("hello world"), + }; + assert_eq!(template.render().unwrap(), "hello world"); +} + +#[derive(Template)] +#[template(source = "{{ value()? }}", ext = "txt")] +struct FailStr { + value: fn() -> Result<&'static str, &'static str>, +} + +#[test] +fn fail_str() { + let template = FailStr { + value: || Err("FAIL"), + }; + assert!(matches!(template.render(), Err(askama::Error::Custom(_)))); + assert_eq!(format!("{}", &template.render().unwrap_err()), "FAIL"); + + let template = FailStr { + value: || Ok("hello world"), + }; + assert_eq!(template.render().unwrap(), "hello world"); +} -- cgit