Updated tests

andrewwhitehead · Nov 24, 2020 · 8b0a499 · 8b0a499
1 parent 7b916ad
commit 8b0a499
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/rep.rs b/src/rep.rs
@@ -149,7 +149,7 @@ impl BlockingRecv for RepSocket {
                     return Ok(messages.pop().unwrap());
                 }
                 Some((_peer_id, _)) => todo!(),
-                None => todo!(),
+                None => return Err(ZmqError::NoMessage),
             };
         }
     }

diff --git a/tests/pub_sub.rs b/tests/pub_sub.rs
@@ -36,6 +36,7 @@ async fn test_pub_sub_sockets() {
 
                 pub_socket
                     .send(cloned_payload.clone().into())
+                    .await
                     .expect("Failed to send");
                 tokio::time::delay_for(Duration::from_millis(1)).await;
             }

diff --git a/tests/req_rep.rs b/tests/req_rep.rs
@@ -10,7 +10,9 @@ async fn run_rep_server(mut rep_socket: RepSocket) -> Result<(), Box<dyn Error>>
 
     for i in 0..10i32 {
         let mess: String = rep_socket.recv().await?.try_into()?;
-        rep_socket.send(format!("{} Rep - {}", mess, i).into())?;
+        rep_socket
+            .send(format!("{} Rep - {}", mess, i).into())
+            .await?;
     }
     // yield for a moment to ensure that server has some time to flush socket
     let errs = rep_socket.close().await;
@@ -73,7 +75,7 @@ async fn test_many_req_rep_sockets() -> Result<(), Box<dyn Error>> {
 
     for _ in 0..10000i32 {
         let mess: String = rep_socket.recv().await?.try_into()?;
-        rep_socket.send(format!("{} Rep", mess).into())?;
+        rep_socket.send(format!("{} Rep", mess).into()).await?;
     }
     Ok(())
 }