feat: parse impl in mock macro as actual impl block

1 files changed, 6 insertions, 10 deletions

diff --git a/macros/src/mock.rs b/macros/src/mock.rs
index 88d3434..1b6dd67 100644
--- a/macros/src/mock.rs
+++ b/macros/src/mock.rs
@@ -16,7 +16,6 @@ use syn::{
     ReturnType,
     Signature,
     Token,
-    TraitItemMethod,
     Type,
     TypePath,
     TypeReference,
@@ -43,18 +42,15 @@ use crate::util::create_unit_type_tuple;
 pub struct Mock
 {
     ident: Ident,
-    mocked_trait: TypePath,
+    mocked_trait: Path,
     expectations_fields: Vec<ExpectationsField>,
-    item_methods: Vec<TraitItemMethod>,
+    item_methods: Vec<ImplItemMethod>,
 }
 
 impl Mock
 {
-    pub fn new(
-        ident: Ident,
-        mocked_trait: TypePath,
-        item_methods: &[TraitItemMethod],
-    ) -> Self
+    pub fn new(ident: Ident, mocked_trait: Path, item_methods: &[ImplItemMethod])
+        -> Self
     {
         let expectations_fields = item_methods
             .iter()
@@ -179,7 +175,7 @@ impl ToTokens for ExpectationsField
     }
 }
 
-fn create_mock_function(item_method: TraitItemMethod) -> ImplItemMethod
+fn create_mock_function(item_method: ImplItemMethod) -> ImplItemMethod
 {
     let func_ident = &item_method.sig.ident;
 
@@ -247,7 +243,7 @@ fn create_mock_function(item_method: TraitItemMethod) -> ImplItemMethod
     }
 }
 
-fn create_expect_function(mock: &Ident, item_method: &TraitItemMethod) -> ImplItemMethod
+fn create_expect_function(mock: &Ident, item_method: &ImplItemMethod) -> ImplItemMethod
 {
     let signature = Signature::new(
         format_ident!("expect_{}", item_method.sig.ident),