mirror of
https://github.com/serai-dex/serai.git
synced 2025-12-08 20:29:23 +00:00
f6f945e747
It's largely unoptimized, and not yet exclusive to validators, yet has basic sanity (using message content for ID instead of sender + index). Fixes bugs as found. Notably, we used a time in milliseconds where the Tributary expected seconds. Also has Tributary::new jump to the presumed round number. This reduces slashes when starting new chains (whose times will be before the current time) and was the only way I was able to observe successful confirmations given current surrounding infrastructure.
68 lines
2.0 KiB
Rust
68 lines
2.0 KiB
Rust
use core::time::Duration;
|
|
use std::{sync::Arc, collections::HashMap};
|
|
|
|
use rand_core::OsRng;
|
|
|
|
use ciphersuite::{Ciphersuite, Ristretto};
|
|
|
|
use tokio::{sync::RwLock, time::sleep};
|
|
|
|
use serai_db::MemDb;
|
|
|
|
use tributary::Tributary;
|
|
|
|
use crate::{
|
|
tributary::Transaction,
|
|
ActiveTributary, handle_p2p,
|
|
tests::{
|
|
LocalP2p,
|
|
tributary::{new_keys, new_spec, new_tributaries},
|
|
},
|
|
};
|
|
|
|
#[tokio::test]
|
|
async fn handle_p2p_test() {
|
|
let keys = new_keys(&mut OsRng);
|
|
let spec = new_spec(&mut OsRng, &keys);
|
|
|
|
let mut tributaries = new_tributaries(&keys, &spec).await;
|
|
|
|
let mut tributary_arcs = vec![];
|
|
for (i, (p2p, tributary)) in tributaries.drain(..).enumerate() {
|
|
let tributary = Arc::new(RwLock::new(tributary));
|
|
tributary_arcs.push(tributary.clone());
|
|
tokio::spawn(handle_p2p(
|
|
Ristretto::generator() * *keys[i],
|
|
p2p,
|
|
Arc::new(RwLock::new(HashMap::from([(
|
|
spec.genesis(),
|
|
ActiveTributary { spec: spec.clone(), tributary },
|
|
)]))),
|
|
));
|
|
}
|
|
let tributaries = tributary_arcs;
|
|
|
|
// After two blocks of time, we should have a new block
|
|
// We don't wait one block of time as we may have missed the chance for this block
|
|
sleep(Duration::from_secs((2 * Tributary::<MemDb, Transaction, LocalP2p>::block_time()).into()))
|
|
.await;
|
|
let tip = tributaries[0].read().await.tip().await;
|
|
assert!(tip != spec.genesis());
|
|
|
|
// Sleep one second to make sure this block propagates
|
|
sleep(Duration::from_secs(1)).await;
|
|
// Make sure every tributary has it
|
|
for tributary in &tributaries {
|
|
assert!(tributary.read().await.reader().block(&tip).is_some());
|
|
}
|
|
|
|
// Then after another block of time, we should have yet another new block
|
|
sleep(Duration::from_secs(Tributary::<MemDb, Transaction, LocalP2p>::block_time().into())).await;
|
|
let new_tip = tributaries[0].read().await.tip().await;
|
|
assert!(new_tip != tip);
|
|
sleep(Duration::from_secs(1)).await;
|
|
for tributary in tributaries {
|
|
assert!(tributary.read().await.reader().block(&new_tip).is_some());
|
|
}
|
|
}
|