replace self in match arms

Signed-off-by: Gregory Hill <gregorydhill@outlook.com>
BGluth · Jan 20, 2021 · 4529a3e · 4529a3e
1 parent 7cf550c
commit 4529a3e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "mocktopus"
-version = "0.7.9"
+version = "0.7.10"
 authors = [
     "CodeSandwich <igor.zuk@protonmail.com>",
     "gregdhill <gregorydhill@outlook.com>"
@@ -21,7 +21,7 @@ travis-ci = { repository = "CodeSandwich/Mocktopus" }
 doctest = false
 
 [dependencies]
-mocktopus_macros = "0.7.9"
+mocktopus_macros = "0.7.10"
 
 [dev-dependencies]
 tokio = { version = "0.2", features = ["full"] }

diff --git a/macros/Cargo.toml b/macros/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "mocktopus_macros"
-version = "0.7.9"
+version = "0.7.10"
 authors = [
     "CodeSandwich <igor.zuk@protonmail.com>",
     "gregdhill <gregorydhill@outlook.com>"

diff --git a/macros/src/item_injector.rs b/macros/src/item_injector.rs
@@ -486,6 +486,9 @@ fn replace_self_in_expr(expr: &mut syn::Expr) {
         }
         syn::Expr::Match(expr) => {
             replace_self_in_expr(&mut expr.expr);
+            for arm in &mut expr.arms {
+                replace_self_in_expr(&mut arm.body);
+            }
         }
         syn::Expr::MethodCall(expr) => {
             replace_self_in_expr(&mut expr.receiver);