From fa45424bb3f8208db024f789bf91a7df24afd970 Mon Sep 17 00:00:00 2001
From: Roman <zajic@zajic.net>
Date: Tue, 3 Sep 2024 18:31:50 +0800
Subject: [PATCH] test: prepare to call stream disperse

---
 .../protocols/dispersal/executor/behaviour.rs | 55 +++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/nomos-da/network/core/src/protocols/dispersal/executor/behaviour.rs b/nomos-da/network/core/src/protocols/dispersal/executor/behaviour.rs
index 242791054..107e16a5e 100644
--- a/nomos-da/network/core/src/protocols/dispersal/executor/behaviour.rs
+++ b/nomos-da/network/core/src/protocols/dispersal/executor/behaviour.rs
@@ -235,6 +235,7 @@ where
         message: DaBlob,
         subnetwork_id: SubnetworkId,
     ) -> Result<StreamHandlerFutureSuccess, DispersalError> {
+        println!("I have been called!");
         let blob_id = message.id();
         let blob = bincode::serialize(&message).map_err(|error| DispersalError::Serialization {
             error,
@@ -485,3 +486,57 @@ impl<M: MembershipHandler<Id = PeerId, NetworkId = SubnetworkId> + 'static> Netw
         }
     }
 }
+
+#[cfg(test)]
+pub mod test {
+    use crate::protocols::dispersal::executor::behaviour::DispersalExecutorBehaviour;
+    use crate::test_utils::AllNeighbours;
+    use crate::SubnetworkId;
+    use kzgrs_backend::common::blob::DaBlob;
+    use kzgrs_backend::common::{Chunk, Column};
+    use libp2p::PeerId;
+
+    #[tokio::test]
+    async fn test_stream_disperse_error_cases() {
+        let k1 = libp2p::identity::Keypair::generate_ed25519();
+        let k2 = libp2p::identity::Keypair::generate_ed25519();
+        let _validator_peer = PeerId::from_public_key(&k2.public());
+        let neighbours = AllNeighbours {
+            neighbours: [
+                PeerId::from_public_key(&k1.public()),
+                PeerId::from_public_key(&k2.public()),
+            ]
+            .into_iter()
+            .collect(),
+        };
+        let mut executor = DispersalExecutorBehaviour::new(neighbours.clone());
+        let chunk = Chunk(Vec::with_capacity(10));
+        let chunks: Vec<Chunk> = (0..100).map(|_| chunk.clone()).collect();
+        let column = Column(chunks.clone());
+        let blob = DaBlob {
+            column_idx: 0,
+            column,
+            column_commitment: Default::default(),
+            aggregated_column_commitment: Default::default(),
+            aggregated_column_proof: Default::default(),
+            rows_commitments: vec![],
+            rows_proofs: vec![],
+        };
+
+        if let Some(stream) = executor
+            .idle_streams
+            .remove(&PeerId::from_public_key(&k1.public()))
+        {
+            let network_id = SubnetworkId::from(0u32);
+            let dispersed = DispersalExecutorBehaviour::<AllNeighbours>::stream_disperse(
+                stream,
+                blob.clone(),
+                network_id,
+            )
+            .await
+            .is_ok();
+
+            assert!(dispersed);
+        };
+    }
+}