From 8c202c6d43963255e25440f5c42f0ca9be5993fd Mon Sep 17 00:00:00 2001
From: pool2win <pool2win@users.noreply.github.com>
Date: Tue, 3 Dec 2024 12:42:45 +0100
Subject: [PATCH] Add cloned mock to fix flakey test

---
 src/node.rs | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/node.rs b/src/node.rs
index 13fd2de..d079938 100644
--- a/src/node.rs
+++ b/src/node.rs
@@ -443,8 +443,13 @@ mod node_tests {
         let ctx = EchoBroadcastHandle::start_context();
         ctx.expect().returning(|| {
             let mut mock = EchoBroadcastHandle::default();
-            mock.expect_clone()
-                .returning(|| EchoBroadcastHandle::default());
+            mock.expect_clone().returning(|| {
+                let mut cloned = EchoBroadcastHandle::default();
+                cloned
+                    .expect_clone()
+                    .returning(|| EchoBroadcastHandle::default());
+                cloned
+            });
             mock
         });
 
@@ -487,8 +492,13 @@ mod node_tests {
         let ctx = EchoBroadcastHandle::start_context();
         ctx.expect().returning(|| {
             let mut mock = EchoBroadcastHandle::default();
-            mock.expect_clone()
-                .returning(|| EchoBroadcastHandle::default());
+            mock.expect_clone().returning(|| {
+                let mut cloned = EchoBroadcastHandle::default();
+                cloned
+                    .expect_clone()
+                    .returning(|| EchoBroadcastHandle::default());
+                cloned
+            });
             mock
         });