From c1775e45ebe8c7e9c808f767babfe8634aa7a8bf Mon Sep 17 00:00:00 2001 From: Igor Katson Date: Mon, 19 Aug 2024 11:13:16 +0100 Subject: [PATCH] E2E_NUM_SERVERS env var --- crates/librqbit/src/tests/e2e.rs | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/crates/librqbit/src/tests/e2e.rs b/crates/librqbit/src/tests/e2e.rs index 4c5cf0c..abce8f9 100644 --- a/crates/librqbit/src/tests/e2e.rs +++ b/crates/librqbit/src/tests/e2e.rs @@ -41,14 +41,17 @@ async fn test_e2e_download() { .await .unwrap(); - let num_servers = 128; + let num_servers = std::env::var("E2E_NUM_SERVERS") + .ok() + .and_then(|v| v.parse().ok()) + .unwrap_or(128u8); let torrent_file_bytes = torrent_file.as_bytes().unwrap(); let mut futs = Vec::new(); // 2. Start N servers that are serving that torrent, and return their IP:port combos. // Disable DHT on each. - for i in 0u8..num_servers { + for i in 0..num_servers { let torrent_file_bytes = torrent_file_bytes.clone(); let tempdir = tempdir.path().to_owned(); let fut = spawn(