Add/test max_nodes_per_connection feature

backend/test_utils/src/workspace/start_server.rs

@@ -1,7 +1,7 @@
 use super::commands;
 use crate::server::{self, Server, Command};
 
-/// Additional options to pass to the feed command.
+/// Additional options to pass to the core command.
 pub struct CoreOpts {
     pub feed_timeout: Option<u64>
 }
@@ -14,6 +14,19 @@ impl Default for CoreOpts {
     }
 }
 
+/// Additional options to pass to the shard command.
+pub struct ShardOpts {
+    pub max_nodes_per_connection: Option<usize>
+}
+
+impl Default for ShardOpts {
+    fn default() -> Self {
+        Self {
+            max_nodes_per_connection: None
+        }
+    }
+}
+
 /// Start a telemetry server. We'll use `cargo run` by default, but you can also provide
 /// env vars to configure the binary that runs for the shard and core process. Either:
 ///
@@ -31,7 +44,7 @@ impl Default for CoreOpts {
 /// - `TELEMETRY_SUBMIT_HOSTS` - hosts (comma separated) to connect to for telemetry `/submit`s.
 /// - `TELEMETRY_FEED_HOST` - host to connect to for feeds (eg 127.0.0.1:3000)
 ///
-pub async fn start_server(release_mode: bool, core_opts: CoreOpts) -> Server {
+pub async fn start_server(release_mode: bool, core_opts: CoreOpts, shard_opts: ShardOpts) -> Server {
     // Start to a single process:
     if let Ok(bin) = std::env::var("TELEMETRY_BIN") {
         return Server::start(server::StartOpts::SingleProcess {
@@ -52,10 +65,17 @@ pub async fn start_server(release_mode: bool, core_opts: CoreOpts) -> Server {
     }
 
     // Build the shard command
-    let shard_command = std::env::var("TELEMETRY_SHARD_BIN")
+    let mut shard_command = std::env::var("TELEMETRY_SHARD_BIN")
         .map(|val| Command::new(val))
         .unwrap_or_else(|_| commands::cargo_run_telemetry_shard(release_mode).expect("must be in rust workspace to run shard command"));
 
+    // Append additional opts to the shard command
+    if let Some(max_nodes_per_connection) = shard_opts.max_nodes_per_connection {
+        shard_command = shard_command
+            .arg("--max-nodes-per-connection")
+            .arg(max_nodes_per_connection.to_string());
+    }
+
     // Build the core command
     let mut core_command = std::env::var("TELEMETRY_CORE_BIN")
         .map(|val| Command::new(val))
@@ -77,10 +97,10 @@ pub async fn start_server(release_mode: bool, core_opts: CoreOpts) -> Server {
 
 /// Start a telemetry core server in debug mode. see [`start_server`] for details.
 pub async fn start_server_debug() -> Server {
-    start_server(false, CoreOpts::default()).await
+    start_server(false, CoreOpts::default(), ShardOpts::default()).await
 }
 
 /// Start a telemetry core server in release mode. see [`start_server`] for details.
 pub async fn start_server_release() -> Server {
-    start_server(true, CoreOpts::default()).await
+    start_server(true, CoreOpts::default(), ShardOpts::default()).await
 }