Ecosystem/serai
Ecosystem/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-03-12 09:26:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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
No known key found for this signature in database
GPG key ID: F9F1386DB1E119B6
1 changed files with 18 additions and 12 deletions
Show stats 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 frost::Curve;
use crate::{
NetworkError, Network,
Coin, coins::monero::Monero,
@ -55,19 +57,17 @@ impl Network for LocalNetwork {
}
}
#[tokio::test]
async fn test() {
let monero = Monero::new("http://127.0.0.1:18081".to_string());
async fn test_send<C: Coin + Clone>(coin: C) {
// Mine a block so there's a confirmed height
monero.mine_block(monero.address(dalek_ff_group::EdwardsPoint::generator())).await;
let height = monero.get_height().await.unwrap();
coin.mine_block(coin.address(<C::Curve as Curve>::G::generator())).await;
let height = coin.get_height().await.unwrap();
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![];
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.add_keys(
&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
while (height + Monero::CONFIRMATIONS) > monero.get_height().await.unwrap() {
monero.mine_block(monero.address(dalek_ff_group::EdwardsPoint::generator())).await;
while (height + C::CONFIRMATIONS) > coin.get_height().await.unwrap() {
coin.mine_block(coin.address(<C::Curve as Curve>::G::generator())).await;
}
for wallet in wallets.iter_mut() {
@ -85,23 +85,29 @@ async fn test() {
wallet.poll().await.unwrap();
}
monero.test_send(wallets[0].address()).await;
coin.test_send(wallets[0].address()).await;
let mut futures = vec![];
for (i, network) in networks.iter_mut().enumerate() {
let wallet = &mut wallets[i];
wallet.poll().await.unwrap();
let height = monero.get_height().await.unwrap();
let height = coin.get_height().await.unwrap();
wallet.acknowledge_height(1, height - 10);
let signable = wallet.prepare_sends(
1,
vec![(wallet.address(), 10000000000)]
).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!(
"{:?}",
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;
}