Add mtx call for start_context tests

pool2win · Dec 3, 2024 · 47f7141 · 47f7141
1 parent 8c202c6
commit 47f7141
Showing 1 changed file with 7 additions and 14 deletions.
diff --git a/src/node.rs b/src/node.rs
@@ -443,13 +443,8 @@ mod node_tests {
         let ctx = EchoBroadcastHandle::start_context();
         ctx.expect().returning(|| {
             let mut mock = EchoBroadcastHandle::default();
-            mock.expect_clone().returning(|| {
-                let mut cloned = EchoBroadcastHandle::default();
-                cloned
-                    .expect_clone()
-                    .returning(|| EchoBroadcastHandle::default());
-                cloned
-            });
+            mock.expect_clone()
+                .returning(|| EchoBroadcastHandle::default());
             mock
         });
 
@@ -492,13 +487,8 @@ mod node_tests {
         let ctx = EchoBroadcastHandle::start_context();
         ctx.expect().returning(|| {
             let mut mock = EchoBroadcastHandle::default();
-            mock.expect_clone().returning(|| {
-                let mut cloned = EchoBroadcastHandle::default();
-                cloned
-                    .expect_clone()
-                    .returning(|| EchoBroadcastHandle::default());
-                cloned
-            });
+            mock.expect_clone()
+                .returning(|| EchoBroadcastHandle::default());
             mock
         });
 
@@ -508,6 +498,7 @@ mod node_tests {
 
     #[tokio::test]
     async fn it_should_respond_to_unicast_messages() {
+        let _m = MTX.lock();
         let ctx = EchoBroadcastHandle::start_context();
         ctx.expect().returning(|| {
             let mut mock = EchoBroadcastHandle::default();
@@ -543,6 +534,8 @@ mod node_tests {
 
     #[tokio::test]
     async fn it_should_respond_to_broadcast_messages() {
+        let _m = MTX.lock();
+
         let ctx = EchoBroadcastHandle::start_context();
         ctx.expect().returning(EchoBroadcastHandle::default);