Function expand_wrapper

src/ore-proc/src/test.rs:72–106

Emit code for a wrapper function around a test function.

(inner_test: &TokenStream2, wrappee: &ItemFn)

/// Emit code for a wrapper function around a test function.
fn expand_wrapper(inner_test: &TokenStream2, wrappee: &ItemFn) -> TokenStream {
    let attrs = &wrappee.attrs;
    let async_ = &wrappee.sig.asyncness;
    let await_ = if async_.is_some() {
        quote! {.await}
    } else {
        quote! {}
    };
    let body = &wrappee.block;
    let test_name = &wrappee.sig.ident;

    // Note that Rust does not allow us to have a test function with
    // #[should_panic] that has a non-unit return value.
    let ret = match &wrappee.sig.output {
        ReturnType::Default => quote! {},
        ReturnType::Type(_, type_) => quote! {-> #type_},
    };

    let logging_init = expand_logging_init();

    let result = quote! {
      #[#inner_test]
      #(#attrs)*
      #async_ fn #test_name() #ret {
        #async_ fn test_impl() #ret {
          #body
        }

        #logging_init

        test_impl()#await_
      }
    };
    result.into()
}