From 1c6e757f02488c8e90387f721d1bd3f79ef28e93 Mon Sep 17 00:00:00 2001
From: patrini32 <171664803+patrini32@users.noreply.github.com>
Date: Wed, 5 Jun 2024 15:34:20 +0300
Subject: [PATCH] Use assert_bob_punished

---
 .../alice_manually_punishes_after_bob_dead_and_bob_cancels.rs   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swap/tests/alice_manually_punishes_after_bob_dead_and_bob_cancels.rs b/swap/tests/alice_manually_punishes_after_bob_dead_and_bob_cancels.rs
index 5eacefd1..84b1ef7f 100644
--- a/swap/tests/alice_manually_punishes_after_bob_dead_and_bob_cancels.rs
+++ b/swap/tests/alice_manually_punishes_after_bob_dead_and_bob_cancels.rs
@@ -76,7 +76,7 @@ async fn alice_manually_punishes_after_bob_dead_and_bob_cancels() {
 
         let state =
             cli::cancel_and_refund(bob_swap_id, bob_swap.bitcoin_wallet, bob_swap.db).await?;
-        assert!(matches!(state, BobState::BtcPunished { .. }));
+        ctx.assert_bob_punished(state).await;
         Ok(())
     })
     .await;