test: extend block_exchange tests

Signed-off-by: ljedrz <ljedrz@gmail.com>

tests/block_exchange.rs

@@ -7,29 +7,41 @@ use tokio::time::timeout;
 mod common;
 use common::{spawn_nodes, Topology};
 
-#[tokio::test(max_threads = 1)]
-async fn exchange_block() {
-    tracing_subscriber::fmt::init();
-
+fn create_block() -> Block {
     let data = b"hello block\n".to_vec().into_boxed_slice();
     let cid = Cid::new_v1(Codec::Raw, Sha2_256::digest(&data));
 
+    Block { cid, data }
+}
+
+// verify that a put block can be received via get_block and the data matches
+#[tokio::test(max_threads = 1)]
+async fn two_node_put_get() {
     let nodes = spawn_nodes(2, Topology::Line).await;
+    let block = create_block();
 
-    nodes[0]
-        .put_block(Block {
-            cid: cid.clone(),
-            data: data.clone(),
-        })
-        .await
-        .unwrap();
-
-    let f = timeout(Duration::from_secs(10), nodes[1].get_block(&cid));
-
-    let Block { data: data2, .. } = f
+    nodes[0].put_block(block.clone()).await.unwrap();
+    let found_block = timeout(Duration::from_secs(10), nodes[1].get_block(&block.cid))
         .await
         .expect("get_block did not complete in time")
         .unwrap();
 
-    assert_eq!(data, data2);
+    assert_eq!(block.data, found_block.data);
+}
+
+// check that a long line of nodes still works with get_block
+#[tokio::test(max_threads = 1)]
+async fn long_get_block() {
+    // this number could be higher, but it starts hanging above ~24
+    const N: usize = 10;
+    let nodes = spawn_nodes(N, Topology::Line).await;
+    let block = create_block();
+
+    // the first node should get the block from the last one...
+    nodes[N - 1].put_block(block.clone()).await.unwrap();
+    nodes[0].get_block(&block.cid).await.unwrap();
+
+    // ...and the last one from the first one
+    nodes[0].put_block(block.clone()).await.unwrap();
+    nodes[N - 1].get_block(&block.cid).await.unwrap();
 }