test: use channel response to test received message

Author: TypeLevelConsoli

crates/tap-agent/src/agent/sender_account.rs

@@ -1564,7 +1564,11 @@ pub mod tests {
         sender_account
             .cast(SenderAccountMessage::UpdateAllocationIds(HashSet::new()))
             .unwrap();
-        msg_receiver.recv().await.expect("Channel failed");
+        let msg = msg_receiver.recv().await.expect("Channel failed");
+        assert_eq!(
+            msg,
+            SenderAccountMessage::UpdateAllocationIds(HashSet::new())
+        );
 
         let actor_ref = ActorRef::<SenderAllocationMessage>::where_is(sender_allocation_id.clone());
         assert!(actor_ref.is_none());
@@ -2366,7 +2370,14 @@ pub mod tests {
                 ReceiptFees::NewReceipt(TRIGGER_VALUE, get_current_timestamp_u64_ns()),
             ))
             .unwrap();
-        msg_receiver.recv().await.expect("Channel failed");
+        let msg = msg_receiver.recv().await.expect("Channel failed");
+        assert!(matches!(
+            msg,
+            SenderAccountMessage::UpdateReceiptFees(
+                ALLOCATION_ID_0,
+                ReceiptFees::NewReceipt(TRIGGER_VALUE, _)
+            )
+        ));
 
         let deny = call!(sender_account, SenderAccountMessage::GetDeny).unwrap();
         assert!(deny, "should be blocked");

crates/tap-agent/tests/sender_account_test.rs

@@ -48,14 +48,17 @@ async fn sender_account_layer_test(pgpool: PgPool) {
         .await;
 
     // we expect it to create a sender allocation
+    let allocation_ids = HashSet::from_iter([AllocationId::Legacy(ALLOCATION_ID_0)]);
     sender_account
         .cast(SenderAccountMessage::UpdateAllocationIds(
-            vec![AllocationId::Legacy(ALLOCATION_ID_0)]
-                .into_iter()
-                .collect(),
+            allocation_ids.clone(),
         ))
         .unwrap();
-    msg_receiver.recv().await.expect("Channel failed");
+    let msg = msg_receiver.recv().await.expect("Channel failed");
+    assert_eq!(
+        msg,
+        SenderAccountMessage::UpdateAllocationIds(allocation_ids)
+    );
 
     mock_server
         .register(