Add flag for overriding API environment in tests

Author: dlon

test/test-manager/src/main.rs

@@ -74,6 +74,11 @@ enum Commands {
         #[arg(long, group = "display_args")]
         display: bool,
 
+        /// API and conncheck environment to use, such as `mullvad.net` or `stagemole.eu`.
+        /// The domain name will be prefixed with `api.` and `ipv4.am.i.`
+        #[arg(long)]
+        environment: 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,
+            environment,
             vnc,
             account,
             app_package,
@@ -256,6 +262,10 @@ async fn main() -> Result<()> {
                 (true, true) => unreachable!("invalid combination"),
             };
 
+            if let Some(environment) = environment {
+                // Override environment using flag
+                config.mullvad_host = Some(environment);
+            }
             let mullvad_host = config.get_host();
             log::debug!("Mullvad host: {mullvad_host}");