pezkuwi-telemetry/backend/test_utils/src/workspace/commands.rs

//! Commands that we can use when running `cargo test` style tests in this workspace
//! that want to test the current code.
use crate::server::Command;
use std::path::PathBuf;

/// Runs `cargo run` in the current workspace to start up a telemetry shard process.
///
/// Note: The CWD must be somewhere within this backend workspace for the command to work.
pub fn cargo_run_telemetry_shard(release_mode: bool) -> Result<Command, std::io::Error> {
    telemetry_command("telemetry_shard", release_mode)
}

/// Runs `cargo run` in the current workspace to start up a telemetry core process.
///
/// Note: The CWD must be somewhere within this backend workspace for the command to work.
pub fn cargo_run_telemetry_core(release_mode: bool) -> Result<Command, std::io::Error> {
    telemetry_command("telemetry_core", release_mode)
}

fn telemetry_command(bin: &'static str, release_mode: bool) -> Result<Command, std::io::Error> {
    let mut workspace_dir = try_find_workspace_dir()?;
    workspace_dir.push("Cargo.toml");

    let mut cmd = Command::new("cargo").arg("run");

    // Release mode?
    if release_mode {
        cmd = cmd.arg("--release");
    }

    cmd = cmd.arg("--bin")
        .arg(bin)
        .arg("--manifest-path")
        .arg(workspace_dir)
        .arg("--");

    Ok(cmd)
}

/// A _very_ naive way to find the workspace ("backend") directory
/// from the current path (which is assumed to be inside it).
fn try_find_workspace_dir() -> Result<PathBuf, std::io::Error> {
    let mut dir = std::env::current_dir()?;
    while !dir.ends_with("backend") && dir.pop() {}
    Ok(dir)
}