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

Rename ExecutionParams to EnvironmentConfig

Browse Source
This commit is contained in:
Thomas Eizinger 2021-03-17 14:55:42 +11:00
parent bc43ed6ebd
commit 09c41f89c4
No known key found for this signature in database
GPG Key ID: 651AC83A6C6C8B96
14 changed files with 132 additions and 136 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
swap/src/bin/asb.rs
View File

@ -24,13 +24,13 @@ use swap::asb::config::{
initial_setup, query_user_for_initial_testnet_config, read_config, Config, ConfigNotInitialized,
};
use swap::database::Database;
use swap::execution_params::{ExecutionParams, GetExecutionParams};
use swap::env::GetConfig;
use swap::fs::default_config_path;
use swap::monero::Amount;
use swap::protocol::alice::{run, EventLoop};
use swap::seed::Seed;
use swap::trace::init_tracing;
use swap::{bitcoin, execution_params, kraken, monero};
use swap::{bitcoin, env, kraken, monero};
use tracing::{info, warn};
use tracing_subscriber::filter::LevelFilter;
@ -76,13 +76,13 @@ async fn main() -> Result<()> {
let seed = Seed::from_file_or_generate(&config.data.dir)
.expect("Could not retrieve/initialize seed");
let exec_params = execution_params::Testnet::get_execution_params();
let env_config = env::Testnet::get_config();
let (bitcoin_wallet, monero_wallet) = init_wallets(
config.clone(),
&wallet_data_dir,
seed.derive_extended_private_key(exec_params.bitcoin_network)?,
exec_params,
seed.derive_extended_private_key(env_config.bitcoin_network)?,
env_config,
)
.await?;
@ -96,7 +96,7 @@ async fn main() -> Result<()> {
let (event_loop, mut swap_receiver) = EventLoop::new(
config.network.listen,
seed,
exec_params,
env_config,
Arc::new(bitcoin_wallet),
Arc::new(monero_wallet),
Arc::new(db),
@ -146,13 +146,13 @@ async fn init_wallets(
config: Config,
bitcoin_wallet_data_dir: &Path,
key: impl DerivableKey<Segwitv0> + Clone,
exec_params: ExecutionParams,
env_config: env::Config,
) -> Result<(bitcoin::Wallet, monero::Wallet)> {
let bitcoin_wallet = bitcoin::Wallet::new(
config.bitcoin.electrum_rpc_url,
bitcoin_wallet_data_dir,
key,
exec_params,
env_config,
)
.await?;
@ -167,7 +167,7 @@ async fn init_wallets(
let monero_wallet = monero::Wallet::new(
config.monero.wallet_rpc_url.clone(),
DEFAULT_WALLET_NAME.to_string(),
exec_params,
env_config,
);
// Setup the Monero wallet

42
swap/src/bin/swap.rs
View File

@ -23,12 +23,12 @@ use structopt::StructOpt;
use swap::bitcoin::{Amount, TxLock};
use swap::cli::command::{AliceConnectParams, Arguments, Command, Data, MoneroParams};
use swap::database::Database;
use swap::execution_params::{ExecutionParams, GetExecutionParams};
use swap::env::{Config, GetConfig};
use swap::network::quote::BidQuote;
use swap::protocol::bob;
use swap::protocol::bob::{Builder, EventLoop};
use swap::seed::Seed;
use swap::{bitcoin, execution_params, monero};
use swap::{bitcoin, env, monero};
use tracing::{debug, error, info, warn, Level};
use tracing_subscriber::FmtSubscriber;
use url::Url;
@ -76,7 +76,7 @@ async fn main() -> Result<()> {
let seed =
Seed::from_file_or_generate(data_dir.as_path()).context("Failed to read in seed file")?;
let exec_params = execution_params::Testnet::get_execution_params();
let env_config = env::Testnet::get_config();
match args.cmd {
Command::BuyXmr {
@ -92,18 +92,18 @@ async fn main() -> Result<()> {
},
electrum_rpc_url,
} => {
if receive_monero_address.network != exec_params.monero_network {
if receive_monero_address.network != env_config.monero_network {
bail!(
"Given monero address is on network {:?}, expected address on network {:?}",
receive_monero_address.network,
exec_params.monero_network
env_config.monero_network
)
}
let bitcoin_wallet =
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir.clone(), exec_params).await?;
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir.clone(), env_config).await?;
let (monero_wallet, _process) =
init_monero_wallet(data_dir, monero_daemon_host, exec_params).await?;
init_monero_wallet(data_dir, monero_daemon_host, env_config).await?;
let bitcoin_wallet = Arc::new(bitcoin_wallet);
let (event_loop, mut event_loop_handle) = EventLoop::new(
&seed.derive_libp2p_identity(),
@ -135,7 +135,7 @@ async fn main() -> Result<()> {
Uuid::new_v4(),
bitcoin_wallet.clone(),
Arc::new(monero_wallet),
exec_params,
env_config,
event_loop_handle,
receive_monero_address,
)
@ -178,14 +178,14 @@ async fn main() -> Result<()> {
},
electrum_rpc_url,
} => {
if receive_monero_address.network != exec_params.monero_network {
bail!("The given monero address is on network {:?}, expected address of network {:?}.", receive_monero_address.network, exec_params.monero_network)
if receive_monero_address.network != env_config.monero_network {
bail!("The given monero address is on network {:?}, expected address of network {:?}.", receive_monero_address.network, env_config.monero_network)
}
let bitcoin_wallet =
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir.clone(), exec_params).await?;
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir.clone(), env_config).await?;
let (monero_wallet, _process) =
init_monero_wallet(data_dir, monero_daemon_host, exec_params).await?;
init_monero_wallet(data_dir, monero_daemon_host, env_config).await?;
let bitcoin_wallet = Arc::new(bitcoin_wallet);
let (event_loop, event_loop_handle) = EventLoop::new(
@ -201,7 +201,7 @@ async fn main() -> Result<()> {
swap_id,
bitcoin_wallet.clone(),
Arc::new(monero_wallet),
exec_params,
env_config,
event_loop_handle,
receive_monero_address,
)
@ -223,7 +223,7 @@ async fn main() -> Result<()> {
electrum_rpc_url,
} => {
let bitcoin_wallet =
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir, exec_params).await?;
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir, env_config).await?;
let resume_state = db.get_state(swap_id)?.try_into_bob()?.into();
let cancel =
@ -248,7 +248,7 @@ async fn main() -> Result<()> {
electrum_rpc_url,
} => {
let bitcoin_wallet =
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir, exec_params).await?;
init_bitcoin_wallet(electrum_rpc_url, seed, data_dir, env_config).await?;
let resume_state = db.get_state(swap_id)?.try_into_bob()?.into();
@ -262,15 +262,15 @@ async fn init_bitcoin_wallet(
electrum_rpc_url: Url,
seed: Seed,
data_dir: PathBuf,
exec_params: ExecutionParams,
env_config: Config,
) -> Result<bitcoin::Wallet> {
let wallet_dir = data_dir.join("wallet");
let wallet = bitcoin::Wallet::new(
electrum_rpc_url.clone(),
&wallet_dir,
seed.derive_extended_private_key(exec_params.bitcoin_network)?,
exec_params,
seed.derive_extended_private_key(env_config.bitcoin_network)?,
env_config,
)
.await
.context("Failed to initialize Bitcoin wallet")?;
@ -283,9 +283,9 @@ async fn init_bitcoin_wallet(
async fn init_monero_wallet(
data_dir: PathBuf,
monero_daemon_host: String,
exec_params: ExecutionParams,
env_config: Config,
) -> Result<(monero::Wallet, monero::WalletRpcProcess)> {
let network = exec_params.monero_network;
let network = env_config.monero_network;
const MONERO_BLOCKCHAIN_MONITORING_WALLET_NAME: &str = "swap-tool-blockchain-monitoring-wallet";
@ -298,7 +298,7 @@ async fn init_monero_wallet(
let monero_wallet = monero::Wallet::new(
monero_wallet_rpc_process.endpoint(),
MONERO_BLOCKCHAIN_MONITORING_WALLET_NAME.to_string(),
exec_params,
env_config,
);
monero_wallet.open_or_create().await?;

8
swap/src/bitcoin/wallet.rs
View File

@ -1,6 +1,6 @@
use crate::bitcoin::timelocks::BlockHeight;
use crate::bitcoin::{Address, Amount, Transaction};
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use ::bitcoin::util::psbt::PartiallySignedTransaction;
use ::bitcoin::Txid;
use anyhow::{anyhow, bail, Context, Result};
@ -33,7 +33,7 @@ impl Wallet {
electrum_rpc_url: Url,
wallet_dir: &Path,
key: impl DerivableKey<Segwitv0> + Clone,
exec_params: ExecutionParams,
env_config: Config,
) -> Result<Self> {
// Workaround for https://github.com/bitcoindevkit/rust-electrum-client/issues/47.
let config = electrum_client::ConfigBuilder::default().retry(2).build();
@ -47,7 +47,7 @@ impl Wallet {
let bdk_wallet = bdk::Wallet::new(
bdk::template::BIP84(key.clone(), KeychainKind::External),
Some(bdk::template::BIP84(key, KeychainKind::Internal)),
exec_params.bitcoin_network,
env_config.bitcoin_network,
db,
ElectrumBlockchain::from(client),
)?;
@ -60,7 +60,7 @@ impl Wallet {
Ok(Self {
wallet: Arc::new(Mutex::new(bdk_wallet)),
client: Arc::new(Mutex::new(Client::new(electrum, interval)?)),
finality_confirmations: exec_params.bitcoin_finality_confirmations,
finality_confirmations: env_config.bitcoin_finality_confirmations,
})
}

26
swap/src/execution_params.rs → swap/src/env.rs
View File

@ -3,7 +3,7 @@ use std::time::Duration;
use time::NumericalStdDurationShort;
#[derive(Debug, Copy, Clone)]
pub struct ExecutionParams {
pub struct Config {
pub bob_time_to_act: Duration,
pub bitcoin_finality_confirmations: u32,
pub bitcoin_avg_block_time: Duration,
@ -15,8 +15,8 @@ pub struct ExecutionParams {
pub monero_network: monero::Network,
}
pub trait GetExecutionParams {
fn get_execution_params() -> ExecutionParams;
pub trait GetConfig {
fn get_config() -> Config;
}
#[derive(Clone, Copy)]
@ -28,9 +28,9 @@ pub struct Testnet;
#[derive(Clone, Copy)]
pub struct Regtest;
impl GetExecutionParams for Mainnet {
fn get_execution_params() -> ExecutionParams {
ExecutionParams {
impl GetConfig for Mainnet {
fn get_config() -> Config {
Config {
bob_time_to_act: 10.minutes(),
bitcoin_finality_confirmations: 3,
bitcoin_avg_block_time: 10.minutes(),
@ -44,9 +44,9 @@ impl GetExecutionParams for Mainnet {
}
}
impl GetExecutionParams for Testnet {
fn get_execution_params() -> ExecutionParams {
ExecutionParams {
impl GetConfig for Testnet {
fn get_config() -> Config {
Config {
bob_time_to_act: 60.minutes(),
bitcoin_finality_confirmations: 1,
bitcoin_avg_block_time: 5.minutes(),
@ -60,9 +60,9 @@ impl GetExecutionParams for Testnet {
}
}
impl GetExecutionParams for Regtest {
fn get_execution_params() -> ExecutionParams {
ExecutionParams {
impl GetConfig for Regtest {
fn get_config() -> Config {
Config {
bob_time_to_act: 30.seconds(),
bitcoin_finality_confirmations: 1,
bitcoin_avg_block_time: 5.seconds(),
@ -71,7 +71,7 @@ impl GetExecutionParams for Regtest {
bitcoin_network: bitcoin::Network::Regtest,
monero_avg_block_time: 1.seconds(),
monero_finality_confirmations: 10,
monero_network: monero::Network::Testnet,
monero_network: monero::Network::Mainnet, // yes this is strange
}
}
}

2
swap/src/lib.rs
View File

@ -20,7 +20,7 @@ pub mod asb;
pub mod bitcoin;
pub mod cli;
pub mod database;
pub mod execution_params;
pub mod env;
pub mod fs;
pub mod kraken;
pub mod monero;

16
swap/src/monero/wallet.rs
View File

@ -1,4 +1,4 @@
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::monero::{
Amount, InsufficientFunds, PrivateViewKey, PublicViewKey, TransferProof, TxHash,
};
@ -24,20 +24,16 @@ pub struct Wallet {
}
impl Wallet {
pub fn new(url: Url, name: String, exec_params: ExecutionParams) -> Self {
Self::new_with_client(Client::new(url), name, exec_params)
pub fn new(url: Url, name: String, env_config: Config) -> Self {
Self::new_with_client(Client::new(url), name, env_config)
}
pub fn new_with_client(
client: wallet::Client,
name: String,
exec_params: ExecutionParams,
) -> Self {
pub fn new_with_client(client: wallet::Client, name: String, env_config: Config) -> Self {
Self {
inner: Mutex::new(client),
network: exec_params.monero_network,
network: env_config.monero_network,
name,
avg_block_time: exec_params.monero_avg_block_time,
avg_block_time: env_config.monero_avg_block_time,
}
}

4
swap/src/protocol/alice.rs
View File

@ -1,7 +1,7 @@
//! Run an XMR/BTC swap in the role of Alice.
//! Alice holds XMR and wishes receive BTC.
use crate::database::Database;
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::{bitcoin, monero};
use std::sync::Arc;
use uuid::Uuid;
@ -28,7 +28,7 @@ pub struct Swap {
pub event_loop_handle: EventLoopHandle,
pub bitcoin_wallet: Arc<bitcoin::Wallet>,
pub monero_wallet: Arc<monero::Wallet>,
pub exec_params: ExecutionParams,
pub env_config: Config,
pub swap_id: Uuid,
pub db: Arc<Database>,
}

6
swap/src/protocol/alice/behaviour.rs
View File

@ -1,4 +1,4 @@
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::network::quote::BidQuote;
use crate::network::{peer_tracker, quote, spot_price};
use crate::protocol::alice::{
@ -206,11 +206,11 @@ impl Behaviour {
peer: PeerId,
btc: bitcoin::Amount,
xmr: monero::Amount,
exec_params: ExecutionParams,
env_config: Config,
bitcoin_wallet: &bitcoin::Wallet,
rng: &mut (impl RngCore + CryptoRng),
) -> Result<()> {
let state0 = State0::new(btc, xmr, exec_params, bitcoin_wallet, rng).await?;
let state0 = State0::new(btc, xmr, env_config, bitcoin_wallet, rng).await?;
tracing::info!(
%peer,

12
swap/src/protocol/alice/event_loop.rs
View File

@ -1,6 +1,6 @@
use crate::asb::{FixedRate, Rate};
use crate::database::Database;
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::monero::BalanceTooLow;
use crate::network::quote::BidQuote;
use crate::network::{spot_price, transport, TokioExecutor};
@ -23,7 +23,7 @@ use uuid::Uuid;
pub struct EventLoop<RS> {
swarm: libp2p::Swarm<Behaviour>,
peer_id: PeerId,
exec_params: ExecutionParams,
env_config: Config,
bitcoin_wallet: Arc<bitcoin::Wallet>,
monero_wallet: Arc<monero::Wallet>,
db: Arc<Database>,
@ -53,7 +53,7 @@ where
pub fn new(
listen_address: Multiaddr,
seed: Seed,
exec_params: ExecutionParams,
env_config: Config,
bitcoin_wallet: Arc<bitcoin::Wallet>,
monero_wallet: Arc<monero::Wallet>,
db: Arc<Database>,
@ -81,7 +81,7 @@ where
let event_loop = EventLoop {
swarm,
peer_id,
exec_params,
env_config,
bitcoin_wallet,
monero_wallet,
db,
@ -133,7 +133,7 @@ where
}
}
match self.swarm.start_execution_setup(peer, btc, xmr, self.exec_params, self.bitcoin_wallet.as_ref(), &mut OsRng).await {
match self.swarm.start_execution_setup(peer, btc, xmr, self.env_config, self.bitcoin_wallet.as_ref(), &mut OsRng).await {
Ok(_) => {},
Err(e) => {
tracing::warn!(%peer, "failed to start execution setup: {:#}", e);
@ -241,7 +241,7 @@ where
event_loop_handle: handle,
bitcoin_wallet: self.bitcoin_wallet.clone(),
monero_wallet: self.monero_wallet.clone(),
exec_params: self.exec_params,
env_config: self.env_config,
db: self.db.clone(),
state: initial_state,
swap_id,

8
swap/src/protocol/alice/state.rs
View File

@ -1,7 +1,7 @@
use crate::bitcoin::{
current_epoch, CancelTimelock, ExpiredTimelocks, PunishTimelock, TxCancel, TxPunish, TxRefund,
};
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::protocol::alice::{Message1, Message3};
use crate::protocol::bob::{Message0, Message2, Message4};
use crate::protocol::CROSS_CURVE_PROOF_SYSTEM;
@ -96,7 +96,7 @@ impl State0 {
pub async fn new<R>(
btc: bitcoin::Amount,
xmr: monero::Amount,
exec_params: ExecutionParams,
env_config: Config,
bitcoin_wallet: &bitcoin::Wallet,
rng: &mut R,
) -> Result<Self>
@ -124,8 +124,8 @@ impl State0 {
punish_address,
btc,
xmr,
cancel_timelock: exec_params.bitcoin_cancel_timelock,
punish_timelock: exec_params.bitcoin_punish_timelock,
cancel_timelock: env_config.bitcoin_cancel_timelock,
punish_timelock: env_config.bitcoin_punish_timelock,
})
}

28
swap/src/protocol/alice/swap.rs
View File

@ -2,7 +2,7 @@
//! Alice holds XMR and wishes receive BTC.
use crate::bitcoin::{ExpiredTimelocks, TxRedeem};
use crate::database::Database;
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::monero_ext::ScalarExt;
use crate::protocol::alice;
use crate::protocol::alice::event_loop::EventLoopHandle;
@ -51,7 +51,7 @@ pub async fn run_until(
swap.event_loop_handle,
swap.bitcoin_wallet,
swap.monero_wallet,
swap.exec_params,
swap.env_config,
swap.swap_id,
swap.db,
)
@ -67,7 +67,7 @@ async fn run_until_internal(
mut event_loop_handle: EventLoopHandle,
bitcoin_wallet: Arc<bitcoin::Wallet>,
monero_wallet: Arc<monero::Wallet>,
exec_params: ExecutionParams,
env_config: Config,
swap_id: Uuid,
db: Arc<Database>,
) -> Result<AliceState> {
@ -81,7 +81,7 @@ async fn run_until_internal(
bob_peer_id,
} => {
timeout(
exec_params.bob_time_to_act,
env_config.bob_time_to_act,
bitcoin_wallet
.watch_until_status(&state3.tx_lock, |status| status.has_been_seen()),
)
@ -90,7 +90,7 @@ async fn run_until_internal(
bitcoin_wallet
.watch_until_status(&state3.tx_lock, |status| {
status.is_confirmed_with(exec_params.bitcoin_finality_confirmations)
status.is_confirmed_with(env_config.bitcoin_finality_confirmations)
})
.await?;
@ -108,7 +108,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet,
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -144,7 +144,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet,
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -192,7 +192,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet.clone(),
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -258,7 +258,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet,
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -291,7 +291,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet,
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -326,7 +326,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet.clone(),
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -355,7 +355,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet.clone(),
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -432,7 +432,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet.clone(),
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)
@ -452,7 +452,7 @@ async fn run_until_internal(
event_loop_handle,
bitcoin_wallet.clone(),
monero_wallet,
exec_params,
env_config,
swap_id,
db,
)

12
swap/src/protocol/bob.rs
View File

@ -1,5 +1,5 @@
use crate::database::Database;
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::network::{peer_tracker, spot_price};
use crate::protocol::alice::TransferProof;
use crate::protocol::bob;
@ -37,7 +37,7 @@ pub struct Swap {
pub db: Database,
pub bitcoin_wallet: Arc<bitcoin::Wallet>,
pub monero_wallet: Arc<monero::Wallet>,
pub exec_params: ExecutionParams,
pub env_config: Config,
pub swap_id: Uuid,
pub receive_monero_address: ::monero::Address,
}
@ -50,7 +50,7 @@ pub struct Builder {
monero_wallet: Arc<monero::Wallet>,
init_params: InitParams,
exec_params: ExecutionParams,
env_config: Config,
event_loop_handle: EventLoopHandle,
@ -69,7 +69,7 @@ impl Builder {
swap_id: Uuid,
bitcoin_wallet: Arc<bitcoin::Wallet>,
monero_wallet: Arc<monero::Wallet>,
exec_params: ExecutionParams,
env_config: Config,
event_loop_handle: EventLoopHandle,
receive_monero_address: ::monero::Address,
) -> Self {
@ -79,7 +79,7 @@ impl Builder {
bitcoin_wallet,
monero_wallet,
init_params: InitParams::None,
exec_params,
env_config,
event_loop_handle,
receive_monero_address,
}
@ -105,7 +105,7 @@ impl Builder {
bitcoin_wallet: self.bitcoin_wallet.clone(),
monero_wallet: self.monero_wallet.clone(),
swap_id: self.swap_id,
exec_params: self.exec_params,
env_config: self.env_config,
receive_monero_address: self.receive_monero_address,
})
}

34
swap/src/protocol/bob/swap.rs
View File

@ -1,6 +1,6 @@
use crate::bitcoin::ExpiredTimelocks;
use crate::database::{Database, Swap};
use crate::execution_params::ExecutionParams;
use crate::env::Config;
use crate::monero::InsufficientFunds;
use crate::protocol::bob;
use crate::protocol::bob::event_loop::EventLoopHandle;
@ -41,7 +41,7 @@ pub async fn run_until(
swap.bitcoin_wallet,
swap.monero_wallet,
swap.swap_id,
swap.exec_params,
swap.env_config,
swap.receive_monero_address,
)
.await
@ -58,7 +58,7 @@ async fn run_until_internal(
bitcoin_wallet: Arc<bitcoin::Wallet>,
monero_wallet: Arc<monero::Wallet>,
swap_id: Uuid,
exec_params: ExecutionParams,
env_config: Config,
receive_monero_address: monero::Address,
) -> Result<BobState> {
trace!("Current state: {}", state);
@ -74,7 +74,7 @@ async fn run_until_internal(
let state2 = request_price_and_setup(
btc_amount,
&mut event_loop_handle,
exec_params,
env_config,
bitcoin_refund_address,
)
.await?;
@ -90,7 +90,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -121,7 +121,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -177,7 +177,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -232,7 +232,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -275,7 +275,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -311,7 +311,7 @@ async fn run_until_internal(
bitcoin_wallet.clone(),
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -342,7 +342,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -368,7 +368,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -398,7 +398,7 @@ async fn run_until_internal(
bitcoin_wallet,
monero_wallet,
swap_id,
exec_params,
env_config,
receive_monero_address,
)
.await
@ -414,7 +414,7 @@ async fn run_until_internal(
pub async fn request_price_and_setup(
btc: bitcoin::Amount,
event_loop_handle: &mut EventLoopHandle,
exec_params: ExecutionParams,
env_config: Config,
bitcoin_refund_address: bitcoin::Address,
) -> Result<bob::state::State2> {
let xmr = event_loop_handle.request_spot_price(btc).await?;
@ -425,10 +425,10 @@ pub async fn request_price_and_setup(
&mut OsRng,
btc,
xmr,
exec_params.bitcoin_cancel_timelock,
exec_params.bitcoin_punish_timelock,
env_config.bitcoin_cancel_timelock,
env_config.bitcoin_punish_timelock,
bitcoin_refund_address,
exec_params.monero_finality_confirmations,
env_config.monero_finality_confirmations,
);
let state2 = event_loop_handle.execution_setup(state0).await?;

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

@ -16,12 +16,12 @@ use std::time::Duration;
use swap::asb::FixedRate;
use swap::bitcoin::{CancelTimelock, PunishTimelock};
use swap::database::Database;
use swap::execution_params::{ExecutionParams, GetExecutionParams};
use swap::env::{Config, GetConfig};
use swap::protocol::alice::{AliceState, Swap};
use swap::protocol::bob::BobState;
use swap::protocol::{alice, bob};
use swap::seed::Seed;
use swap::{bitcoin, execution_params, monero};
use swap::{bitcoin, env, monero};
use tempfile::tempdir;
use testcontainers::clients::Cli;
use testcontainers::{Container, Docker, RunArgs};
@ -52,7 +52,7 @@ struct BobParams {
monero_wallet: Arc<monero::Wallet>,
alice_address: Multiaddr,
alice_peer_id: PeerId,
exec_params: ExecutionParams,
env_config: Config,
}
impl BobParams {
@ -64,7 +64,7 @@ impl BobParams {
self.swap_id,
self.bitcoin_wallet.clone(),
self.monero_wallet.clone(),
self.exec_params,
self.env_config,
event_loop_handle,
receive_address,
))
@ -311,13 +311,13 @@ pub async fn setup_test<T, F, C>(_config: C, testfn: T)
where
T: Fn(TestContext) -> F,
F: Future<Output = Result<()>>,
C: GetExecutionParams,
C: GetConfig,
{
let cli = Cli::default();
let _guard = init_tracing();
let exec_params = C::get_execution_params();
let env_config = C::get_config();
let (monero, containers) = testutils::init_containers(&cli).await;
@ -351,7 +351,7 @@ where
tempdir().unwrap().path(),
electrs_rpc_port,
alice_seed,
exec_params,
env_config,
)
.await;
@ -373,14 +373,14 @@ where
tempdir().unwrap().path(),
electrs_rpc_port,
bob_seed,
exec_params,
env_config,
)
.await;
let (alice_event_loop, alice_swap_handle) = alice::EventLoop::new(
alice_listen_address.clone(),
alice_seed,
exec_params,
env_config,
alice_bitcoin_wallet.clone(),
alice_monero_wallet.clone(),
alice_db,
@ -401,7 +401,7 @@ where
monero_wallet: bob_monero_wallet.clone(),
alice_address: alice_listen_address,
alice_peer_id,
exec_params,
env_config,
};
let test = TestContext {
@ -580,7 +580,7 @@ async fn init_test_wallets(
datadir: &Path,
electrum_rpc_port: u16,
seed: Seed,
exec_params: ExecutionParams,
env_config: Config,
) -> (Arc<bitcoin::Wallet>, Arc<monero::Wallet>) {
monero
.init(vec![(name, starting_balances.xmr.as_piconero())])
@ -590,7 +590,7 @@ async fn init_test_wallets(
let xmr_wallet = swap::monero::Wallet::new_with_client(
monero.wallet(name).unwrap().client(),
name.to_string(),
exec_params,
env_config,
);
let electrum_rpc_url = {
@ -601,9 +601,9 @@ async fn init_test_wallets(
let btc_wallet = swap::bitcoin::Wallet::new(
electrum_rpc_url,
datadir,
seed.derive_extended_private_key(exec_params.bitcoin_network)
seed.derive_extended_private_key(env_config.bitcoin_network)
.expect("Could not create extended private key from seed"),
exec_params,
env_config,
)
.await
.expect("could not init btc wallet");
@ -702,34 +702,34 @@ pub mod bob_run_until {
pub struct SlowCancelConfig;
impl GetExecutionParams for SlowCancelConfig {
fn get_execution_params() -> ExecutionParams {
ExecutionParams {
impl GetConfig for SlowCancelConfig {
fn get_config() -> Config {
Config {
bitcoin_cancel_timelock: CancelTimelock::new(180),
..execution_params::Regtest::get_execution_params()
..env::Regtest::get_config()
}
}
}
pub struct FastCancelConfig;
impl GetExecutionParams for FastCancelConfig {
fn get_execution_params() -> ExecutionParams {
ExecutionParams {
impl GetConfig for FastCancelConfig {
fn get_config() -> Config {
Config {
bitcoin_cancel_timelock: CancelTimelock::new(1),
..execution_params::Regtest::get_execution_params()
..env::Regtest::get_config()
}
}
}
pub struct FastPunishConfig;
impl GetExecutionParams for FastPunishConfig {
fn get_execution_params() -> ExecutionParams {
ExecutionParams {
impl GetConfig for FastPunishConfig {
fn get_config() -> Config {
Config {
bitcoin_cancel_timelock: CancelTimelock::new(1),
bitcoin_punish_timelock: PunishTimelock::new(1),
..execution_params::Regtest::get_execution_params()
..env::Regtest::get_config()
}
}
}