From 97be7056d3740d6f266953d13c2018cb14de38e6 Mon Sep 17 00:00:00 2001 From: Pat Hickey Date: Fri, 1 Sep 2023 14:45:33 -0700 Subject: [PATCH] tcp tests: need a blocking_read to get messages --- crates/test-programs/wasi-sockets-tests/src/bin/tcp_v4.rs | 4 ++-- crates/test-programs/wasi-sockets-tests/src/bin/tcp_v6.rs | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v4.rs b/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v4.rs index 907b8986b7de..c0eb0afdc473 100644 --- a/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v4.rs +++ b/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v4.rs @@ -67,7 +67,7 @@ fn main() { assert!(empty_data.is_empty()); assert_eq!(status, streams::StreamStatus::Open); - let (data, status) = streams::read(input, first_message.len() as u64).unwrap(); + let (data, status) = streams::blocking_read(input, first_message.len() as u64).unwrap(); assert_eq!(status, streams::StreamStatus::Open); tcp::drop_tcp_socket(accepted); @@ -96,7 +96,7 @@ fn main() { wait(sub); let (accepted, input, output) = tcp::accept(sock).unwrap(); - let (data, status) = streams::read(input, second_message.len() as u64).unwrap(); + let (data, status) = streams::blocking_read(input, second_message.len() as u64).unwrap(); assert_eq!(status, streams::StreamStatus::Open); streams::drop_input_stream(input); diff --git a/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v6.rs b/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v6.rs index 47a569aed30c..36e465f0fac7 100644 --- a/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v6.rs +++ b/crates/test-programs/wasi-sockets-tests/src/bin/tcp_v6.rs @@ -69,7 +69,7 @@ fn main() { assert!(empty_data.is_empty()); assert_eq!(status, streams::StreamStatus::Open); - let (data, status) = streams::read(input, first_message.len() as u64).unwrap(); + let (data, status) = streams::blocking_read(input, first_message.len() as u64).unwrap(); assert_eq!(status, streams::StreamStatus::Open); tcp::drop_tcp_socket(accepted); @@ -98,7 +98,7 @@ fn main() { wait(sub); let (accepted, input, output) = tcp::accept(sock).unwrap(); - let (data, status) = streams::read(input, second_message.len() as u64).unwrap(); + let (data, status) = streams::blocking_read(input, second_message.len() as u64).unwrap(); assert_eq!(status, streams::StreamStatus::Open); streams::drop_input_stream(input);