dioxus/packages/autofmt/tests/sink.rs

use dioxus_autofmt::*;
use proc_macro2::TokenStream as TokenStream2;
use syn::{Attribute, Meta};

#[test]
fn formats_block() {
    let block = r#"
        div {
                                    div {
                                    class: "asd",
                                    class: "asd",class: "asd",class: "asd",class: "asd",class: "asd",
                                    key: "ddd",
                                    onclick: move |_| {
                                        let blah = 120;
                                                    true
                                    },
                                    blah: 123,
                                    onclick: move |_| {
                                        let blah = 120;
                                                    true
                                    },
                                    onclick: move |_| {
                                        let blah = 120;
                                                    true
                                    },
                                    onclick: move |_| {
                                        let blah = 120;
                                                    true
                                    },

                                    div {
                                        div {
                                            "hi"
                                        }
                                        h2 {
                            class: "asd",
                                        }
                                    }
            }
        }
    "#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn parse_comment() {
    let block = r#"
    div {
        adsasd: "asd", // this is a comment
    }
        "#;

    let parsed: TokenStream2 = syn::parse_str(block).unwrap();

    dbg!(parsed);
}

#[test]
fn formats_component() {
    let block = r#"
    Component {
        adsasd: "asd", // this is a comment
        onclick: move |_| {
            let blah = 120;
            let blah = 120;
        },
    }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_element() {
    let block = r#"
    div {
        a: "1234567891012345678910123456789101234567891012345678910123456789101234567891012345678910123456789101234567891012345678910",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
    }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_element_short() {
    let block = r#"
    div {
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
        a: "123",
    }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_element_nested() {
    let block = r#"
    h3 {
        class: "mb-2 text-xl font-bold",
        "Invite Member"
    }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_element_nested_no_trailing_tabs() {
    let block = r#"
    img { class: "mb-6 mx-auto h-24", src: "artemis-assets/images/friends.png", alt: "",                }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_element_with_correct_indent() {
    let block = r###"
    div {

                        a { class: "py-2 px-3 bg-indigo-500 hover:bg-indigo-600 rounded text-xs text-white", href: "#",
                            "Send invitation"
          }
    }

"###;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn small_elements_and_text_are_small() {
    let block = r###"
                        a { class: " text-white",
                            "Send invitation"
          }
"###;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_component_man_props() {
    let block = r#"
    Component {
        ..MyProps {
            val: 123
        },
        adsasd: "asd", // this is a comment
        onclick: move |_| {
            let blah = 120;
        },
    }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_component_tiny() {
    let block = r#"
    Component { a: 123
    }
"#;

    let formatted = fmt_block(block).unwrap();

    print!("{formatted}");
}

#[test]
fn formats_document() {
    let block = r#"
rsx!{
    Component {
        adsasd: "asd", // this is a comment
        onclick: move |_| {
            let blah = 120;
        },
    }
}


"#;

    let formatted = get_format_blocks(block);

    print!("{formatted:?}");
}

#[test]
fn component_path_mod_style() {
    let block = r#"
rsx!{
    my::thing::Component {
        adsasd: "asd", // this is a comment
        onclick: move |_| {
            let blah = 120;
        },
    }
}
"#;

    let formatted = get_format_blocks(block);

    print!("{formatted:?}");
}

#[test]
fn formats_valid_rust_src() {
    let src = r#"
//
rsx! {
    div {}
    div {
        h3 {"asd"
        }
    }
}
"#;

    let formatted = get_format_blocks(src);

    println!("{formatted:?}");
}

#[test]
fn formats_valid_rust_src_with_indents() {
    let mut src = r#"
#[inline_props]
fn NavItem<'a>(cx: Scope, to: &'static str, children: Element<'a>, icon: Shape) -> Element {
    const ICON_SIZE: u32 = 36;

    rsx! {
        div {

            h1 {"thing"}


        }




    }
}
"#
    .to_string();

    let formatted = get_format_blocks(&src);

    let block = formatted.into_iter().next().unwrap();

    src.replace_range(
        block.start - 1..block.end + 1,
        &format!("{{ {}    }}", &block.formatted),
    );
}

#[test]
fn formats_multiple_blocks() {
    let mut src = r#"
#[inline_props]
fn NavItem<'a>(cx: Scope, to: &'static str, children: Element<'a>, icon: Shape) -> Element {
    const ICON_SIZE: u32 = 36;

    rsx! {
        div {

            h1 {"thing"}


        }


    }

    rsx! {
        div {

            Ball {
                a: rsx!{
                    "asdasd"
                }
            }
        }
    }
}

#[inline_props]
fn NavItem<'a>(cx: Scope, to: &'static str, children: Element<'a>, icon: Shape) -> Element {
    const ICON_SIZE: u32 = 36;

    rsx! {
        div {

            h1 {"thing"}


        }


    }

    rsx! {
        div {

            Ball {
                a: rsx!{
                    "asdasd"
                }
            }
        }
    }
}
"#
    .to_string();

    let formatted = get_format_blocks(&src);

    dbg!(&formatted);

    let block = formatted.into_iter().next().unwrap();

    src.replace_range(
        block.start - 1..block.end + 1,
        &format!("{{ {}    }}", &block.formatted),
    );
}