pub mod harness; use harness::alice_run_until::is_xmr_lock_transaction_sent; use harness::bob_run_until::is_btc_locked; use harness::FastPunishConfig; use swap::protocol::alice::AliceState; use swap::protocol::bob::BobState; use swap::protocol::{alice, bob}; /// Bob locks Btc and Alice locks Xmr. Bob does not act; he fails to send Alice /// the encsig and fail to refund or redeem. Alice punishes. #[tokio::test] async fn alice_punishes_after_restart_if_punish_timelock_expired() { harness::setup_test(FastPunishConfig, |mut ctx| async move { let (bob_swap, bob_join_handle) = ctx.bob_swap().await; let bob_swap_id = bob_swap.id; let bob_swap = tokio::spawn(bob::run_until(bob_swap, is_btc_locked)); let alice_swap = ctx.alice_next_swap().await; let alice_bitcoin_wallet = alice_swap.bitcoin_wallet.clone(); let alice_swap = tokio::spawn(alice::run_until(alice_swap, is_xmr_lock_transaction_sent)); let bob_state = bob_swap.await??; assert!(matches!(bob_state, BobState::BtcLocked { .. })); let alice_state = alice_swap.await??; // Ensure punish timelock is expired if let AliceState::XmrLockTransactionSent { state3, .. } = alice_state { alice_bitcoin_wallet .subscribe_to(state3.tx_lock) .await .wait_until_confirmed_with(state3.punish_timelock) .await?; } else { panic!("Alice in unexpected state {}", alice_state); } ctx.restart_alice().await; let alice_swap = ctx.alice_next_swap().await; let alice_swap = tokio::spawn(alice::run(alice_swap)); let alice_state = alice_swap.await??; ctx.assert_alice_punished(alice_state).await; // Restart Bob after Alice punished to ensure Bob transitions to // punished and does not run indefinitely let (bob_swap, _) = ctx .stop_and_resume_bob_from_db(bob_join_handle, bob_swap_id) .await; assert!(matches!(bob_swap.state, BobState::BtcLocked { .. })); let bob_state = bob::run(bob_swap).await?; ctx.assert_bob_punished(bob_state).await; Ok(()) }) .await; }