leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Session is now properly cleaned up

Browse source
This commit is contained in:
Igor Katson 2024-08-21 17:25:00 +01:00
parent a5abe97735
commit d5ddf4d294
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
3 changed files with 21 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

13
crates/librqbit/src/tests/e2e.rs
View file

@ -38,7 +38,7 @@ async fn test_e2e_download() {
.unwrap();
// Wait to ensure everything is dropped.
tokio::time::sleep(Duration::from_secs(10)).await;
tokio::time::sleep(Duration::from_secs(1)).await;
let metrics = tokio::runtime::Handle::current().metrics();
assert_eq!(metrics.num_alive_tasks(), 1);
@ -157,12 +157,13 @@ async fn _test_e2e_download(drop_checks: &DropChecks) {
}
}
info!("torrent is live");
Ok::<_, anyhow::Error>(SocketAddr::new(
let addr = SocketAddr::new(
std::net::IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)),
session
.tcp_listen_port()
.context("expected session.tcp_listen_port() to be set")?,
))
);
Ok::<_, anyhow::Error>((session, addr))
}
.instrument(error_span!("server", id = i)),
);
@ -170,12 +171,15 @@ async fn _test_e2e_download(drop_checks: &DropChecks) {
}
let mut peers = Vec::new();
// This is around just not to drop.
let mut _servers = Vec::new();
for (id, peer) in futures::future::join_all(futs)
.await
.into_iter()
.enumerate()
{
let peer = peer
let (server, peer) = peer
.with_context(|| format!("join error, server={id}"))
.unwrap()
.with_context(|| format!("timeout, server={id}"))
@ -183,6 +187,7 @@ async fn _test_e2e_download(drop_checks: &DropChecks) {
.with_context(|| format!("server couldn't start, server={id}"))
.unwrap();
peers.push(peer);
_servers.push(server);
}
info!("started all servers, starting client");