test(dgw): add AI Gateway smoke tests (#1590)

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: CBenoit <3809077+CBenoit@users.noreply.github.com>

Author: Copilot

testsuite/src/dgw_config.rs

@@ -21,6 +21,23 @@ impl fmt::Display for VerbosityProfile {
     }
 }
 
+/// Configuration for the AI Gateway feature in tests.
+#[derive(Clone, Default, TypedBuilder)]
+pub struct AiGatewayConfig {
+    /// Whether the AI gateway is enabled.
+    #[builder(default = false)]
+    pub enabled: bool,
+    /// Optional API key for gateway-level authentication.
+    #[builder(default, setter(into))]
+    pub gateway_api_key: Option<String>,
+    /// Custom endpoint for the OpenAI provider (for mock server).
+    #[builder(default, setter(into))]
+    pub openai_endpoint: Option<String>,
+    /// API key for OpenAI provider.
+    #[builder(default, setter(into))]
+    pub openai_api_key: Option<String>,
+}
+
 #[derive(TypedBuilder)]
 pub struct DgwConfig {
     #[builder(default, setter(into))]
@@ -31,6 +48,12 @@ pub struct DgwConfig {
     disable_token_validation: bool,
     #[builder(default = VerbosityProfile::DEFAULT)]
     verbosity_profile: VerbosityProfile,
+    /// AI gateway configuration (requires enable_unstable: true to work).
+    #[builder(default, setter(into))]
+    ai_gateway: Option<AiGatewayConfig>,
+    /// Enable unstable features (required for AI gateway).
+    #[builder(default = false)]
+    enable_unstable: bool,
 }
 
 fn find_unused_port() -> u16 {
@@ -60,6 +83,8 @@ impl DgwConfigHandle {
             http_port,
             disable_token_validation,
             verbosity_profile,
+            ai_gateway,
+            enable_unstable,
         } = config;
 
         let tempdir = tempfile::tempdir().context("create tempdir")?;
@@ -68,26 +93,71 @@ impl DgwConfigHandle {
         let tcp_port = tcp_port.unwrap_or_else(find_unused_port);
         let http_port = http_port.unwrap_or_else(find_unused_port);
 
+        // Build AI gateway config JSON if provided.
+        let ai_gateway_json = if let Some(ai_config) = ai_gateway {
+            let gateway_api_key_json = ai_config
+                .gateway_api_key
+                .map(|k| format!(r#""GatewayApiKey": "{k}","#))
+                .unwrap_or_default();
+
+            let openai_provider_json = if ai_config.openai_endpoint.is_some() || ai_config.openai_api_key.is_some() {
+                let endpoint_json = ai_config
+                    .openai_endpoint
+                    .map(|e| format!(r#""Endpoint": "{e}""#))
+                    .unwrap_or_default();
+                let api_key_json = ai_config
+                    .openai_api_key
+                    .map(|k| format!(r#""ApiKey": "{k}""#))
+                    .unwrap_or_default();
+                let comma = if !endpoint_json.is_empty() && !api_key_json.is_empty() {
+                    ","
+                } else {
+                    ""
+                };
+                format!(
+                    r#""Providers": {{
+                        "Openai": {{ {endpoint_json}{comma}{api_key_json} }}
+                    }},"#
+                )
+            } else {
+                String::new()
+            };
+
+            format!(
+                r#",
+    "AiGateway": {{
+        "Enabled": {},
+        {gateway_api_key_json}
+        {openai_provider_json}
+        "RequestTimeoutSecs": 30
+    }}"#,
+                ai_config.enabled
+            )
+        } else {
+            String::new()
+        };
+
         let config = format!(
-            "{{
-    \"ProvisionerPublicKeyData\": {{
-        \"Value\": \"mMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA4vuqLOkl1pWobt6su1XO9VskgCAwevEGs6kkNjJQBwkGnPKYLmNF1E/af1yCocfVn/OnPf9e4x+lXVyZ6LMDJxFxu+axdgOq3Ld392J1iAEbfvwlyRFnEXFOJNyylqg3bY6LvnWHL/XZczVdMD9xYfq2sO9bg3xjRW4s7r9EEYOFjqVT3VFznH9iWJVtcSEKukmS/3uKoO6lGhacvu0HgjXXdgq0R8zvR4XRJ9Fcnf0f9Ypoc+i6L80NVjrRCeVOH+Ld/2fA9bocpfLarcVqG3RjS+qgOtpyCc0jWVFF4zaGQ7LUDFkEIYILkICeMMn2ll29hmZNzsJzZJ9s6NocgQIDAQAB\"
+            r#"{{
+    "ProvisionerPublicKeyData": {{
+        "Value": "mMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA4vuqLOkl1pWobt6su1XO9VskgCAwevEGs6kkNjJQBwkGnPKYLmNF1E/af1yCocfVn/OnPf9e4x+lXVyZ6LMDJxFxu+axdgOq3Ld392J1iAEbfvwlyRFnEXFOJNyylqg3bY6LvnWHL/XZczVdMD9xYfq2sO9bg3xjRW4s7r9EEYOFjqVT3VFznH9iWJVtcSEKukmS/3uKoO6lGhacvu0HgjXXdgq0R8zvR4XRJ9Fcnf0f9Ypoc+i6L80NVjrRCeVOH+Ld/2fA9bocpfLarcVqG3RjS+qgOtpyCc0jWVFF4zaGQ7LUDFkEIYILkICeMMn2ll29hmZNzsJzZJ9s6NocgQIDAQAB"
     }},
-    \"Listeners\": [
+    "Listeners": [
         {{
-            \"InternalUrl\": \"tcp://127.0.0.1:{tcp_port}\",
-            \"ExternalUrl\": \"tcp://127.0.0.1:{tcp_port}\"
+            "InternalUrl": "tcp://127.0.0.1:{tcp_port}",
+            "ExternalUrl": "tcp://127.0.0.1:{tcp_port}"
         }},
         {{
-            \"InternalUrl\": \"http://127.0.0.1:{http_port}\",
-            \"ExternalUrl\": \"http://127.0.0.1:{http_port}\"
+            "InternalUrl": "http://127.0.0.1:{http_port}",
+            "ExternalUrl": "http://127.0.0.1:{http_port}"
         }}
     ],
-    \"VerbosityProfile\": \"{verbosity_profile}\",
-    \"__debug__\": {{
-        \"disable_token_validation\": {disable_token_validation}
-    }}
-}}"
+    "VerbosityProfile": "{verbosity_profile}",
+    "__debug__": {{
+        "disable_token_validation": {disable_token_validation},
+        "enable_unstable": {enable_unstable}
+    }}{ai_gateway_json}
+}}"#
         );
 
         std::fs::write(&config_path, config).with_context(|| format!("write config into {}", config_path.display()))?;