kreuzberg 4.3.8

//! Tests for environment variable overrides.

#![allow(unsafe_code)]

use crate::core::ServerConfig;

#[serial_test::serial]
#[test]
fn test_apply_env_host_override() {
    let original = std::env::var("KREUZBERG_HOST").ok();
    unsafe {
        std::env::set_var("KREUZBERG_HOST", "192.168.1.1");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.host, "192.168.1.1");

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_HOST", orig);
        } else {
            std::env::remove_var("KREUZBERG_HOST");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_port_override() {
    let original = std::env::var("KREUZBERG_PORT").ok();
    unsafe {
        std::env::set_var("KREUZBERG_PORT", "5000");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.port, 5000);

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_PORT", orig);
        } else {
            std::env::remove_var("KREUZBERG_PORT");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_port_invalid() {
    let original = std::env::var("KREUZBERG_PORT").ok();
    unsafe {
        std::env::set_var("KREUZBERG_PORT", "not_a_number");
    }

    let mut config = ServerConfig::default();
    let result = config.apply_env_overrides();

    assert!(result.is_err());
    assert!(
        result
            .unwrap_err()
            .to_string()
            .contains("KREUZBERG_PORT must be a valid u16")
    );

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_PORT", orig);
        } else {
            std::env::remove_var("KREUZBERG_PORT");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_cors_origins_override() {
    let original = std::env::var("KREUZBERG_CORS_ORIGINS").ok();
    unsafe {
        std::env::set_var("KREUZBERG_CORS_ORIGINS", "https://example.com, https://other.com");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.cors_origins.len(), 2);
    assert!(config.cors_origins.contains(&"https://example.com".to_string()));
    assert!(config.cors_origins.contains(&"https://other.com".to_string()));

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_CORS_ORIGINS", orig);
        } else {
            std::env::remove_var("KREUZBERG_CORS_ORIGINS");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_max_request_body_bytes_override() {
    let original = std::env::var("KREUZBERG_MAX_REQUEST_BODY_BYTES").ok();
    unsafe {
        std::env::set_var("KREUZBERG_MAX_REQUEST_BODY_BYTES", "52428800");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.max_request_body_bytes, 52_428_800);

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_MAX_REQUEST_BODY_BYTES", orig);
        } else {
            std::env::remove_var("KREUZBERG_MAX_REQUEST_BODY_BYTES");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_max_multipart_field_bytes_override() {
    let original = std::env::var("KREUZBERG_MAX_MULTIPART_FIELD_BYTES").ok();
    unsafe {
        std::env::set_var("KREUZBERG_MAX_MULTIPART_FIELD_BYTES", "78643200");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.max_multipart_field_bytes, 78_643_200);

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_MAX_MULTIPART_FIELD_BYTES", orig);
        } else {
            std::env::remove_var("KREUZBERG_MAX_MULTIPART_FIELD_BYTES");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_legacy_max_upload_size_mb_override() {
    let original = std::env::var("KREUZBERG_MAX_UPLOAD_SIZE_MB").ok();
    unsafe {
        std::env::set_var("KREUZBERG_MAX_UPLOAD_SIZE_MB", "75");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.max_upload_mb, Some(75));
    assert_eq!(config.max_multipart_field_bytes, 75 * 1_048_576);

    // Cleanup
    unsafe {
        if let Some(orig) = original {
            std::env::set_var("KREUZBERG_MAX_UPLOAD_SIZE_MB", orig);
        } else {
            std::env::remove_var("KREUZBERG_MAX_UPLOAD_SIZE_MB");
        }
    }
}

#[serial_test::serial]
#[test]
fn test_apply_env_multiple_overrides() {
    let host_orig = std::env::var("KREUZBERG_HOST").ok();
    let port_orig = std::env::var("KREUZBERG_PORT").ok();
    let cors_orig = std::env::var("KREUZBERG_CORS_ORIGINS").ok();

    unsafe {
        std::env::set_var("KREUZBERG_HOST", "0.0.0.0");
        std::env::set_var("KREUZBERG_PORT", "4000");
        std::env::set_var("KREUZBERG_CORS_ORIGINS", "https://api.example.com");
    }

    let mut config = ServerConfig::default();
    config.apply_env_overrides().unwrap();

    assert_eq!(config.host, "0.0.0.0");
    assert_eq!(config.port, 4000);
    assert_eq!(config.cors_origins.len(), 1);
    assert_eq!(config.cors_origins[0], "https://api.example.com");

    // Cleanup
    unsafe {
        if let Some(orig) = host_orig {
            std::env::set_var("KREUZBERG_HOST", orig);
        } else {
            std::env::remove_var("KREUZBERG_HOST");
        }
        if let Some(orig) = port_orig {
            std::env::set_var("KREUZBERG_PORT", orig);
        } else {
            std::env::remove_var("KREUZBERG_PORT");
        }
        if let Some(orig) = cors_orig {
            std::env::set_var("KREUZBERG_CORS_ORIGINS", orig);
        } else {
            std::env::remove_var("KREUZBERG_CORS_ORIGINS");
        }
    }
}