Add flag for overriding API environment in tests

1 files changed, 11 insertions, 1 deletions

diff --git a/test/test-manager/src/main.rs b/test/test-manager/src/main.rs
index 88203cd8c2..8628502eff 100644
--- a/test/test-manager/src/main.rs
+++ b/test/test-manager/src/main.rs
@@ -12,7 +12,7 @@ mod vm;
 use std::{net::SocketAddr, path::PathBuf};
 
 use anyhow::{Context, Result};
-use clap::Parser;
+use clap::{builder::PossibleValuesParser, Parser};
 use tests::{config::TEST_CONFIG, get_filtered_tests};
 use vm::provision;
 
@@ -74,6 +74,11 @@ enum Commands {
         #[arg(long, group = "display_args")]
         display: bool,
 
+        /// API and conncheck environment to use. The domain name will be prefixed with "api." and
+        /// "ipv4.am.i.".
+        #[arg(long, value_parser = PossibleValuesParser::new(&["mullvad.net", "stagemole.eu", "devmole.eu"]))]
+        mullvad_host: Option<String>,
+
         /// Run VNC server on a specified port
         #[arg(long, group = "display_args")]
         vnc: Option<u16>,
@@ -236,6 +241,7 @@ async fn main() -> Result<()> {
         Commands::RunTests {
             vm,
             display,
+            mullvad_host,
             vnc,
             account,
             app_package,
@@ -256,6 +262,10 @@ async fn main() -> Result<()> {
                 (true, true) => unreachable!("invalid combination"),
             };
 
+            if let Some(mullvad_host) = mullvad_host {
+                log::trace!("Setting Mullvad host using --mullvad-host flag");
+                config.mullvad_host = Some(mullvad_host);
+            }
             let mullvad_host = config.get_host();
             log::debug!("Mullvad host: {mullvad_host}");