test: refactor and use channels

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

tap-agent/src/agent/sender_accounts_manager.rs

@@ -598,8 +598,8 @@ mod tests {
     use sqlx::postgres::PgListener;
     use sqlx::PgPool;
     use std::collections::{HashMap, HashSet};
-    use std::sync::{Arc, Mutex};
     use std::time::Duration;
+    use tokio::sync::mpsc;
 
     const DUMMY_URL: &str = "http://localhost:1234";
 
@@ -931,12 +931,12 @@ mod tests {
             PREFIX_ID.fetch_add(1, std::sync::atomic::Ordering::SeqCst)
         );
 
-        let last_message_emitted = Arc::new(Mutex::new(vec![]));
+        let (last_message_emitted, mut rx) = mpsc::channel(64);
 
         let (sender_account, join_handle) = MockSenderAccount::spawn(
             Some(format!("{}:{}", prefix.clone(), SENDER.1,)),
             MockSenderAccount {
-                last_message_emitted: last_message_emitted.clone(),
+                last_message_emitted,
             },
             (),
         )
@@ -958,8 +958,8 @@ mod tests {
         tokio::time::sleep(Duration::from_millis(10)).await;
 
         assert_eq!(
-            last_message_emitted.lock().unwrap().last().unwrap(),
-            &SenderAccountMessage::NewAllocationId(*ALLOCATION_ID_0)
+            rx.recv().await.unwrap(),
+            SenderAccountMessage::NewAllocationId(*ALLOCATION_ID_0)
         );
         sender_account.stop_and_wait(None, None).await.unwrap();
         join_handle.await.unwrap();