Fix some flaky tests

alexcrichton · Oct 9, 2023 · 1d0fc19 · 1d0fc19
1 parent 116673f
commit 1d0fc19
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/crates/test-programs/src/bin/preview2_tcp_sample_application.rs b/crates/test-programs/src/bin/preview2_tcp_sample_application.rs
@@ -25,7 +25,7 @@ fn test_sample_application(family: IpAddressFamily, bind_address: IpSocketAddres
     }
 
     {
-        let (_accepted, input, _output) = listener.accept().unwrap();
+        let (_accepted, input, _output) = listener.blocking_accept().unwrap();
 
         let empty_data = input.read(0).unwrap();
         assert!(empty_data.is_empty());

diff --git a/crates/test-programs/src/bin/preview2_tcp_sockopts.rs b/crates/test-programs/src/bin/preview2_tcp_sockopts.rs
@@ -104,7 +104,7 @@ fn test_tcp_sockopt_inheritance(net: &Network, family: IpAddressFamily) {
     let bound_addr = listener.local_address().unwrap();
     let client = TcpSocket::new(family).unwrap();
     client.blocking_connect(&net, bound_addr).unwrap();
-    let (accepted_client, _, _) = listener.accept().unwrap();
+    let (accepted_client, _, _) = listener.blocking_accept().unwrap();
 
     // Verify options on accepted socket:
     {
@@ -158,7 +158,7 @@ fn test_tcp_sockopt_after_listen(net: &Network, family: IpAddressFamily) {
 
     let client = TcpSocket::new(family).unwrap();
     client.blocking_connect(&net, bound_addr).unwrap();
-    let (accepted_client, _, _) = listener.accept().unwrap();
+    let (accepted_client, _, _) = listener.blocking_accept().unwrap();
 
     // Verify options on accepted socket:
     {