test fix

holochain · Oct 16, 2024 · c511c65 · c511c65
1 parent 6c37511
commit c511c65
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/crates/tx5/src/test.rs b/crates/tx5/src/test.rs
@@ -1,5 +1,7 @@
 use crate::*;
 
+const DISCONNECT: &[u8] = b"<<<test-disconnect>>>";
+
 struct TestEp {
     ep: Endpoint,
     task: tokio::task::JoinHandle<()>,
@@ -48,7 +50,7 @@ impl TestEp {
                     }
                     EndpointEvent::Disconnected { peer_url } => {
                         if send
-                            .send((peer_url, b"<<<test-disconnect>>>".to_vec()))
+                            .send((peer_url, DISCONNECT.to_vec()))
                             .is_err()
                         {
                             break;
@@ -469,6 +471,10 @@ async fn ep_messages_contiguous() {
     loop {
         let (peer_url, message) = dest_recv.recv().await.unwrap();
 
+        if message == DISCONNECT {
+            continue;
+        }
+
         assert_eq!(((16 * 1024) - 4) * CHUNK_COUNT, message.len());
 
         for chunk_id in 0..CHUNK_COUNT {