actix/actix-web
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-23 16:21:06 +01:00
Code Issues Releases Wiki Activity

test: fix test_h2_connection_drop spurious hang

Browse Source 
fixes #3061
This commit is contained in:
Rob Ede 2023-08-03 06:54:50 +01:00
parent 1c60978a89
commit 12dbda986e
No known key found for this signature in database
GPG Key ID: 97C636207D3EF933
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
awc/src/client/connection.rs
View File

@ -394,6 +394,8 @@ mod test {
#[actix_rt::test]
async fn test_h2_connection_drop() {
env_logger::try_init().ok();
let addr = "127.0.0.1:0".parse::<net::SocketAddr>().unwrap();
let listener = net::TcpListener::bind(addr).unwrap();
let local = listener.local_addr().unwrap();
@ -428,11 +430,18 @@ mod test {
if this.start_from.elapsed() > Duration::from_secs(10) {
panic!("connection should be gone and can not be ready");
} else {
let _ = this.interval.poll_tick(cx);
Poll::Pending
match this.interval.poll_tick(cx) {
Poll::Ready(_) => {
// prevents spurious test hang
this.interval.reset();
Poll::Pending
}
Poll::Pending => Poll::Pending,
}
}
}
Err(_) => Poll::Ready(()),
Err(err) => Poll::Ready(()),
}
}
}