untest_macro 0.1.1

Macros for untest crate
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use proc_macro::TokenStream;
use quote::{format_ident, quote};
use syn::{parse_macro_input, ItemFn};

#[proc_macro_attribute]
pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let input = &parse_macro_input!(item as ItemFn);

    match generate_test(input) {
        Ok(generated) => generated,
        Err(err) => err.to_compile_error().into(),
    }
}

fn generate_test(ast: &ItemFn) -> Result<TokenStream, syn::Error> {
    let ident = &ast.sig.ident;
    let gen_pub = match ast.vis {
        syn::Visibility::Inherited => Some(quote! { pub }),
        _ => None,
    };

    let mod_name = format_ident!("untest_generated_test_{}", ident);

    let gen = quote! {
        //extern crate test;
        #[allow(non_upper_case_globals)]
        pub const #ident: test::TestDescAndFn = test::TestDescAndFn {
          desc: test::TestDesc {
            name: test::StaticTestName(stringify!(#ident)),
            ignore: false,
            ignore_message: ::core::option::Option::None,
            compile_fail: false,
            no_run: false,
            should_panic: test::ShouldPanic::No,
            test_type: test::TestType::IntegrationTest,
          },
          testfn: test::StaticTestFn(|| test::assert_test_result(
            #mod_name::#ident()
          )),
        };
        mod #mod_name {
          use super::*;
          #gen_pub #ast
        }
    };

    Ok(gen.into())
}