mry_macros 0.14.0

Macro crate for mry, a simple but powerful mocking library that supports struct, trait, and function.
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

use proc_macro2::{Span, TokenStream};
use quote::quote;
use syn::{Ident, Index};

use crate::MAX_ARGUMENT_COUNT;

pub(crate) fn create() -> TokenStream {
    let items = (0..=MAX_ARGUMENT_COUNT).map(|nargs| {
        let (args, types): (Vec<_>, Vec<_>) = (1..=nargs)
            .map(|i| {
                let name = format!("Arg{i}");
                (
                    Ident::new(&name.to_lowercase(), Span::call_site()),
                    Ident::new(&name, Span::call_site()),
                )
            })
            .unzip();
        let matchers: Vec<_> = types.iter().map(|ty| quote![ArgMatcher<#ty>]).collect();
        let trait_bounds: Vec<_> = types.iter().map(|ty| quote![#ty: Send + 'static]).collect();
        let matchers = quote![#(#matchers,)*];
        let matches = args.iter().enumerate().map(|(index, arg)| {
            let index = Index::from(index);
            quote![self.#index.matches(#arg)]
        });
        let args = quote![#(#args,)*];
        quote! {
            impl<#(#trait_bounds),*> Match<(#(#types,)*)> for (#matchers) {
                fn matches(&self, (#args): &(#(#types,)*)) -> bool {
                    #(#matches &&)* true
                }
            }

            impl<#(#trait_bounds),*> From<(#matchers)> for Matcher<(#(#types,)*)> {
                fn from((#args): (#matchers)) -> Self {
                    Matcher(Box::new((#args)))
                }
            }
        }
    });
    quote![#(#items)*]
}