diff options
| author | HampusM <hampus@hampusmat.com> | 2023-03-19 16:29:23 +0100 |
|---|---|---|
| committer | HampusM <hampus@hampusmat.com> | 2023-03-19 16:54:57 +0100 |
| commit | ad2a6d1dc517407939ed022bba9f3352efc678ce (patch) | |
| tree | 8bf7acc6cd79f2d686865832a3b44adb22c01d60 /macros/src/mock.rs | |
| parent | 657673f4a25a2a7299d3751d54d9597635bc529d (diff) | |
feat: add call count expectations to expectations
Diffstat (limited to 'macros/src/mock.rs')
| -rw-r--r-- | macros/src/mock.rs | 10 |
1 files changed, 1 insertions, 9 deletions
diff --git a/macros/src/mock.rs b/macros/src/mock.rs index 8828b17..d2eb451 100644 --- a/macros/src/mock.rs +++ b/macros/src/mock.rs @@ -247,15 +247,7 @@ fn create_mock_function( )) .with_generic_params::<#(#type_param_idents,)*>(); - let Some(returning) = &expectation.returning else { - panic!(concat!( - "Expectation for function", - stringify!(#func_ident), - " is missing a function to call") - ); - }; - - returning(#(#args),*) + expectation.call_returning(#(#args),*) } }) .unwrap_or_abort(), |