Git-Mirrors/xmr-btc-swap
1
0
Fork 0
mirror of https://github.com/comit-network/xmr-btc-swap.git synced 2024-12-26 07:59:28 -05:00
Code Issues Packages Projects Releases Wiki Activity

Rename Test to TestContext and argument to ctx

Browse Source
This commit is contained in:
Daniel Karzel 2021-01-20 10:40:40 +11:00
parent e91987e23f
commit acfd43ee79
7 changed files with 39 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
swap/tests/happy_path.rs
View File

@ -7,17 +7,17 @@ pub mod testutils;
#[tokio::test] #[tokio::test]
async fn happy_path() { async fn happy_path() {
testutils::setup_test(|test| async move { testutils::setup_test(|ctx| async move {
let alice_swap = test.new_swap_as_alice().await; let alice_swap = ctx.new_swap_as_alice().await;
let bob_swap = test.new_swap_as_bob().await; let bob_swap = ctx.new_swap_as_bob().await;
let alice = alice::run(alice_swap); let alice = alice::run(alice_swap);
let bob = bob::run(bob_swap); let bob = bob::run(bob_swap);
let (alice_state, bob_state) = join!(alice, bob); let (alice_state, bob_state) = join!(alice, bob);
test.assert_alice_redeemed(alice_state.unwrap()).await; ctx.assert_alice_redeemed(alice_state.unwrap()).await;
test.assert_bob_redeemed(bob_state.unwrap()).await; ctx.assert_bob_redeemed(bob_state.unwrap()).await;
}) })
.await; .await;
} }

12
swap/tests/happy_path_restart_alice.rs
View File

@ -4,9 +4,9 @@ pub mod testutils;
#[tokio::test] #[tokio::test]
async fn given_alice_restarts_after_encsig_is_learned_resume_swap() { async fn given_alice_restarts_after_encsig_is_learned_resume_swap() {
testutils::setup_test(|test| async move { testutils::setup_test(|ctx| async move {
let alice_swap = test.new_swap_as_alice().await; let alice_swap = ctx.new_swap_as_alice().await;
let bob_swap = test.new_swap_as_bob().await; let bob_swap = ctx.new_swap_as_bob().await;
let bob = bob::run(bob_swap); let bob = bob::run(bob_swap);
let bob_handle = tokio::spawn(bob); let bob_handle = tokio::spawn(bob);
@ -16,15 +16,15 @@ async fn given_alice_restarts_after_encsig_is_learned_resume_swap() {
.unwrap(); .unwrap();
assert!(matches!(alice_state, AliceState::EncSigLearned {..})); assert!(matches!(alice_state, AliceState::EncSigLearned {..}));
let alice_swap = test.recover_alice_from_db().await; let alice_swap = ctx.recover_alice_from_db().await;
assert!(matches!(alice_swap.state, AliceState::EncSigLearned {..})); assert!(matches!(alice_swap.state, AliceState::EncSigLearned {..}));
let alice_state = alice::run(alice_swap).await.unwrap(); let alice_state = alice::run(alice_swap).await.unwrap();
test.assert_alice_redeemed(alice_state).await; ctx.assert_alice_redeemed(alice_state).await;
let bob_state = bob_handle.await.unwrap(); let bob_state = bob_handle.await.unwrap();
test.assert_bob_redeemed(bob_state.unwrap()).await ctx.assert_bob_redeemed(bob_state.unwrap()).await
}) })
.await; .await;
} }

12
swap/tests/happy_path_restart_bob_after_comm.rs
View File

@ -4,9 +4,9 @@ pub mod testutils;
#[tokio::test] #[tokio::test]
async fn given_bob_restarts_after_encsig_is_sent_resume_swap() { async fn given_bob_restarts_after_encsig_is_sent_resume_swap() {
testutils::setup_test(|test| async move { testutils::setup_test(|ctx| async move {
let alice_swap = test.new_swap_as_alice().await; let alice_swap = ctx.new_swap_as_alice().await;
let bob_swap = test.new_swap_as_bob().await; let bob_swap = ctx.new_swap_as_bob().await;
let alice = alice::run(alice_swap); let alice = alice::run(alice_swap);
let alice_handle = tokio::spawn(alice); let alice_handle = tokio::spawn(alice);
@ -17,15 +17,15 @@ async fn given_bob_restarts_after_encsig_is_sent_resume_swap() {
assert!(matches!(bob_state, BobState::EncSigSent {..})); assert!(matches!(bob_state, BobState::EncSigSent {..}));
let bob_swap = test.recover_bob_from_db().await; let bob_swap = ctx.recover_bob_from_db().await;
assert!(matches!(bob_swap.state, BobState::EncSigSent {..})); assert!(matches!(bob_swap.state, BobState::EncSigSent {..}));
let bob_state = bob::run(bob_swap).await.unwrap(); let bob_state = bob::run(bob_swap).await.unwrap();
test.assert_bob_redeemed(bob_state).await; ctx.assert_bob_redeemed(bob_state).await;
let alice_state = alice_handle.await.unwrap(); let alice_state = alice_handle.await.unwrap();
test.assert_alice_redeemed(alice_state.unwrap()).await; ctx.assert_alice_redeemed(alice_state.unwrap()).await;
}) })
.await; .await;
} }

12
swap/tests/happy_path_restart_bob_before_comm.rs
View File

@ -7,9 +7,9 @@ pub mod testutils;
#[tokio::test] #[tokio::test]
async fn given_bob_restarts_after_xmr_is_locked_resume_swap() { async fn given_bob_restarts_after_xmr_is_locked_resume_swap() {
testutils::setup_test(|test| async move { testutils::setup_test(|ctx| async move {
let alice_swap = test.new_swap_as_alice().await; let alice_swap = ctx.new_swap_as_alice().await;
let bob_swap = test.new_swap_as_bob().await; let bob_swap = ctx.new_swap_as_bob().await;
let alice_handle = alice::run(alice_swap); let alice_handle = alice::run(alice_swap);
let alice_swap_handle = tokio::spawn(alice_handle); let alice_swap_handle = tokio::spawn(alice_handle);
@ -18,15 +18,15 @@ async fn given_bob_restarts_after_xmr_is_locked_resume_swap() {
assert!(matches!(bob_state, BobState::XmrLocked {..})); assert!(matches!(bob_state, BobState::XmrLocked {..}));
let bob_swap = test.recover_bob_from_db().await; let bob_swap = ctx.recover_bob_from_db().await;
assert!(matches!(bob_swap.state, BobState::XmrLocked {..})); assert!(matches!(bob_swap.state, BobState::XmrLocked {..}));
let bob_state = bob::run(bob_swap).await.unwrap(); let bob_state = bob::run(bob_swap).await.unwrap();
test.assert_bob_redeemed(bob_state).await; ctx.assert_bob_redeemed(bob_state).await;
let alice_state = alice_swap_handle.await.unwrap(); let alice_state = alice_swap_handle.await.unwrap();
test.assert_alice_redeemed(alice_state.unwrap()).await; ctx.assert_alice_redeemed(alice_state.unwrap()).await;
}) })
.await; .await;
} }

12
swap/tests/punish.rs
View File

@ -9,9 +9,9 @@ pub mod testutils;
/// the encsig and fail to refund or redeem. Alice punishes. /// the encsig and fail to refund or redeem. Alice punishes.
#[tokio::test] #[tokio::test]
async fn alice_punishes_if_bob_never_acts_after_fund() { async fn alice_punishes_if_bob_never_acts_after_fund() {
testutils::setup_test(|test| async move { testutils::setup_test(|ctx| async move {
let alice_swap = test.new_swap_as_alice().await; let alice_swap = ctx.new_swap_as_alice().await;
let bob_swap = test.new_swap_as_bob().await; let bob_swap = ctx.new_swap_as_bob().await;
let alice = alice::run(alice_swap); let alice = alice::run(alice_swap);
let alice_handle = tokio::spawn(alice); let alice_handle = tokio::spawn(alice);
@ -21,16 +21,16 @@ async fn alice_punishes_if_bob_never_acts_after_fund() {
assert!(matches!(bob_state, BobState::BtcLocked {..})); assert!(matches!(bob_state, BobState::BtcLocked {..}));
let alice_state = alice_handle.await.unwrap(); let alice_state = alice_handle.await.unwrap();
test.assert_alice_punished(alice_state.unwrap()).await; ctx.assert_alice_punished(alice_state.unwrap()).await;
// Restart Bob after Alice punished to ensure Bob transitions to // Restart Bob after Alice punished to ensure Bob transitions to
// punished and does not run indefinitely // punished and does not run indefinitely
let bob_swap = test.recover_bob_from_db().await; let bob_swap = ctx.recover_bob_from_db().await;
assert!(matches!(bob_swap.state, BobState::BtcLocked {..})); assert!(matches!(bob_swap.state, BobState::BtcLocked {..}));
let bob_state = bob::run(bob_swap).await.unwrap(); let bob_state = bob::run(bob_swap).await.unwrap();
test.assert_bob_punished(bob_state).await; ctx.assert_bob_punished(bob_state).await;
}) })
.await; .await;
} }

12
swap/tests/refund_restart_alice.rs
View File

@ -6,9 +6,9 @@ pub mod testutils;
/// then also refunds. /// then also refunds.
#[tokio::test] #[tokio::test]
async fn given_alice_restarts_after_xmr_is_locked_abort_swap() { async fn given_alice_restarts_after_xmr_is_locked_abort_swap() {
testutils::setup_test(|test| async move { testutils::setup_test(|ctx| async move {
let alice_swap = test.new_swap_as_alice().await; let alice_swap = ctx.new_swap_as_alice().await;
let bob_swap = test.new_swap_as_bob().await; let bob_swap = ctx.new_swap_as_bob().await;
let bob = bob::run(bob_swap); let bob = bob::run(bob_swap);
let bob_handle = tokio::spawn(bob); let bob_handle = tokio::spawn(bob);
@ -22,7 +22,7 @@ async fn given_alice_restarts_after_xmr_is_locked_abort_swap() {
let bob_state = bob_handle.await.unwrap(); let bob_state = bob_handle.await.unwrap();
// Once bob has finished Alice is restarted and refunds as well // Once bob has finished Alice is restarted and refunds as well
let alice_swap = test.recover_alice_from_db().await; let alice_swap = ctx.recover_alice_from_db().await;
assert!(matches!(alice_swap.state, AliceState::XmrLocked {..})); assert!(matches!(alice_swap.state, AliceState::XmrLocked {..}));
let alice_state = alice::run(alice_swap).await.unwrap(); let alice_state = alice::run(alice_swap).await.unwrap();
@ -31,8 +31,8 @@ async fn given_alice_restarts_after_xmr_is_locked_abort_swap() {
// refunded, not at the end because this can cause side-effects! // refunded, not at the end because this can cause side-effects!
// We have to properly wait for the refund tx's finality inside the assertion, // We have to properly wait for the refund tx's finality inside the assertion,
// which requires storing the refund_tx_id in the the state! // which requires storing the refund_tx_id in the the state!
test.assert_bob_refunded(bob_state.unwrap()).await; ctx.assert_bob_refunded(bob_state.unwrap()).await;
test.assert_alice_refunded(alice_state).await; ctx.assert_alice_refunded(alice_state).await;
}) })
.await; .await;
} }

8
swap/tests/testutils/mod.rs
View File

@ -19,14 +19,14 @@ use tracing_core::dispatcher::DefaultGuard;
use tracing_log::LogTracer; use tracing_log::LogTracer;
use uuid::Uuid; use uuid::Uuid;
pub struct Test { pub struct TestContext {
swap_amounts: SwapAmounts, swap_amounts: SwapAmounts,
alice_swap_factory: alice::SwapFactory, alice_swap_factory: alice::SwapFactory,
bob_swap_factory: bob::SwapFactory, bob_swap_factory: bob::SwapFactory,
} }
impl Test { impl TestContext {
pub async fn new_swap_as_alice(&self) -> alice::Swap { pub async fn new_swap_as_alice(&self) -> alice::Swap {
let (swap, mut event_loop) = self let (swap, mut event_loop) = self
.alice_swap_factory .alice_swap_factory
@ -303,7 +303,7 @@ impl Test {
pub async fn setup_test<T, F>(testfn: T) pub async fn setup_test<T, F>(testfn: T)
where where
T: Fn(Test) -> F, T: Fn(TestContext) -> F,
F: Future<Output = ()>, F: Future<Output = ()>,
{ {
let cli = Cli::default(); let cli = Cli::default();
@ -377,7 +377,7 @@ where
alice_swap_factory.peer_id(), alice_swap_factory.peer_id(),
); );
let test = Test { let test = TestContext {
swap_amounts, swap_amounts,
alice_swap_factory, alice_swap_factory,
bob_swap_factory, bob_swap_factory,