@@ -4,6 +4,11 @@ use std::sync::{Arc, Mutex, Barrier};
44use std:: thread;
55use crate :: core:: testutil:: fixtures:: random_identifier;
66use crate :: network:: mock:: hub:: NetworkHub ;
7+ use crate :: network:: Payload :: TestMessage ;
8+ use crate :: network:: { Message , MessageProcessor , Network } ;
9+ use std:: collections:: HashSet ;
10+ use std:: sync:: { Arc , Barrier , Mutex } ;
11+ use std:: thread;
712
813#[ derive( Debug ) ]
914struct MockMessageProcessor {
@@ -125,9 +130,8 @@ fn test_concurrent_message_sending() {
125130 let mock_net_2 = NetworkHub :: new_mock_network ( hub, id_2) . unwrap ( ) ;
126131
127132 // Create 10 different message contents
128- let message_contents: Vec < String > = ( 0 ..10 )
129- . map ( |i| format ! ( "Concurrent message {i}" ) )
130- . collect ( ) ;
133+ let message_contents: Vec < String > =
134+ ( 0 ..10 ) . map ( |i| format ! ( "Concurrent message {i}" ) ) . collect ( ) ;
131135
132136 // Set up a barrier to synchronize all threads
133137 let barrier = Arc :: new ( Barrier :: new ( 10 ) ) ;
@@ -142,7 +146,7 @@ fn test_concurrent_message_sending() {
142146
143147 let handle = thread:: spawn ( move || {
144148 let message = Message {
145- payload : Payload :: TestMessage ( content) ,
149+ payload : TestMessage ( content) ,
146150 target_node_id : id_1_copy,
147151 } ;
148152
@@ -165,7 +169,10 @@ fn test_concurrent_message_sending() {
165169 // Verify that all messages were received
166170 let processor = msg_proc_1. lock ( ) . unwrap ( ) ;
167171 for content in message_contents {
168- assert ! ( processor. has_seen( & content) , "Message '{content}' was not received" ) ;
172+ assert ! (
173+ processor. has_seen( & content) ,
174+ "Message '{content}' was not received"
175+ ) ;
169176 println ! ( "Message '{content}' was successfully processed" ) ;
170177 }
171178}
0 commit comments