absolutebi/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-12-08 12:19:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make processor's test for an arbitrary coin, instead of just Monero

Browse Source
This commit is contained in:
Luke Parker
2022-06-09 04:34:31 -04:00
parent 75fb9b3198
commit 8e8bfabc83
1 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
processor/src/tests/mod.rs
View File

@@ -6,6 +6,8 @@ use rand::rngs::OsRng;
use group::Group; use group::Group;
use frost::Curve;
use crate::{ use crate::{
NetworkError, Network, NetworkError, Network,
Coin, coins::monero::Monero, Coin, coins::monero::Monero,
@@ -55,19 +57,17 @@ impl Network for LocalNetwork {
} }
} }
#[tokio::test] async fn test_send<C: Coin + Clone>(coin: C) {
async fn test() {
let monero = Monero::new("http://127.0.0.1:18081".to_string());
// Mine a block so there's a confirmed height // Mine a block so there's a confirmed height
monero.mine_block(monero.address(dalek_ff_group::EdwardsPoint::generator())).await; coin.mine_block(coin.address(<C::Curve as Curve>::G::generator())).await;
let height = monero.get_height().await.unwrap(); let height = coin.get_height().await.unwrap();
let mut networks = LocalNetwork::new(3); let mut networks = LocalNetwork::new(3);
let mut keys = frost::tests::key_gen::<_, <Monero as Coin>::Curve>(&mut OsRng); let mut keys = frost::tests::key_gen::<_, C::Curve>(&mut OsRng);
let mut wallets = vec![]; let mut wallets = vec![];
for i in 1 ..= 3 { for i in 1 ..= 3 {
let mut wallet = Wallet::new(MemCoinDb::new(), monero.clone()); let mut wallet = Wallet::new(MemCoinDb::new(), coin.clone());
wallet.acknowledge_height(0, height); wallet.acknowledge_height(0, height);
wallet.add_keys( wallet.add_keys(
&WalletKeys::new(Arc::try_unwrap(keys.remove(&i).take().unwrap()).unwrap(), 0) &WalletKeys::new(Arc::try_unwrap(keys.remove(&i).take().unwrap()).unwrap(), 0)
@@ -76,8 +76,8 @@ async fn test() {
} }
// Get the chain to a height where blocks have sufficient confirmations // Get the chain to a height where blocks have sufficient confirmations
while (height + Monero::CONFIRMATIONS) > monero.get_height().await.unwrap() { while (height + C::CONFIRMATIONS) > coin.get_height().await.unwrap() {
monero.mine_block(monero.address(dalek_ff_group::EdwardsPoint::generator())).await; coin.mine_block(coin.address(<C::Curve as Curve>::G::generator())).await;
} }
for wallet in wallets.iter_mut() { for wallet in wallets.iter_mut() {
@@ -85,23 +85,29 @@ async fn test() {
wallet.poll().await.unwrap(); wallet.poll().await.unwrap();
} }
monero.test_send(wallets[0].address()).await; coin.test_send(wallets[0].address()).await;
let mut futures = vec![]; let mut futures = vec![];
for (i, network) in networks.iter_mut().enumerate() { for (i, network) in networks.iter_mut().enumerate() {
let wallet = &mut wallets[i]; let wallet = &mut wallets[i];
wallet.poll().await.unwrap(); wallet.poll().await.unwrap();
let height = monero.get_height().await.unwrap(); let height = coin.get_height().await.unwrap();
wallet.acknowledge_height(1, height - 10); wallet.acknowledge_height(1, height - 10);
let signable = wallet.prepare_sends( let signable = wallet.prepare_sends(
1, 1,
vec![(wallet.address(), 10000000000)] vec![(wallet.address(), 10000000000)]
).await.unwrap().1.swap_remove(0); ).await.unwrap().1.swap_remove(0);
futures.push(monero.attempt_send(network, signable, &[1, 2, 3])); futures.push(coin.attempt_send(network, signable, &[1, 2, 3]));
} }
println!( println!(
"{:?}", "{:?}",
hex::encode(futures::future::join_all(futures).await.swap_remove(0).unwrap().0) hex::encode(futures::future::join_all(futures).await.swap_remove(0).unwrap().0)
); );
} }
#[tokio::test]
async fn monero() {
test_send(Monero::new("http://127.0.0.1:18081".to_string())).await;
}