Add cloned mock to fix flakey test

pool2win · Dec 3, 2024 · 8c202c6 · 8c202c6
1 parent f4e5cf2
commit 8c202c6
Showing 1 changed file with 14 additions and 4 deletions.
diff --git 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
         });