merge

Author: evanliu048

crates/chat-cli/src/cli/mod.rs

@@ -171,6 +171,7 @@ pub struct Cli {
 impl Cli {
     pub async fn execute(self) -> Result<ExitCode> {
         let subcommand = self.subcommand.unwrap_or_default();
+
         // Initialize our logger and keep around the guard so logging can perform as expected.
         let _log_guard = initialize_logging(LogArgs {
             log_level: match self.verbose > 0 {
@@ -218,6 +219,7 @@ impl Cli {
         if subcommand.valid_for_telemetry() {
             telemetry.send_cli_subcommand_executed(&subcommand).ok();
         }
+
         let cli_context = CliContext::new();
 
         let result = match subcommand {

crates/chat-cli/src/telemetry/cognito.rs

@@ -27,6 +27,7 @@ pub async fn get_cognito_credentials_send(
     telemetry_stage: &TelemetryStage,
 ) -> Result<Credentials, CredentialsError> {
     trace!("Creating new cognito credentials");
+
     let conf = aws_sdk_cognitoidentity::Config::builder()
         .behavior_version(BehaviorVersion::v2025_01_17())
         .region(telemetry_stage.region.clone())
@@ -161,3 +162,17 @@ fn is_expired(expiration: Option<&String>) -> bool {
         },
     }
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[tokio::test]
+    async fn pools() {
+        for telemetry_stage in [TelemetryStage::BETA, TelemetryStage::EXTERNAL_PROD] {
+            get_cognito_credentials_send(&mut Database::new().await.unwrap(), &telemetry_stage)
+                .await
+                .unwrap();
+        }
+    }
+}