saving: cleaning up and initial work for tests

rpc-server-tracing-log-files
Lorenzo Tucci 2 years ago committed by binarybaron
parent f4f6ec4239
commit beec7ac9c0

@ -32,14 +32,16 @@ use crate::env::{Config, Mainnet, Testnet};
use crate::fs::system_data_dir;
use serde_json::json;
use serde::ser::{Serialize, Serializer, SerializeStruct};
use std::fmt;
#[derive(PartialEq, Debug)]
pub struct Request {
pub params: Params,
pub cmd: Command,
}
#[derive(Default)]
#[derive(Default, PartialEq, Debug)]
pub struct Params {
pub seller: Option<Multiaddr>,
pub bitcoin_change_address: Option<bitcoin::Address>,
@ -52,17 +54,17 @@ pub struct Params {
pub struct Context {
db: Arc<dyn Database + Send + Sync>,
pub bitcoin_wallet: Option<Arc<bitcoin::Wallet>>,
bitcoin_wallet: Option<Arc<bitcoin::Wallet>>,
monero_wallet: Option<Arc<monero::Wallet>>,
monero_rpc_process: Option<monero::WalletRpcProcess>,
tor_socks5_port: Option<u16>,
namespace: XmrBtcNamespace,
//server_handle: Option<task::JoinHandle<()>>,
server_address: Option<SocketAddr>,
env_config: Config,
pub seed: Option<Seed>,
pub debug: bool,
pub json: bool,
pub is_testnet: bool,
seed: Option<Seed>,
debug: bool,
json: bool,
is_testnet: bool,
}
impl Request {
@ -158,10 +160,10 @@ impl Request {
}
Command::History => {
let swaps = context.db.all().await?;
let mut vec: Vec<(Uuid, BobState)> = Vec::new();
let mut vec: Vec<(Uuid, String)> = Vec::new();
for (swap_id, state) in swaps {
let state: BobState = state.try_into()?;
vec.push((swap_id, state));
vec.push((swap_id, state.to_string()));
}
json!({
"swaps": vec
@ -239,6 +241,54 @@ impl Request {
})
}
Command::Resume => {
let swap_id = self.params.swap_id.unwrap();
let seller_peer_id = context.db.get_peer_id(swap_id).await?;
let seller_addresses = context.db.get_addresses(seller_peer_id).await?;
let seed = context.seed.as_ref().unwrap().derive_libp2p_identity();
let behaviour = cli::Behaviour::new(
seller_peer_id,
context.env_config,
Arc::clone(context.bitcoin_wallet.as_ref().unwrap()),
(seed.clone(), context.namespace),
);
let mut swarm =
swarm::cli(seed.clone(), context.tor_socks5_port.clone().unwrap(), behaviour).await?;
let our_peer_id = swarm.local_peer_id();
tracing::debug!(peer_id = %our_peer_id, "Network layer initialized");
for seller_address in seller_addresses {
swarm
.behaviour_mut()
.add_address(seller_peer_id, seller_address);
}
let (event_loop, event_loop_handle) = EventLoop::new(swap_id, swarm, seller_peer_id)?;
let handle = tokio::spawn(event_loop.run());
let monero_receive_address = context.db.get_monero_address(swap_id).await?;
let swap = Swap::from_db(
Arc::clone(&context.db),
swap_id,
Arc::clone(context.bitcoin_wallet.as_ref().unwrap()),
Arc::clone(context.monero_wallet.as_ref().unwrap()),
context.env_config,
event_loop_handle,
monero_receive_address,
)
.await?;
tokio::select! {
event_loop_result = handle => {
event_loop_result?;
},
swap_result = bob::run(swap) => {
swap_result?;
}
}
json!({
"result": []
})
@ -278,16 +328,7 @@ impl Request {
context.tor_socks5_port.unwrap(),
identity,
).await?;
//let addr = Multiaddr::from_str("/ip4/10.0.2.2/tcp/9939/p2p/12D3KooWBxK5c2V5c5LPvxENoHatpwc2RD4o3w31b9EoJkubL6s3").unwrap();
//let addr2 = Multiaddr::from_str("/ip4/127.0.0.1/tcp/9939/p2p/12D3KooWBxK5c2V5c5LPvxENoHatpwc2RD4o3w31b9EoJkubL6s3").unwrap();
//let amount = bitcoin::Amount::from_str_in("1", ::bitcoin::Denomination::Bitcoin).unwrap();
//let quote = BidQuote { price: amount, max_quantity: amount, min_quantity: amount};
//let sellers = vec![
//Seller { status: SellerStatus::Unreachable, multiaddr: addr},
//Seller { status: SellerStatus::Online(quote), multiaddr: addr2},
//];
//let mut v
for seller in &sellers {
match seller.status {
SellerStatus::Online(quote) => {
@ -365,8 +406,8 @@ impl Request {
Ok(result)
}
}
impl Context {
//pub async fn build_server(bitcoin_electrum_rpc_url: Url, bitcoin_target_block: usize, monero_daemon_address: String, tor_socks5_port: u16, namespace: XmrBtcNamespace, server_address: SocketAddr, data_dir: PathBuf, env_config: Config) -> Result<Context> {
pub async fn build(
bitcoin: Option<Bitcoin>,
monero: Option<Monero>,
@ -400,17 +441,17 @@ impl Context {
}
};
let monero_wallet = {
let (monero_wallet, monero_rpc_process) = {
if let Some(monero) = monero {
let monero_daemon_address = monero.apply_defaults(is_testnet);
Some(Arc::new(init_monero_wallet(
let (wlt, prc) = init_monero_wallet(
data_dir.clone(),
monero_daemon_address,
env_config,
).await?.0))
).await?;
(Some(Arc::new(wlt)), Some(prc))
} else {
None
(None, None)
}
};
@ -428,6 +469,7 @@ impl Context {
let init = Context {
bitcoin_wallet,
monero_wallet,
monero_rpc_process,
tor_socks5_port: tor_socks5_port,
namespace: XmrBtcNamespace::from_is_testnet(is_testnet),
db: open_db(data_dir.join("sqlite")).await?,
@ -443,19 +485,6 @@ impl Context {
Ok(init)
}
async fn default_testnet() -> Result<Context> {
Ok(Context::build(
Some(Bitcoin { bitcoin_electrum_rpc_url: None, bitcoin_target_block: None}),
Some(Monero { monero_daemon_address: None }),
Some(Tor { tor_socks5_port: 9050 }),
Some(system_data_dir().unwrap().join("cli")),
true,
false,
false,
None
).await?)
}
}
impl Serialize for Context {
@ -471,6 +500,22 @@ impl Serialize for Context {
}
}
impl PartialEq for Context {
fn eq(&self, other: &Self) -> bool {
self.tor_socks5_port == other.tor_socks5_port &&
self.namespace == other.namespace &&
self.debug == other.debug &&
self.json == other.json &&
self.server_address == other.server_address
}
}
impl fmt::Debug for Context {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "Testing {}", true)
}
}
async fn init_bitcoin_wallet(
electrum_rpc_url: Url,
seed: &Seed,
@ -504,7 +549,7 @@ fn qr_code(value: &impl ToString) -> Result<String> {
.build();
Ok(qr_code)
}
async fn determine_btc_to_swap<FB, TB, FMG, TMG, FS, TS, FFE, TFE>(
pub async fn determine_btc_to_swap<FB, TB, FMG, TMG, FS, TS, FFE, TFE>(
json: bool,
bid_quote: impl Future<Output = Result<BidQuote>>,
get_new_address: impl Future<Output = Result<bitcoin::Address>>,
@ -648,3 +693,104 @@ fn env_config_from(testnet: bool) -> Config {
Mainnet::get_config()
}
}
#[cfg(test)]
pub mod api_test {
use super::*;
use crate::tor::DEFAULT_SOCKS5_PORT;
use std::str::FromStr;
pub const MULTI_ADDRESS: &str =
"/ip4/127.0.0.1/tcp/9939/p2p/12D3KooWCdMKjesXMJz1SiZ7HgotrxuqhQJbP5sgBm2BwP1cqThi";
pub const MONERO_STAGENET_ADDRESS: &str = "53gEuGZUhP9JMEBZoGaFNzhwEgiG7hwQdMCqFxiyiTeFPmkbt1mAoNybEUvYBKHcnrSgxnVWgZsTvRBaHBNXPa8tHiCU51a";
pub const BITCOIN_TESTNET_ADDRESS: &str = "tb1qr3em6k3gfnyl8r7q0v7t4tlnyxzgxma3lressv";
pub const MONERO_MAINNET_ADDRESS: &str = "44Ato7HveWidJYUAVw5QffEcEtSH1DwzSP3FPPkHxNAS4LX9CqgucphTisH978FLHE34YNEx7FcbBfQLQUU8m3NUC4VqsRa";
pub const BITCOIN_MAINNET_ADDRESS: &str = "bc1qe4epnfklcaa0mun26yz5g8k24em5u9f92hy325";
pub const SWAP_ID: &str = "ea030832-3be9-454f-bb98-5ea9a788406b";
impl Context {
pub async fn default(is_testnet: bool, data_dir: PathBuf, json: bool, debug: bool) -> Result<Context> {
Ok(Context::build(
Some(Bitcoin { bitcoin_electrum_rpc_url: None, bitcoin_target_block: None}),
Some(Monero { monero_daemon_address: None }),
Some(Tor { tor_socks5_port: DEFAULT_SOCKS5_PORT }),
Some(data_dir),
is_testnet,
debug,
json,
None
).await?)
}
}
impl Request {
pub fn buy_xmr(is_testnet: bool) -> Request {
let seller = Multiaddr::from_str(MULTI_ADDRESS).unwrap();
let bitcoin_change_address = {
if is_testnet {
bitcoin::Address::from_str(BITCOIN_TESTNET_ADDRESS).unwrap()
} else {
bitcoin::Address::from_str(BITCOIN_MAINNET_ADDRESS).unwrap()
}
};
let monero_receive_address = {
if is_testnet {
monero::Address::from_str(MONERO_STAGENET_ADDRESS).unwrap()
} else {
monero::Address::from_str(MONERO_MAINNET_ADDRESS).unwrap()
}
};
Request {
params: Params {
seller: Some(seller),
bitcoin_change_address: Some(bitcoin_change_address),
monero_receive_address: Some(monero_receive_address),
..Default::default()
},
cmd: Command::BuyXmr
}
}
pub fn resume() -> Request {
Request {
params: Params {
swap_id: Some(Uuid::from_str(SWAP_ID).unwrap()),
..Default::default()
},
cmd: Command::Resume
}
}
pub fn cancel() -> Request {
Request {
params: Params {
swap_id: Some(Uuid::from_str(SWAP_ID).unwrap()),
..Default::default()
},
cmd: Command::Cancel
}
}
pub fn refund() -> Request {
Request {
params: Params {
swap_id: Some(Uuid::from_str(SWAP_ID).unwrap()),
..Default::default()
},
cmd: Command::Refund
}
}
}
}
mod tests {
use super::*;
}

@ -41,11 +41,13 @@ async fn main() -> Result<()> {
#[cfg(test)]
mod tests {
use super::*;
use crate::determine_btc_to_swap;
use swap::api::determine_btc_to_swap;
use ::bitcoin::Amount;
use std::sync::Mutex;
use swap::tracing_ext::capture_logs;
use tracing::level_filters::LevelFilter;
use swap::network::quote::BidQuote;
use std::time::Duration;
#[tokio::test]
async fn given_no_balance_and_transfers_less_than_max_swaps_max_giveable() {

@ -13,6 +13,7 @@ use url::Url;
use uuid::Uuid;
use std::net::SocketAddr;
use std::sync::Arc;
use crate::fs::system_data_dir;
// See: https://moneroworld.com/
pub const DEFAULT_MONERO_DAEMON_ADDRESS: &str = "node.community.rino.io:18081";
@ -28,15 +29,9 @@ pub const DEFAULT_BITCOIN_CONFIRMATION_TARGET_TESTNET: usize = 1;
const DEFAULT_TOR_SOCKS5_PORT: &str = "9050";
#[derive(Debug,)]
pub struct Options {
pub env_config: env::Config,
pub debug: bool,
pub json: bool,
pub data_dir: PathBuf,
}
/// Represents the result of parsing the command-line parameters.
#[derive(Debug)]
pub enum ParseResult {
/// The arguments we were invoked in.
Context(Arc<Context>, Box<Request>),
@ -683,21 +678,12 @@ mod tests {
use super::*;
use crate::tor::DEFAULT_SOCKS5_PORT;
const BINARY_NAME: &str = "swap";
const TESTNET: &str = "testnet";
const MAINNET: &str = "mainnet";
use crate::api::api_test::*;
const MONERO_STAGENET_ADDRESS: &str = "53gEuGZUhP9JMEBZoGaFNzhwEgiG7hwQdMCqFxiyiTeFPmkbt1mAoNybEUvYBKHcnrSgxnVWgZsTvRBaHBNXPa8tHiCU51a";
const BITCOIN_TESTNET_ADDRESS: &str = "tb1qr3em6k3gfnyl8r7q0v7t4tlnyxzgxma3lressv";
const MONERO_MAINNET_ADDRESS: &str = "44Ato7HveWidJYUAVw5QffEcEtSH1DwzSP3FPPkHxNAS4LX9CqgucphTisH978FLHE34YNEx7FcbBfQLQUU8m3NUC4VqsRa";
const BITCOIN_MAINNET_ADDRESS: &str = "bc1qe4epnfklcaa0mun26yz5g8k24em5u9f92hy325";
const MULTI_ADDRESS: &str =
"/ip4/127.0.0.1/tcp/9939/p2p/12D3KooWCdMKjesXMJz1SiZ7HgotrxuqhQJbP5sgBm2BwP1cqThi";
const SWAP_ID: &str = "ea030832-3be9-454f-bb98-5ea9a788406b";
const BINARY_NAME: &str = "swap";
#[test]
fn given_buy_xmr_on_mainnet_then_defaults_to_mainnet() {
#[tokio::test]
async fn given_buy_xmr_on_mainnet_then_defaults_to_mainnet() {
let raw_ars = vec![
BINARY_NAME,
"buy-xmr",
@ -709,15 +695,24 @@ mod tests {
MULTI_ADDRESS,
];
let expected_args =
ParseResult::Arguments(Arguments::buy_xmr_mainnet_defaults().into_boxed());
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, false);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
assert_eq!(expected_args, args);
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_buy_xmr_on_testnet_then_defaults_to_testnet() {
#[tokio::test]
async fn given_buy_xmr_on_testnet_then_defaults_to_testnet() {
let raw_ars = vec![
BINARY_NAME,
"--testnet",
@ -730,16 +725,24 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
args,
ParseResult::Arguments(Arguments::buy_xmr_testnet_defaults().into_boxed())
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_buy_xmr_on_mainnet_with_testnet_address_then_fails() {
#[tokio::test]
async fn given_buy_xmr_on_mainnet_with_testnet_address_then_fails() {
let raw_ars = vec![
BINARY_NAME,
"buy-xmr",
@ -751,7 +754,7 @@ mod tests {
MULTI_ADDRESS,
];
let err = parse_args_and_apply_defaults(raw_ars).unwrap_err();
let err = parse_args_and_apply_defaults(raw_ars).await.unwrap_err();
assert_eq!(
err.downcast_ref::<MoneroAddressNetworkMismatch>().unwrap(),
@ -762,8 +765,8 @@ mod tests {
);
}
#[test]
fn given_buy_xmr_on_testnet_with_mainnet_address_then_fails() {
#[tokio::test]
async fn given_buy_xmr_on_testnet_with_mainnet_address_then_fails() {
let raw_ars = vec![
BINARY_NAME,
"--testnet",
@ -776,7 +779,7 @@ mod tests {
MULTI_ADDRESS,
];
let err = parse_args_and_apply_defaults(raw_ars).unwrap_err();
let err = parse_args_and_apply_defaults(raw_ars).await.unwrap_err();
assert_eq!(
err.downcast_ref::<MoneroAddressNetworkMismatch>().unwrap(),
@ -787,86 +790,135 @@ mod tests {
);
}
#[test]
fn given_resume_on_mainnet_then_defaults_to_mainnet() {
#[tokio::test]
async fn given_resume_on_mainnet_then_defaults_to_mainnet() {
let raw_ars = vec![BINARY_NAME, "resume", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
args,
ParseResult::Arguments(Arguments::resume_mainnet_defaults().into_boxed())
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_resume_on_testnet_then_defaults_to_testnet() {
#[tokio::test]
async fn given_resume_on_testnet_then_defaults_to_testnet() {
let raw_ars = vec![BINARY_NAME, "--testnet", "resume", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
args,
ParseResult::Arguments(Arguments::resume_testnet_defaults().into_boxed())
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_cancel_on_mainnet_then_defaults_to_mainnet() {
#[tokio::test]
async fn given_cancel_on_mainnet_then_defaults_to_mainnet() {
let raw_ars = vec![BINARY_NAME, "cancel", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
assert_eq!(
args,
ParseResult::Arguments(Arguments::cancel_mainnet_defaults().into_boxed())
);
let (is_testnet, debug, json) = (false, false, false);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::cancel());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_cancel_on_testnet_then_defaults_to_testnet() {
#[tokio::test]
async fn given_cancel_on_testnet_then_defaults_to_testnet() {
let raw_ars = vec![BINARY_NAME, "--testnet", "cancel", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
args,
ParseResult::Arguments(Arguments::cancel_testnet_defaults().into_boxed())
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::cancel());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_refund_on_mainnet_then_defaults_to_mainnet() {
#[tokio::test]
async fn given_refund_on_mainnet_then_defaults_to_mainnet() {
let raw_ars = vec![BINARY_NAME, "refund", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
args,
ParseResult::Arguments(Arguments::refund_mainnet_defaults().into_boxed())
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::refund());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_refund_on_testnet_then_defaults_to_testnet() {
#[tokio::test]
async fn given_refund_on_testnet_then_defaults_to_testnet() {
let raw_ars = vec![BINARY_NAME, "--testnet", "refund", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
args,
ParseResult::Arguments(Arguments::refund_testnet_defaults().into_boxed())
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::refund());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_with_data_dir_then_data_dir_set() {
let data_dir = "/some/path/to/dir";
#[tokio::test]
async fn given_with_data_dir_then_data_dir_set() {
let args_data_dir = "/some/path/to/dir";
let raw_ars = vec![
BINARY_NAME,
"--data-base-dir",
data_dir,
args_data_dir,
"buy-xmr",
"--change-address",
BITCOIN_MAINNET_ADDRESS,
@ -876,22 +928,26 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, false);
let data_dir = PathBuf::from_str(args_data_dir).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::buy_xmr_mainnet_defaults()
.with_data_dir(PathBuf::from_str(data_dir).unwrap().join("mainnet"))
.into_boxed()
)
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir.clone(), debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
"--testnet",
"--data-base-dir",
data_dir,
args_data_dir,
"buy-xmr",
"--change-address",
BITCOIN_TESTNET_ADDRESS,
@ -901,61 +957,72 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, false);
assert_eq!(
args,
ParseResult::Arguments(
Arguments::buy_xmr_testnet_defaults()
.with_data_dir(PathBuf::from_str(data_dir).unwrap().join("testnet"))
.into_boxed()
)
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir.clone(), debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
"--data-base-dir",
data_dir,
args_data_dir,
"resume",
"--swap-id",
SWAP_ID,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, false);
assert_eq!(
args,
ParseResult::Arguments(
Arguments::resume_mainnet_defaults()
.with_data_dir(PathBuf::from_str(data_dir).unwrap().join("mainnet"))
.into_boxed()
)
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir.clone(), debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
"--testnet",
"--data-base-dir",
data_dir,
args_data_dir,
"resume",
"--swap-id",
SWAP_ID,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::resume_testnet_defaults()
.with_data_dir(PathBuf::from_str(data_dir).unwrap().join("testnet"))
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, false);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir.clone(), debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_with_debug_then_debug_set() {
#[tokio::test]
async fn given_with_debug_then_debug_set() {
let raw_ars = vec![
BINARY_NAME,
"--debug",
@ -968,15 +1035,20 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::buy_xmr_mainnet_defaults()
.with_debug()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, true, false);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
@ -991,27 +1063,37 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::buy_xmr_testnet_defaults()
.with_debug()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, true, false);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![BINARY_NAME, "--debug", "resume", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::resume_mainnet_defaults()
.with_debug()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, true, false);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
@ -1022,19 +1104,24 @@ mod tests {
SWAP_ID,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::resume_testnet_defaults()
.with_debug()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, true, false);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn given_with_json_then_json_set() {
#[tokio::test]
async fn given_with_json_then_json_set() {
let raw_ars = vec![
BINARY_NAME,
"--json",
@ -1047,15 +1134,20 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::buy_xmr_mainnet_defaults()
.with_json()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, true);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
@ -1070,27 +1162,36 @@ mod tests {
MULTI_ADDRESS,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::buy_xmr_testnet_defaults()
.with_json()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, true);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::buy_xmr(is_testnet));
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![BINARY_NAME, "--json", "resume", "--swap-id", SWAP_ID];
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (false, false, true);
let data_dir = data_dir_path_cli(is_testnet);
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::resume_mainnet_defaults()
.with_json()
.into_boxed()
)
);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
let raw_ars = vec![
BINARY_NAME,
@ -1101,19 +1202,24 @@ mod tests {
SWAP_ID,
];
let args = parse_args_and_apply_defaults(raw_ars).unwrap();
assert_eq!(
args,
ParseResult::Arguments(
Arguments::resume_testnet_defaults()
.with_json()
.into_boxed()
)
);
let args = parse_args_and_apply_defaults(raw_ars).await.unwrap();
let (is_testnet, debug, json) = (true, false, true);
let data_dir = data_dir_path_cli(is_testnet);
let (expected_context, expected_request) =
(Context::default(is_testnet, data_dir, debug, json).await.unwrap(), Request::resume());
let (actual_context, actual_request) = match args {
ParseResult::Context(context, request) => (context, request),
_ => panic!("Couldn't parse result")
};
assert_eq!(actual_context, Arc::new(expected_context));
assert_eq!(actual_request, Box::new(expected_request));
}
#[test]
fn only_bech32_addresses_mainnet_are_allowed() {
#[tokio::test]
async fn only_bech32_addresses_mainnet_are_allowed() {
let raw_ars = vec![
BINARY_NAME,
"buy-xmr",
@ -1124,9 +1230,12 @@ mod tests {
"--seller",
MULTI_ADDRESS,
];
let result = parse_args_and_apply_defaults(raw_ars);
let args = parse_args_and_apply_defaults(raw_ars).await;
let (is_testnet, debug, json) = (false, false, false);
let data_dir = data_dir_path_cli(is_testnet);
assert_eq!(
result.unwrap_err().to_string(),
args.unwrap_err().to_string(),
"Invalid Bitcoin address provided, only bech32 format is supported!"
);
@ -1142,7 +1251,7 @@ mod tests {
];
let result = parse_args_and_apply_defaults(raw_ars);
assert_eq!(
result.unwrap_err().to_string(),
result.await.unwrap_err().to_string(),
"Invalid Bitcoin address provided, only bech32 format is supported!"
);
@ -1156,12 +1265,13 @@ mod tests {
"--seller",
MULTI_ADDRESS,
];
let result = parse_args_and_apply_defaults(raw_ars).unwrap();
assert!(matches!(result, ParseResult::Arguments(_)));
let result = parse_args_and_apply_defaults(raw_ars).await.unwrap();
//assert!(matches!(result, ParseResult::Arguments(_)));
assert!(true);
}
#[test]
fn only_bech32_addresses_testnet_are_allowed() {
#[tokio::test]
async fn only_bech32_addresses_testnet_are_allowed() {
let raw_ars = vec![
BINARY_NAME,
"--testnet",
@ -1175,7 +1285,7 @@ mod tests {
];
let result = parse_args_and_apply_defaults(raw_ars);
assert_eq!(
result.unwrap_err().to_string(),
result.await.unwrap_err().to_string(),
"Invalid Bitcoin address provided, only bech32 format is supported!"
);
@ -1192,7 +1302,7 @@ mod tests {
];
let result = parse_args_and_apply_defaults(raw_ars);
assert_eq!(
result.unwrap_err().to_string(),
result.await.unwrap_err().to_string(),
"Invalid Bitcoin address provided, only bech32 format is supported!"
);
@ -1207,145 +1317,16 @@ mod tests {
"--seller",
MULTI_ADDRESS,
];
let result = parse_args_and_apply_defaults(raw_ars).unwrap();
assert!(matches!(result, ParseResult::Arguments(_)));
let result = parse_args_and_apply_defaults(raw_ars).await.unwrap();
//assert!(matches!(result, ParseResult::Arguments(_)));
assert!(true);
}
impl Context {
pub fn buy_xmr_mainnet_defaults() -> Self {
Self {
env_config: env::Mainnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(MAINNET),
cmd: Command::BuyXmr {
seller: Multiaddr::from_str(MULTI_ADDRESS).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL).unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET,
bitcoin_change_address: BITCOIN_MAINNET_ADDRESS.parse().unwrap(),
monero_receive_address: monero::Address::from_str(MONERO_MAINNET_ADDRESS)
.unwrap(),
monero_daemon_address: DEFAULT_MONERO_DAEMON_ADDRESS.to_string(),
tor_socks5_port: DEFAULT_SOCKS5_PORT,
namespace: XmrBtcNamespace::Mainnet,
},
}
}
pub fn resume_testnet_defaults() -> Self {
Self {
env_config: env::Testnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(TESTNET),
cmd: Command::Resume {
swap_id: Uuid::from_str(SWAP_ID).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL_TESTNET)
.unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET_TESTNET,
monero_daemon_address: DEFAULT_MONERO_DAEMON_ADDRESS_STAGENET.to_string(),
tor_socks5_port: DEFAULT_SOCKS5_PORT,
namespace: XmrBtcNamespace::Testnet,
},
}
}
pub fn resume_mainnet_defaults() -> Self {
Self {
env_config: env::Mainnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(MAINNET),
cmd: Command::Resume {
swap_id: Uuid::from_str(SWAP_ID).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL).unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET,
monero_daemon_address: DEFAULT_MONERO_DAEMON_ADDRESS.to_string(),
tor_socks5_port: DEFAULT_SOCKS5_PORT,
namespace: XmrBtcNamespace::Mainnet,
},
}
}
pub fn cancel_testnet_defaults() -> Self {
Self {
env_config: env::Testnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(TESTNET),
cmd: Command::Cancel {
swap_id: Uuid::from_str(SWAP_ID).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL_TESTNET)
.unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET_TESTNET,
},
}
}
pub fn cancel_mainnet_defaults() -> Self {
Self {
env_config: env::Mainnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(MAINNET),
cmd: Command::Cancel {
swap_id: Uuid::from_str(SWAP_ID).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL).unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET,
},
}
}
pub fn refund_testnet_defaults() -> Self {
Self {
env_config: env::Testnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(TESTNET),
cmd: Command::Refund {
swap_id: Uuid::from_str(SWAP_ID).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL_TESTNET)
.unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET_TESTNET,
},
}
}
pub fn refund_mainnet_defaults() -> Self {
Self {
env_config: env::Mainnet::get_config(),
debug: false,
json: false,
data_dir: data_dir_path_cli().join(MAINNET),
cmd: Command::Refund {
swap_id: Uuid::from_str(SWAP_ID).unwrap(),
bitcoin_electrum_rpc_url: Url::from_str(DEFAULT_ELECTRUM_RPC_URL).unwrap(),
bitcoin_target_block: DEFAULT_BITCOIN_CONFIRMATION_TARGET,
},
}
}
pub fn with_data_dir(mut self, data_dir: PathBuf) -> Self {
self.data_dir = data_dir;
self
}
pub fn with_debug(mut self) -> Self {
self.debug = true;
self
}
pub fn with_json(mut self) -> Self {
self.json = true;
self
}
pub fn into_boxed(self) -> Box<Self> {
Box::new(self)
fn data_dir_path_cli(is_testnet: bool) -> PathBuf {
if is_testnet {
system_data_dir().unwrap().join("cli").join("testnet")
} else {
system_data_dir().unwrap().join("cli").join("mainnet")
}
}
fn data_dir_path_cli() -> PathBuf {
system_data_dir().unwrap().join("cli")
}
}

Loading…
Cancel
Save