Mirrors/chatmail-core
1
0
Fork 0
mirror of https://github.com/chatmail/core.git synced 2026-08-14 21:03:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

test: add basic tests for run_connection_attempts

Browse Source
This commit is contained in:
link2xt
2026-07-24 18:18:39 +00:00
parent ebff8ce455
commit f9dbc23336
1 changed files with 40 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/net.rs
View File

@@ -264,3 +264,43 @@ pub(crate) async fn connect_tcp(
.map(connect_tcp_inner);
run_connection_attempts(connection_futures).await
}
#[cfg(test)]
mod tests {
use super::*;
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn test_run_connection_attempts() {
let futures: Vec<Pin<Box<dyn Future<Output = Result<u32>> + Send>>> =
vec![Box::pin(async { Ok(1) }), Box::pin(async { Ok(2) })];
assert_eq!(
run_connection_attempts(futures.into_iter()).await.unwrap(),
1
);
let futures: Vec<Pin<Box<dyn Future<Output = Result<u32>> + Send>>> = vec![
Box::pin(async { Err(format_err!("fail")) }),
Box::pin(async { Ok(2) }),
];
assert_eq!(
run_connection_attempts(futures.into_iter()).await.unwrap(),
2
);
let futures: Vec<Pin<Box<dyn Future<Output = Result<u32>> + Send>>> = vec![
Box::pin(async { Err(format_err!("fail")) }),
Box::pin(async { Err(format_err!("fail")) }),
Box::pin(async { Err(format_err!("fail")) }),
Box::pin(async { Err(format_err!("fail")) }),
Box::pin(async { Err(format_err!("fail")) }),
Box::pin(async { Err(format_err!("last")) }),
];
assert!(
run_connection_attempts(futures.into_iter())
.await
.unwrap_err()
.to_string()
.contains("last"),
);
}
}