Git-Mirrors/xmr-btc-swap
1
0
Fork 0
mirror of https://github.com/comit-network/xmr-btc-swap.git synced 2024-10-01 01:45:40 -04:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup resume_from_database functions only used in tests

Browse Source
This commit is contained in:
Daniel Karzel 2020-12-18 10:50:32 +11:00
parent e9681b6e2b
commit 773eebc723
4 changed files with 14 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
swap/src/alice/swap.rs
View File

@ -233,29 +233,6 @@ pub async fn swap(
.await
}
pub async fn resume_from_database(
event_loop_handle: EventLoopHandle,
bitcoin_wallet: Arc<crate::bitcoin::Wallet>,
monero_wallet: Arc<crate::monero::Wallet>,
config: Config,
swap_id: Uuid,
db: Database,
) -> Result<AliceState> {
let db_swap = db.get_state(swap_id)?;
let start_state = AliceState::try_from(db_swap)?;
let state = swap(
start_state,
event_loop_handle,
bitcoin_wallet,
monero_wallet,
config,
swap_id,
db,
)
.await?;
Ok(state)
}
pub fn is_complete(state: &AliceState) -> bool {
matches!(
state,

26
swap/src/bob/swap.rs
View File

@ -122,32 +122,6 @@ where
.await
}
pub async fn resume_from_database<R>(
event_loop_handle: EventLoopHandle,
db: Database,
bitcoin_wallet: Arc<crate::bitcoin::Wallet>,
monero_wallet: Arc<crate::monero::Wallet>,
rng: R,
swap_id: Uuid,
) -> Result<BobState>
where
R: RngCore + CryptoRng + Send,
{
let db_swap = db.get_state(swap_id)?;
let start_state = BobState::try_from(db_swap)?;
let state = swap(
start_state,
event_loop_handle,
db,
bitcoin_wallet,
monero_wallet,
rng,
swap_id,
)
.await?;
Ok(state)
}
pub fn is_complete(state: &BobState) -> bool {
matches!(
state,

7
swap/tests/happy_path_restart_alice.rs
View File

@ -9,6 +9,7 @@ use xmr_btc::config::Config;
pub mod testutils;
use crate::testutils::{init_alice, init_bob};
use std::convert::TryFrom;
use testutils::init_tracing;
#[tokio::test]
@ -114,7 +115,11 @@ async fn given_alice_restarts_after_encsig_is_learned_resume_swap() {
testutils::init_alice_event_loop(alice_multiaddr);
let _alice_swarm_fut = tokio::spawn(async move { event_loop_after_restart.run().await });
let alice_state = alice::swap::resume_from_database(
let db_swap = alice_db.get_state(alice_swap_id).unwrap();
let resume_state = AliceState::try_from(db_swap).unwrap();
let alice_state = alice::swap::swap(
resume_state,
event_loop_handle_after_restart,
alice_btc_wallet,
alice_xmr_wallet,

11
swap/tests/happy_path_restart_bob.rs
View File

@ -9,6 +9,7 @@ use xmr_btc::config::Config;
pub mod testutils;
use crate::testutils::{init_alice, init_bob};
use std::convert::TryFrom;
use swap::bob::swap::BobState;
use testutils::init_tracing;
@ -112,9 +113,13 @@ async fn given_bob_restarts_after_encsig_is_sent_resume_swap() {
let (event_loop_after_restart, event_loop_handle_after_restart) =
testutils::init_bob_event_loop();
let _alice_swarm_fut = tokio::spawn(async move { event_loop_after_restart.run().await });
let _bob_swarm_fut = tokio::spawn(async move { event_loop_after_restart.run().await });
let alice_state = bob::swap::resume_from_database(
let db_swap = bob_db.get_state(bob_swap_id).unwrap();
let resume_state = BobState::try_from(db_swap).unwrap();
let bob_state = bob::swap::swap(
resume_state,
event_loop_handle_after_restart,
bob_db,
bob_btc_wallet,
@ -125,7 +130,7 @@ async fn given_bob_restarts_after_encsig_is_sent_resume_swap() {
.await
.unwrap();
assert!(matches!(alice_state, BobState::XmrRedeemed {..}));
assert!(matches!(bob_state, BobState::XmrRedeemed {..}));
// TODO: Additionally assert balances
}