frontend-utils: Fix flaky socket tests

This commit is contained in:
Nathan Adams 2024-04-13 23:44:35 +02:00
parent 6dbef5170e
commit c10d8b97b0
1 changed files with 3 additions and 6 deletions

View File

@ -702,16 +702,14 @@ mod tests {
Connect(dummy_handle!(), ConnectionState::Connected), Connect(dummy_handle!(), ConnectionState::Connected),
); );
write_server(&mut server_socket, "Hello ").await; write_server(&mut server_socket, "Hello World!").await;
write_server(&mut server_socket, "World!").await;
assert_next_socket_actions!( assert_next_socket_actions!(
client_read; client_read;
Data(dummy_handle!(), "Hello World!".as_bytes().to_vec()), Data(dummy_handle!(), "Hello World!".as_bytes().to_vec()),
); );
write_client(&client_write, "Hello from").await; write_client(&client_write, "Hello from client").await;
write_client(&client_write, " client").await;
assert_eq!(read_server(&mut server_socket).await, "Hello from client"); assert_eq!(read_server(&mut server_socket).await, "Hello from client");
@ -736,8 +734,7 @@ mod tests {
Connect(dummy_handle!(), ConnectionState::Connected), Connect(dummy_handle!(), ConnectionState::Connected),
); );
write_client(&client_write, "Sending some").await; write_client(&client_write, "Sending some data").await;
write_client(&client_write, " data").await;
client_write.close(); client_write.close();
assert_eq!(read_server(&mut server_socket).await, "Sending some data"); assert_eq!(read_server(&mut server_socket).await, "Sending some data");