use genco::prelude::*;

#[test]
fn test_quote() -> genco::fmt::Result {
    let test = quoted("one");

    let tokens: rust::Tokens = quote! {
        fn test() -> u32 {
            println!("{}", $(test));

            42
        }
    };

    assert_eq!(
        "fn test() -> u32 {\n    println!(\"{}\", \"one\");\n\n    42\n}",
        tokens.to_string()?
    );

    let tokens: rust::Tokens = quote! {
        fn test() -> u32 {
            println!("{}", $(quoted("two")));

            42
        }
    };

    assert_eq!(
        "fn test() -> u32 {\n    println!(\"{}\", \"two\");\n\n    42\n}",
        tokens.to_string()?
    );

    let tokens: rust::Tokens = quote! {
        fn test() -> u32 {
            println!("{}", $$(quoted("two")));

            42
        }
    };

    assert_eq!(
        "fn test() -> u32 {\n    println!(\"{}\", $(quoted(\"two\")));\n\n    42\n}",
        tokens.to_string()?
    );

    Ok(())
}

#[test]
fn test_tight_quote() -> genco::fmt::Result {
    let a = "foo";
    let b = "bar";
    let c = "baz";
    let tokens: rust::Tokens = quote!($(a)$(b)$(c));
    assert_eq!("foobarbaz", tokens.to_string()?);
    Ok(())
}

#[test]
fn test_escape() -> genco::fmt::Result {
    let tokens: rust::Tokens = quote!($$$$ $$ $$$$ $$$$ $$ $$ $$[test]);
    assert_eq!("$$ $ $$ $$ $ $ $[test]", tokens.to_string()?);

    Ok(())
}

#[test]
fn test_scope() -> genco::fmt::Result {
    let tokens: rust::Tokens = quote! {
        // Nested factory.
        $(ref tokens {
            quote_in!(*tokens => fn test() -> u32 { 42 });
        })
    };

    assert_eq!("fn test() -> u32 { 42 }", tokens.to_string()?);

    Ok(())
}