Skip to content

Commit 17fddd3

Browse files
committed
test(client): unflake client try_send_request unit test
1 parent c68d424 commit 17fddd3

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

tests/client.rs

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2044,22 +2044,20 @@ mod conn {
20442044
#[tokio::test]
20452045
async fn test_try_send_request() {
20462046
use std::future::Future;
2047-
let (listener, addr) = setup_tk_test_server().await;
20482047
let (done_tx, done_rx) = tokio::sync::oneshot::channel::<()>();
2048+
let (io_srv, io_cli) = tokio_test::io::Builder::new()
2049+
.write(b"GET / HTTP/1.1\r\n\r\n")
2050+
.read(b"HTTP/1.1 200 OK\r\ncontent-length: 0\r\n\r\n")
2051+
.build_with_handle();
20492052

20502053
tokio::spawn(async move {
2051-
let mut sock = listener.accept().await.unwrap().0;
2052-
let mut buf = [0u8; 8192];
2053-
sock.read(&mut buf).await.expect("read 1");
2054-
sock.write_all(b"HTTP/1.1 200 OK\r\ncontent-length: 0\r\n\r\n")
2055-
.await
2056-
.expect("write 1");
2054+
let _io = io_cli;
20572055
let _ = done_rx.await;
20582056
});
20592057

20602058
// make polling fair by putting both in spawns
20612059
tokio::spawn(async move {
2062-
let io = tcp_connect(&addr).await.expect("tcp connect");
2060+
let io = TokioIo::new(io_srv);
20632061
let (mut client, mut conn) = conn::http1::Builder::new()
20642062
.handshake::<_, Empty<Bytes>>(io)
20652063
.await

0 commit comments

Comments
 (0)