jdrouet/mrml

use crate::prelude::print::{PrintableAttributes, PrintableElement};

impl PrintableAttributes for super::MjIncludeBodyAttributes {
    fn print<P: crate::prelude::print::Printer>(&self, printer: &mut P) -> std::fmt::Result {
        printer.push_attribute("path", self.path.as_str())?;
        if !self.kind.is_default() {
            printer.push_attribute("type", self.kind.as_ref())?;
        }
        Ok(())
    }
}

impl PrintableElement for super::MjIncludeBody {
    type Attrs = super::MjIncludeBodyAttributes;
    type Children = ();

    fn tag(&self) -> &str {
        crate::mj_include::NAME
    }

    fn attributes(&self) -> &Self::Attrs {
        &self.0.attributes
    }

    fn children(&self) -> &Self::Children {
        &()
    }
}

#[cfg(test)]
mod tests {
    use crate::mj_button::MjButton;
    use crate::mj_include::body::{
        MjIncludeBody, MjIncludeBodyAttributes, MjIncludeBodyChild, MjIncludeBodyKind,
    };
    use crate::prelude::print::Printable;

    #[test]
    fn kind_string() {
        assert_eq!(MjIncludeBodyKind::Html.as_ref(), "html");
        assert_eq!(MjIncludeBodyKind::Mjml.as_ref(), "mjml");
    }

    #[test]
    fn simple() {
        let elt = MjIncludeBody::new(
            MjIncludeBodyAttributes::new("memory:include.mjml"),
            vec![MjIncludeBodyChild::MjButton(MjButton::default())],
        );
        assert_eq!(
            elt.print_dense().unwrap(),
            "<mj-include path=\"memory:include.mjml\" />"
        );
    }

    #[test]
    fn html_kind() {
        let elt = MjIncludeBody::new(
            MjIncludeBodyAttributes::new("memory:include.html").with_kind(MjIncludeBodyKind::Html),
            vec![MjIncludeBodyChild::MjButton(MjButton::default())],
        );
        assert_eq!(
            elt.print_dense().unwrap(),
            "<mj-include path=\"memory:include.html\" type=\"html\" />"
        );
    }
}