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
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use super::*;
use darling::FromMeta;
use proc_macro::TokenStream;
use proc_macro2::{Ident, Span};
use quote::quote;
use syn::{parse_macro_input, AttributeArgs, ItemFn};

pub fn relation_link_test_impl(attr: TokenStream, input: TokenStream) -> TokenStream {
    let attributes_meta: syn::AttributeArgs = parse_macro_input!(attr as AttributeArgs);
    let args = RelationLinkTestArgs::from_list(&attributes_meta);
    let args = match args {
        Ok(args) => args,
        Err(err) => return err.write_errors().into(),
    };

    if let Err(err) = args.validate(false) {
        return err.write_errors().into();
    };

    let only = args.only;
    let exclude = args.exclude;
    let id_only = args.id_only;
    let on_parent = args.on_parent.relation_field;
    let on_child = args.on_child.relation_field;

    let mut test_function = parse_macro_input!(input as ItemFn);
    if test_function.sig.inputs.len() != 2 {
        return syn::Error::new_spanned(
            test_function.sig,
            "connector test functions must take exactly two arguments: `runner: Runner, dm: &DatamodelWithParams`.",
        )
        .to_compile_error()
        .into();
    }

    if test_function.sig.asyncness.is_none() {
        return syn::Error::new_spanned(test_function.sig, "connector test functions must be async")
            .to_compile_error()
            .into();
    }

    // The shell function retains the name of the original test definition.
    let test_fn_ident = test_function.sig.ident;

    // Rename original test function to run_<orig_name>.
    let runner_fn_ident = Ident::new(&format!("run_{test_fn_ident}"), Span::call_site());
    test_function.sig.ident = runner_fn_ident.clone();

    // The test database name is the name used as the database for data source rendering.
    // Combination of test name and test mod name.
    let test_name = test_fn_ident.to_string();
    let suite_name = args.suite.expect("A test must have a test suite.");
    let required_capabilities = &args.capabilities.idents;

    let ts = quote! {
        #[test]
        fn #test_fn_ident() {
            query_tests_setup::run_relation_link_test(
                #on_parent,
                #on_child,
                #id_only,
                &[#only],
                &[#exclude],
                enumflags2::make_bitflags!(ConnectorCapability::{#(#required_capabilities)|*}),
                (#suite_name, #test_name),
                #runner_fn_ident
            )
        }

        #test_function
    };

    ts.into()
}