forked from lx/netapp
Add a modest integration test
This commit is contained in:
parent
7eea46dcf3
commit
dd881e2e60
4 changed files with 115 additions and 2 deletions
|
@ -39,3 +39,4 @@ steps:
|
|||
- cargo build-all-features
|
||||
- cargo build --example fullmesh
|
||||
- cargo build --example basalt --features "basalt"
|
||||
- cargo test --all-features -- --test-threads 1
|
||||
|
|
|
@ -25,5 +25,8 @@ mod server;
|
|||
pub mod netapp;
|
||||
pub mod peering;
|
||||
|
||||
pub use netapp::*;
|
||||
pub use crate::netapp::*;
|
||||
pub use util::NodeID;
|
||||
|
||||
#[cfg(test)]
|
||||
mod test;
|
||||
|
|
109
src/test.rs
Normal file
109
src/test.rs
Normal file
|
@ -0,0 +1,109 @@
|
|||
use std::net::SocketAddr;
|
||||
use std::sync::{Arc};
|
||||
use std::time::{Duration};
|
||||
|
||||
use tokio::select;
|
||||
use tokio::sync::watch;
|
||||
|
||||
use sodiumoxide::crypto::auth;
|
||||
use sodiumoxide::crypto::sign::ed25519;
|
||||
|
||||
use crate::peering::fullmesh::*;
|
||||
use crate::netapp::*;
|
||||
use crate::NodeID;
|
||||
|
||||
|
||||
#[tokio::test(flavor="current_thread")]
|
||||
async fn test_with_basic_scheduler() {
|
||||
run_test().await
|
||||
}
|
||||
|
||||
#[tokio::test(flavor="multi_thread", worker_threads=4)]
|
||||
async fn test_with_threaded_scheduler() {
|
||||
run_test().await
|
||||
}
|
||||
|
||||
async fn run_test() {
|
||||
select! {
|
||||
_ = run_test_inner() => (),
|
||||
_ = tokio::time::sleep(Duration::from_secs(20)) => panic!("timeout"),
|
||||
}
|
||||
}
|
||||
|
||||
async fn run_test_inner() {
|
||||
let netid = auth::gen_key();
|
||||
|
||||
let (pk1, sk1) = ed25519::gen_keypair();
|
||||
let (pk2, sk2) = ed25519::gen_keypair();
|
||||
let (pk3, sk3) = ed25519::gen_keypair();
|
||||
|
||||
let addr1: SocketAddr = "127.0.0.1:19991".parse().unwrap();
|
||||
let addr2: SocketAddr = "127.0.0.1:19992".parse().unwrap();
|
||||
let addr3: SocketAddr = "127.0.0.1:19993".parse().unwrap();
|
||||
|
||||
let (stop_tx, stop_rx) = watch::channel(false);
|
||||
|
||||
let (thread1, _netapp1, peering1) = run_netapp(netid.clone(), pk1, sk1, addr1, vec![], stop_rx.clone());
|
||||
tokio::time::sleep(Duration::from_secs(2)).await;
|
||||
|
||||
// Connect second node and check it peers with everyone
|
||||
let (thread2, _netapp2, peering2) = run_netapp(netid.clone(), pk2, sk2, addr2, vec![(pk1, addr1)], stop_rx.clone());
|
||||
tokio::time::sleep(Duration::from_secs(5)).await;
|
||||
|
||||
let pl1 = peering1.get_peer_list();
|
||||
println!("A pl1: {:?}", pl1);
|
||||
assert_eq!(pl1.len(), 2);
|
||||
|
||||
let pl2 = peering2.get_peer_list();
|
||||
println!("A pl2: {:?}", pl2);
|
||||
assert_eq!(pl2.len(), 2);
|
||||
|
||||
// Connect third ndoe and check it peers with everyone
|
||||
let (thread3, _netapp3, peering3) = run_netapp(netid, pk3, sk3, addr3, vec![(pk2, addr2)], stop_rx.clone());
|
||||
tokio::time::sleep(Duration::from_secs(5)).await;
|
||||
|
||||
let pl1 = peering1.get_peer_list();
|
||||
println!("B pl1: {:?}", pl1);
|
||||
assert_eq!(pl1.len(), 3);
|
||||
|
||||
let pl2 = peering2.get_peer_list();
|
||||
println!("B pl2: {:?}", pl2);
|
||||
assert_eq!(pl2.len(), 3);
|
||||
|
||||
let pl3 = peering3.get_peer_list();
|
||||
println!("B pl3: {:?}", pl3);
|
||||
assert_eq!(pl3.len(), 3);
|
||||
|
||||
// Send stop signal and wait for everyone to finish
|
||||
stop_tx.send(true).unwrap();
|
||||
thread1.await.unwrap();
|
||||
thread2.await.unwrap();
|
||||
thread3.await.unwrap();
|
||||
}
|
||||
|
||||
fn run_netapp(
|
||||
netid: auth::Key,
|
||||
_pk: NodeID,
|
||||
sk: ed25519::SecretKey,
|
||||
listen_addr: SocketAddr,
|
||||
bootstrap_peers: Vec<(NodeID, SocketAddr)>,
|
||||
must_exit: watch::Receiver<bool>)
|
||||
->
|
||||
(tokio::task::JoinHandle<()>, Arc<NetApp>, Arc<FullMeshPeeringStrategy>)
|
||||
{
|
||||
|
||||
let netapp = NetApp::new(netid, sk);
|
||||
let peering = FullMeshPeeringStrategy::new(netapp.clone(), bootstrap_peers);
|
||||
|
||||
let peering2 = peering.clone();
|
||||
let netapp2 = netapp.clone();
|
||||
let fut = tokio::spawn(async move {
|
||||
tokio::join!(
|
||||
netapp2.listen(listen_addr, None, must_exit.clone()),
|
||||
peering2.run(must_exit.clone()),
|
||||
);
|
||||
});
|
||||
|
||||
(fut, netapp, peering)
|
||||
}
|
||||
|
|
@ -27,7 +27,7 @@ where
|
|||
/// from a watcher that tells us when to exit.
|
||||
/// Usefull in a select statement to interrupt another
|
||||
/// future:
|
||||
/// ```
|
||||
/// ```ignore
|
||||
/// select!(
|
||||
/// _ = a_long_task() => Success,
|
||||
/// _ = await_exit(must_exit) => Interrupted,
|
||||
|
|
Loading…
Reference in a new issue