ci(runner): rewrite collection_runner.sh in rust (#1604)

crates/test_utils/src/main.rs

@@ -0,0 +1,156 @@
+use std::{
+    env,
+    process::{exit, Command as cmd},
+};
+
+use clap::{arg, command, Parser};
+use router::types::ConnectorAuthType;
+use test_utils::connector_auth::ConnectorAuthenticationMap;
+
+// Just by the name of the connector, this function generates the name of the collection
+// Example: CONNECTOR_NAME="stripe" -> OUTPUT: postman/stripe.postman_collection.json
+#[inline]
+fn path_generation(name: impl AsRef<str>) -> String {
+    format!("postman/{}.postman_collection.json", name.as_ref())
+}
+
+#[derive(Parser)]
+#[command(version, about = "Postman collection runner using newman!", long_about = None)]
+struct Args {
+    /// Name of the connector
+    #[arg(short, long = "connector_name")]
+    connector_name: String,
+    /// Base URL of the Hyperswitch environment
+    #[arg(short, long = "base_url")]
+    base_url: String,
+    /// Admin API Key of the environment
+    #[arg(short, long = "admin_api_key")]
+    admin_api_key: String,
+}
+
+fn main() {
+    let args = Args::parse();
+
+    let connector_name = args.connector_name;
+    let base_url = args.base_url;
+    let admin_api_key = args.admin_api_key;
+
+    let collection_path = path_generation(&connector_name);
+    let auth_map = ConnectorAuthenticationMap::new();
+
+    let inner_map = auth_map.inner();
+
+    // Newman runner
+    // Depending on the conditions satisfied, variables are added. Since certificates of stripe have already
+    // been added to the postman collection, those conditions are set to true and collections that have
+    // variables set up for certificate, will consider those variables and will fail.
+
+    let mut newman_command = cmd::new("newman");
+
+    newman_command.args(["run", &collection_path]);
+    newman_command.args(["--env-var", &format!("admin_api_key={admin_api_key}")]);
+    newman_command.args(["--env-var", &format!("baseUrl={base_url}")]);
+
+    if let Some(auth_type) = inner_map.get(&connector_name) {
+        match auth_type {
+            ConnectorAuthType::HeaderKey { api_key } => {
+                newman_command.args(["--env-var", &format!("connector_api_key={api_key}")]);
+            }
+            ConnectorAuthType::BodyKey { api_key, key1 } => {
+                newman_command.args([
+                    "--env-var",
+                    &format!("connector_api_key={api_key}"),
+                    "--env-var",
+                    &format!("connector_key1={key1}"),
+                ]);
+            }
+            ConnectorAuthType::SignatureKey {
+                api_key,
+                key1,
+                api_secret,
+            } => {
+                newman_command.args([
+                    "--env-var",
+                    &format!("connector_api_key={api_key}"),
+                    "--env-var",
+                    &format!("connector_key1={key1}"),
+                    "--env-var",
+                    &format!("connector_api_secret={api_secret}"),
+                ]);
+            }
+            ConnectorAuthType::MultiAuthKey {
+                api_key,
+                key1,
+                key2,
+                api_secret,
+            } => {
+                newman_command.args([
+                    "--env-var",
+                    &format!("connector_api_key={api_key}"),
+                    "--env-var",
+                    &format!("connector_key1={key1}"),
+                    "--env-var",
+                    &format!("connector_key1={key2}"),
+                    "--env-var",
+                    &format!("connector_api_secret={api_secret}"),
+                ]);
+            }
+            // Handle other ConnectorAuthType variants
+            _ => {
+                eprintln!("Invalid authentication type.");
+            }
+        }
+    } else {
+        eprintln!("Connector not found.");
+    }
+
+    // Add additional environment variables if present
+    if let Ok(gateway_merchant_id) = env::var("GATEWAY_MERCHANT_ID") {
+        newman_command.args([
+            "--env-var",
+            &format!("gateway_merchant_id={gateway_merchant_id}"),
+        ]);
+    }
+
+    if let Ok(gpay_certificate) = env::var("GPAY_CERTIFICATE") {
+        newman_command.args(["--env-var", &format!("certificate={gpay_certificate}")]);
+    }
+
+    if let Ok(gpay_certificate_keys) = env::var("GPAY_CERTIFICATE_KEYS") {
+        newman_command.args([
+            "--env-var",
+            &format!("certificate_keys={gpay_certificate_keys}"),
+        ]);
+    }
+
+    newman_command.arg("--delay-request").arg("5");
+
+    // Execute the newman command
+    let output = newman_command.spawn();
+    let mut child = match output {
+        Ok(child) => child,
+        Err(err) => {
+            eprintln!("Failed to execute command: {err}");
+            exit(1);
+        }
+    };
+    let status = child.wait();
+
+    let exit_code = match status {
+        Ok(exit_status) => {
+            if exit_status.success() {
+                println!("Command executed successfully!");
+                exit_status.code().unwrap_or(0)
+            } else {
+                eprintln!("Command failed with exit code: {:?}", exit_status.code());
+                exit_status.code().unwrap_or(1)
+            }
+        }
+        Err(err) => {
+            eprintln!("Failed to wait for command execution: {}", err);
+            exit(1);
+        }
+    };
+
+    exit(exit_code);
+}