diff options
| -rw-r--r-- | examples/automock.rs | 31 | ||||
| -rw-r--r-- | macros/src/lib.rs | 138 | ||||
| -rw-r--r-- | src/lib.rs | 2 |
3 files changed, 138 insertions, 33 deletions
diff --git a/examples/automock.rs b/examples/automock.rs new file mode 100644 index 0000000..c8725ce --- /dev/null +++ b/examples/automock.rs @@ -0,0 +1,31 @@ +use cool_trais::Foo; +use ridicule::predicate::{always, eq}; + +use crate::cool_trais::MockFoo; + +mod cool_trais +{ + use ridicule::automock; + + #[automock] + pub trait Foo + { + fn bar<Something>(&self, num: u128, text: &str) -> Something; + } +} + +fn main() +{ + let mut mock_foo = MockFoo::new(); + + mock_foo + .expect_bar() + .with(eq(1234), always()) + .returning(|_, num, text| { + println!("bar was called with {num} and '{text}'"); + + 9.36f32 + }); + + mock_foo.bar::<f32>(1234, "Hello"); +} diff --git a/macros/src/lib.rs b/macros/src/lib.rs index 7fc062a..bcb4449 100644 --- a/macros/src/lib.rs +++ b/macros/src/lib.rs @@ -1,20 +1,27 @@ //! Macros for Ridicule, a mocking library supporting non-static generics. #![deny(clippy::all, clippy::pedantic, missing_docs)] use proc_macro::TokenStream; +use proc_macro2::Ident; use proc_macro_error::{proc_macro_error, ResultExt}; use quote::{format_ident, quote}; +use syn::token::Brace; use syn::{ parse, + Block, FnArg, GenericArgument, ImplItem, + ImplItemMethod, + ItemTrait, Path, PathArguments, PathSegment, ReturnType, + TraitItem, Type, TypeBareFn, TypeParamBound, + Visibility, }; use crate::expectation::Expectation; @@ -70,9 +77,104 @@ pub fn mock(input_stream: TokenStream) -> TokenStream let mock_mod_ident = format_ident!("__{mock_ident}"); - let method_items = input - .item_impl - .items + let method_items = + get_type_replaced_impl_item_methods(input.item_impl.items, &mock_ident); + + let mock = Mock::new( + mock_ident.clone(), + input.mocked_trait, + &method_items, + input.item_impl.generics.clone(), + ); + + let expectations = method_items.iter().map(|item_method| { + Expectation::new( + &mock_ident, + item_method, + input.item_impl.generics.params.clone(), + ) + }); + + quote! { + mod #mock_mod_ident { + use super::*; + + #mock + + #(#expectations)* + } + + use #mock_mod_ident::#mock_ident; + } + .into() +} + +/// Creates a mock automatically. +#[proc_macro_attribute] +#[proc_macro_error] +pub fn automock(_: TokenStream, input_stream: TokenStream) -> TokenStream +{ + let item_trait = parse::<ItemTrait>(input_stream).unwrap_or_abort(); + + let mock_ident = format_ident!("Mock{}", item_trait.ident); + + let mock_mod_ident = format_ident!("__{mock_ident}"); + + let method_items = get_type_replaced_impl_item_methods( + item_trait.items.iter().filter_map(|item| match item { + TraitItem::Method(item_method) => Some(ImplItem::Method(ImplItemMethod { + attrs: item_method.attrs.clone(), + vis: Visibility::Inherited, + defaultness: None, + sig: item_method.sig.clone(), + block: Block { + brace_token: Brace::default(), + stmts: vec![], + }, + })), + _ => None, + }), + &mock_ident, + ); + + let mock = Mock::new( + mock_ident.clone(), + Path::new( + WithLeadingColons::No, + [PathSegment::new(item_trait.ident.clone(), None)], + ), + &method_items, + item_trait.generics.clone(), + ); + + let expectations = method_items.iter().map(|item_method| { + Expectation::new(&mock_ident, item_method, item_trait.generics.params.clone()) + }); + + let visibility = &item_trait.vis; + + quote! { + #item_trait + + mod #mock_mod_ident { + use super::*; + + #mock + + #(#expectations)* + } + + #visibility use #mock_mod_ident::#mock_ident; + } + .into() +} + +fn get_type_replaced_impl_item_methods( + impl_items: impl IntoIterator<Item = ImplItem>, + mock_ident: &Ident, +) -> Vec<ImplItemMethod> +{ + impl_items .into_iter() .filter_map(|item| match item { ImplItem::Method(mut item_method) => { @@ -117,35 +219,7 @@ pub fn mock(input_stream: TokenStream) -> TokenStream } _ => None, }) - .collect::<Vec<_>>(); - - let mock = Mock::new( - mock_ident.clone(), - input.mocked_trait, - &method_items, - input.item_impl.generics.clone(), - ); - - let expectations = method_items.iter().map(|item_method| { - Expectation::new( - &mock_ident, - item_method, - input.item_impl.generics.params.clone(), - ) - }); - - quote! { - mod #mock_mod_ident { - use super::*; - - #mock - - #(#expectations)* - } - - use #mock_mod_ident::#mock_ident; - } - .into() + .collect() } fn replace_path_in_type(ty: Type, target_path: &Path, replacement_path: &Path) -> Type @@ -2,7 +2,7 @@ #![deny(clippy::all, clippy::pedantic, missing_docs)] pub use predicates::prelude::*; -pub use ridicule_macros::mock; +pub use ridicule_macros::{automock, mock}; #[doc(hidden)] pub mod __private |