use cid::{Cid, Codec};
use ipfs::Block;
use multihash::Sha2_256;
use std::time::Duration;
use tokio::time;
mod common;
use common::{spawn_nodes, Topology};
#[tokio::test(max_threads = 1)]
async fn check_bitswap_cleanups() {
let mut nodes = spawn_nodes(3, Topology::Star).await;
let bitswap_peers = nodes[0].get_bitswap_peers().await.unwrap();
assert_eq!(bitswap_peers.len(), 2);
if let Some(node) = nodes.pop() {
node.shutdown().await;
time::delay_for(Duration::from_millis(200)).await;
}
let bitswap_peers = nodes[0].get_bitswap_peers().await.unwrap();
assert_eq!(bitswap_peers.len(), 1);
if let Some(node) = nodes.pop() {
node.shutdown().await;
time::delay_for(Duration::from_millis(200)).await;
}
let bitswap_peers = nodes[0].get_bitswap_peers().await.unwrap();
assert!(bitswap_peers.is_empty());
}
#[ignore]
#[tokio::test(max_threads = 1)]
async fn bitswap_stress_test() {
fn filter(i: usize) -> bool {
i % 2 == 0
}
tracing_subscriber::fmt::init();
let data = b"hello block\n".to_vec().into_boxed_slice();
let cid = Cid::new_v1(Codec::Raw, Sha2_256::digest(&data));
let nodes = spawn_nodes(5, Topology::Mesh).await;
for (i, node) in nodes.iter().enumerate() {
if filter(i) {
node.put_block(Block {
cid: cid.clone(),
data: data.clone(),
})
.await
.unwrap();
}
}
for (i, node) in nodes.iter().enumerate() {
if !filter(i) {
node.get_block(&cid).await.unwrap();
}
}
}