clickhouse-testing-macros 0.1.4

A crate that simplifies writing integration tests for ClickHouse, inspired by sqlx.
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
49
50
51
52
53
54
55
56
57
58
59
60

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

#[proc_macro_attribute]
pub fn test(_args: TokenStream, input: TokenStream) -> TokenStream {
    let func = parse_macro_input!(input as ItemFn);
    let func_name = &func.sig.ident;
    let func_name_str = func_name.to_string();
    let func_vis = &func.vis;
    let func_attrs = &func.attrs;
    let func_inputs = &func.sig.inputs;
    let func_output = &func.sig.output;
    let func_block = &func.block;

    let inner_name = syn::Ident::new(&format!("__{}_inner", func_name), func_name.span());

    let handle_result = match func_output {
        syn::ReturnType::Type(_, ty) if matches!(ty.as_ref(), syn::Type::Path(type_path) if type_path.path.segments.last().is_some_and(|seg| seg.ident == "Result")) =>
        {
            quote! {
                match test_result {
                    Ok(val) => {
                        if let Err(e) = clickhouse_testing::cleanup_test(&client).await {
                            panic!("Failed to cleanup test {:?} data: {:?}", #func_name_str, e);
                        }
                        val
                    }
                    Err(e) => panic!("Test {:?} failed: {:?}", #func_name_str, e),
                }
            }
        }
        _ => quote! {
            let val = test_result;
            if let Err(e) = clickhouse_testing::cleanup_test(&client).await {
                panic!("Failed to cleanup test {:?} data: {:?}", #func_name_str, e);
            }
            val
        },
    };

    let expanded = quote! {
        #(#func_attrs)*
        #[tokio::test]
        #func_vis async fn #func_name() {
            let module_path = module_path!();
            let client = match clickhouse_testing::init_test(module_path, #func_name_str).await {
                Ok(c) => c,
                Err(e) => panic!("Failed to setup test {:?} client: {:?}", #func_name_str, e),
            };

            let test_result = #inner_name(client.clone()).await;
            #handle_result
        }

        async fn #inner_name(#func_inputs) #func_output #func_block
    };

    expanded.into()
}