use crate::testutils::{init_alice, init_bob}; use futures::future::try_join; use get_port::get_port; use libp2p::Multiaddr; use rand::rngs::OsRng; use swap::{ bitcoin, config::Config, database::Database, monero, protocol::{alice, alice::AliceState, bob, bob::BobState}, }; use tempfile::tempdir; use testcontainers::clients::Cli; use testutils::init_tracing; use tokio::select; use uuid::Uuid; pub mod testutils; // Bob locks btc and Alice locks xmr. Alice fails to act so Bob refunds. Alice // then also refunds. #[tokio::test] async fn given_alice_restarts_after_xmr_is_locked_abort_swap() { let _guard = init_tracing(); let cli = Cli::default(); let ( monero, testutils::Containers { bitcoind, monerods: _monerods, }, ) = testutils::init_containers(&cli).await; let btc_to_swap = bitcoin::Amount::from_sat(1_000_000); let xmr_to_swap = monero::Amount::from_piconero(1_000_000_000_000); let bob_btc_starting_balance = btc_to_swap * 10; let bob_xmr_starting_balance = monero::Amount::from_piconero(0); let alice_btc_starting_balance = bitcoin::Amount::ZERO; let alice_xmr_starting_balance = xmr_to_swap * 10; let port = get_port().expect("Failed to find a free port"); let alice_multiaddr: Multiaddr = format!("/ip4/127.0.0.1/tcp/{}", port) .parse() .expect("failed to parse Alice's address"); let ( alice_state, mut alice_event_loop_1, alice_event_loop_handle_1, alice_btc_wallet, alice_xmr_wallet, _, ) = init_alice( &bitcoind, &monero, btc_to_swap, xmr_to_swap, alice_xmr_starting_balance, alice_multiaddr.clone(), Config::regtest(), ) .await; let (bob_state, bob_event_loop, bob_event_loop_handle, bob_btc_wallet, bob_xmr_wallet, bob_db) = init_bob( alice_multiaddr.clone(), alice_event_loop_1.peer_id(), &bitcoind, &monero, btc_to_swap, bob_btc_starting_balance, xmr_to_swap, Config::regtest(), ) .await; let bob_fut = bob::swap::swap( bob_state, bob_event_loop_handle, bob_db, bob_btc_wallet.clone(), bob_xmr_wallet.clone(), OsRng, Uuid::new_v4(), ); let alice_swap_id = Uuid::new_v4(); let alice_db_datadir = tempdir().unwrap(); let alice_xmr_locked_fut = { let alice_db = Database::open(alice_db_datadir.path()).unwrap(); alice::swap::run_until( alice_state, alice::swap::is_xmr_locked, alice_event_loop_handle_1, alice_btc_wallet.clone(), alice_xmr_wallet.clone(), Config::regtest(), alice_swap_id, alice_db, ) }; tokio::spawn(bob_event_loop.run()); // We are selecting with alice_event_loop_1 so that we stop polling on it once // the try_join is finished. let (bob_state, alice_restart_state) = select! { res = try_join(bob_fut, alice_xmr_locked_fut) => res.unwrap(), _ = alice_event_loop_1.run() => panic!("The event loop should never finish") }; let tx_lock_id = if let BobState::BtcRefunded(state4) = bob_state { state4.tx_lock_id() } else { panic!("Bob in unexpected state"); }; let (mut alice_event_loop_2, alice_event_loop_handle_2) = testutils::init_alice_event_loop(alice_multiaddr); let alice_final_state = { let alice_db = Database::open(alice_db_datadir.path()).unwrap(); alice::swap::swap( alice_restart_state, alice_event_loop_handle_2, alice_btc_wallet.clone(), alice_xmr_wallet.clone(), Config::regtest(), alice_swap_id, alice_db, ) .await .unwrap() }; tokio::spawn(async move { alice_event_loop_2.run().await }); assert!(matches!(alice_final_state, AliceState::XmrRefunded)); let btc_alice_final = alice_btc_wallet.as_ref().balance().await.unwrap(); let btc_bob_final = bob_btc_wallet.as_ref().balance().await.unwrap(); let lock_tx_bitcoin_fee = bob_btc_wallet.transaction_fee(tx_lock_id).await.unwrap(); assert_eq!(btc_alice_final, alice_btc_starting_balance); // Alice or Bob could publish TxCancel. This means Bob could pay tx fees for // TxCancel and TxRefund or only TxRefund let btc_bob_final_alice_submitted_cancel = btc_bob_final == bob_btc_starting_balance - lock_tx_bitcoin_fee - bitcoin::Amount::from_sat(bitcoin::TX_FEE); let btc_bob_final_bob_submitted_cancel = btc_bob_final == bob_btc_starting_balance - lock_tx_bitcoin_fee - bitcoin::Amount::from_sat(2 * bitcoin::TX_FEE); assert!(btc_bob_final_alice_submitted_cancel || btc_bob_final_bob_submitted_cancel); alice_xmr_wallet.as_ref().inner.refresh().await.unwrap(); let xmr_alice_final = alice_xmr_wallet.as_ref().get_balance().await.unwrap(); assert_eq!(xmr_alice_final, xmr_to_swap); bob_xmr_wallet.as_ref().inner.refresh().await.unwrap(); let xmr_bob_final = bob_xmr_wallet.as_ref().get_balance().await.unwrap(); assert_eq!(xmr_bob_final, bob_xmr_starting_balance); }