fix(cortex-cli): update tests for Rust 2024 edition safety requirements (#334)

- Wrap std::env::set_var and std::env::remove_var calls in unsafe blocks
  (required by Rust 2024 edition)
- Add proper imports to doc tests for print_success, print_error,
  print_warning, print_info, print_dim, and styled_label functions

This fixes test compilation failures due to Rust 2024's new safety
requirements for environment variable manipulation functions.

Co-authored-by: Droid Agent <droid@factory.ai>

Author: echobt

cortex-cli/src/styled_output.rs

@@ -201,6 +201,7 @@ fn print_styled_internal(msg_type: MessageType, message: &str, to_stderr: bool,
 ///
 /// # Example
 /// ```
+/// use cortex_cli::styled_output::print_success;
 /// print_success("Operation completed successfully");
 /// // Output: ✓ Operation completed successfully (in green)
 /// ```
@@ -215,6 +216,7 @@ pub fn print_success(message: &str) {
 ///
 /// # Example
 /// ```
+/// use cortex_cli::styled_output::print_error;
 /// print_error("Failed to connect to server");
 /// // Output: ✗ Failed to connect to server (in red)
 /// ```
@@ -229,6 +231,7 @@ pub fn print_error(message: &str) {
 ///
 /// # Example
 /// ```
+/// use cortex_cli::styled_output::print_warning;
 /// print_warning("Configuration file not found, using defaults");
 /// // Output: [WARN] Configuration file not found, using defaults (in amber)
 /// ```
@@ -243,6 +246,7 @@ pub fn print_warning(message: &str) {
 ///
 /// # Example
 /// ```
+/// use cortex_cli::styled_output::print_info;
 /// print_info("Processing 42 files...");
 /// // Output: [INFO] Processing 42 files... (in blue)
 /// ```
@@ -257,6 +261,7 @@ pub fn print_info(message: &str) {
 ///
 /// # Example
 /// ```
+/// use cortex_cli::styled_output::print_dim;
 /// print_dim("Note: Using default configuration");
 /// // Output: · Note: Using default configuration (in gray)
 /// ```
@@ -344,6 +349,7 @@ fn format_styled(msg_type: MessageType, message: &str) -> String {
 ///
 /// # Example
 /// ```
+/// use cortex_cli::styled_output::{styled_label, MessageType};
 /// println!("Status: {}", styled_label(MessageType::Success, "PASSED"));
 /// // Output: Status: ✓ PASSED (with PASSED in green)
 /// ```
@@ -389,33 +395,35 @@ mod tests {
     #[test]
     fn test_format_styled_no_color() {
         // When colors are disabled, should just return icon + message
-        std::env::set_var("NO_COLOR", "1");
+        // SAFETY: These tests run serially and we restore env vars immediately
+        unsafe { std::env::set_var("NO_COLOR", "1") };
         let result = format_styled(MessageType::Success, "test message");
         assert!(result.contains("[OK]"));
         assert!(result.contains("test message"));
-        std::env::remove_var("NO_COLOR");
+        unsafe { std::env::remove_var("NO_COLOR") };
     }
 
     #[test]
     fn test_colors_disabled() {
-        std::env::set_var("NO_COLOR", "1");
+        // SAFETY: These tests run serially and we restore env vars immediately
+        unsafe { std::env::set_var("NO_COLOR", "1") };
         assert!(colors_disabled());
-        std::env::remove_var("NO_COLOR");
+        unsafe { std::env::remove_var("NO_COLOR") };
 
-        std::env::set_var("NO_COLOR", "true");
+        unsafe { std::env::set_var("NO_COLOR", "true") };
         assert!(colors_disabled());
-        std::env::remove_var("NO_COLOR");
+        unsafe { std::env::remove_var("NO_COLOR") };
 
-        std::env::set_var("NO_COLOR", "0");
+        unsafe { std::env::set_var("NO_COLOR", "0") };
         assert!(!colors_disabled());
-        std::env::remove_var("NO_COLOR");
+        unsafe { std::env::remove_var("NO_COLOR") };
 
-        std::env::set_var("NO_COLOR", "false");
+        unsafe { std::env::set_var("NO_COLOR", "false") };
         assert!(!colors_disabled());
-        std::env::remove_var("NO_COLOR");
+        unsafe { std::env::remove_var("NO_COLOR") };
 
-        std::env::set_var("NO_COLOR", "");
+        unsafe { std::env::set_var("NO_COLOR", "") };
         assert!(!colors_disabled());
-        std::env::remove_var("NO_COLOR");
+        unsafe { std::env::remove_var("NO_COLOR") };
     }
 }