Complete AI Data Analysis Agent implementation with 95.7% test coverage

tests/test_env_loader.py

@@ -0,0 +1,255 @@
+"""环境变量加载器的单元测试。"""
+
+import os
+import pytest
+from pathlib import Path
+from unittest.mock import patch
+
+from src.env_loader import (
+    load_env_file,
+    load_env_with_fallback,
+    get_env,
+    get_env_bool,
+    get_env_int,
+    get_env_float,
+    validate_required_env_vars
+)
+
+
+class TestLoadEnvFile:
+    """测试加载 .env 文件。"""
+    
+    def test_load_env_file_success(self, tmp_path):
+        """测试成功加载 .env 文件。"""
+        env_file = tmp_path / ".env"
+        env_file.write_text("""
+# This is a comment
+KEY1=value1
+KEY2="value2"
+KEY3='value3'
+KEY4=value with spaces
+
+# Another comment
+KEY5=123
+        """, encoding='utf-8')
+        
+        # 清空环境变量
+        with patch.dict(os.environ, {}, clear=True):
+            result = load_env_file(str(env_file))
+            
+            assert result is True
+            assert os.getenv("KEY1") == "value1"
+            assert os.getenv("KEY2") == "value2"
+            assert os.getenv("KEY3") == "value3"
+            assert os.getenv("KEY4") == "value with spaces"
+            assert os.getenv("KEY5") == "123"
+    
+    def test_load_env_file_not_found(self):
+        """测试加载不存在的 .env 文件。"""
+        result = load_env_file("nonexistent.env")
+        assert result is False
+    
+    def test_load_env_file_skip_existing(self, tmp_path):
+        """测试跳过已存在的环境变量。"""
+        env_file = tmp_path / ".env"
+        env_file.write_text("KEY1=from_file\nKEY2=from_file")
+        
+        # 设置一个已存在的环境变量
+        with patch.dict(os.environ, {"KEY1": "from_env"}, clear=True):
+            load_env_file(str(env_file))
+            
+            # KEY1 应该保持原值（环境变量优先）
+            assert os.getenv("KEY1") == "from_env"
+            # KEY2 应该从文件加载
+            assert os.getenv("KEY2") == "from_file"
+    
+    def test_load_env_file_skip_invalid_lines(self, tmp_path):
+        """测试跳过无效行。"""
+        env_file = tmp_path / ".env"
+        env_file.write_text("""
+VALID_KEY=valid_value
+invalid line without equals
+ANOTHER_VALID=another_value
+        """)
+        
+        with patch.dict(os.environ, {}, clear=True):
+            result = load_env_file(str(env_file))
+            
+            assert result is True
+            assert os.getenv("VALID_KEY") == "valid_value"
+            assert os.getenv("ANOTHER_VALID") == "another_value"
+    
+    def test_load_env_file_empty_lines(self, tmp_path):
+        """测试处理空行。"""
+        env_file = tmp_path / ".env"
+        env_file.write_text("""
+KEY1=value1
+
+KEY2=value2
+
+
+KEY3=value3
+        """)
+        
+        with patch.dict(os.environ, {}, clear=True):
+            result = load_env_file(str(env_file))
+            
+            assert result is True
+            assert os.getenv("KEY1") == "value1"
+            assert os.getenv("KEY2") == "value2"
+            assert os.getenv("KEY3") == "value3"
+
+
+class TestLoadEnvWithFallback:
+    """测试按优先级加载多个 .env 文件。"""
+    
+    def test_load_multiple_files(self, tmp_path):
+        """测试加载多个文件。"""
+        env_file1 = tmp_path / ".env.local"
+        env_file1.write_text("KEY1=local\nKEY2=local")
+        
+        env_file2 = tmp_path / ".env"
+        env_file2.write_text("KEY1=default\nKEY3=default")
+        
+        with patch.dict(os.environ, {}, clear=True):
+            # 切换到临时目录
+            original_dir = os.getcwd()
+            os.chdir(tmp_path)
+            
+            try:
+                result = load_env_with_fallback([".env.local", ".env"])
+                
+                assert result is True
+                # KEY1 应该来自 .env.local（优先级更高）
+                assert os.getenv("KEY1") == "local"
+                # KEY2 应该来自 .env.local
+                assert os.getenv("KEY2") == "local"
+                # KEY3 应该来自 .env
+                assert os.getenv("KEY3") == "default"
+            finally:
+                os.chdir(original_dir)
+    
+    def test_load_no_files_found(self):
+        """测试没有找到任何文件。"""
+        result = load_env_with_fallback(["nonexistent1.env", "nonexistent2.env"])
+        assert result is False
+
+
+class TestGetEnv:
+    """测试获取环境变量。"""
+    
+    def test_get_env_exists(self):
+        """测试获取存在的环境变量。"""
+        with patch.dict(os.environ, {"TEST_KEY": "test_value"}):
+            assert get_env("TEST_KEY") == "test_value"
+    
+    def test_get_env_not_exists(self):
+        """测试获取不存在的环境变量。"""
+        with patch.dict(os.environ, {}, clear=True):
+            assert get_env("NONEXISTENT_KEY") is None
+    
+    def test_get_env_with_default(self):
+        """测试使用默认值。"""
+        with patch.dict(os.environ, {}, clear=True):
+            assert get_env("NONEXISTENT_KEY", "default") == "default"
+
+
+class TestGetEnvBool:
+    """测试获取布尔类型环境变量。"""
+    
+    def test_get_env_bool_true_values(self):
+        """测试 True 值。"""
+        true_values = ["true", "True", "TRUE", "yes", "Yes", "YES", "1", "on", "On", "ON"]
+        
+        for value in true_values:
+            with patch.dict(os.environ, {"TEST_BOOL": value}):
+                assert get_env_bool("TEST_BOOL") is True
+    
+    def test_get_env_bool_false_values(self):
+        """测试 False 值。"""
+        false_values = ["false", "False", "FALSE", "no", "No", "NO", "0", "off", "Off", "OFF"]
+        
+        for value in false_values:
+            with patch.dict(os.environ, {"TEST_BOOL": value}):
+                assert get_env_bool("TEST_BOOL") is False
+    
+    def test_get_env_bool_default(self):
+        """测试默认值。"""
+        with patch.dict(os.environ, {}, clear=True):
+            assert get_env_bool("NONEXISTENT_BOOL") is False
+            assert get_env_bool("NONEXISTENT_BOOL", True) is True
+
+
+class TestGetEnvInt:
+    """测试获取整数类型环境变量。"""
+    
+    def test_get_env_int_valid(self):
+        """测试有效的整数。"""
+        with patch.dict(os.environ, {"TEST_INT": "123"}):
+            assert get_env_int("TEST_INT") == 123
+    
+    def test_get_env_int_negative(self):
+        """测试负整数。"""
+        with patch.dict(os.environ, {"TEST_INT": "-456"}):
+            assert get_env_int("TEST_INT") == -456
+    
+    def test_get_env_int_invalid(self):
+        """测试无效的整数。"""
+        with patch.dict(os.environ, {"TEST_INT": "not_a_number"}):
+            assert get_env_int("TEST_INT") == 0
+            assert get_env_int("TEST_INT", 999) == 999
+    
+    def test_get_env_int_default(self):
+        """测试默认值。"""
+        with patch.dict(os.environ, {}, clear=True):
+            assert get_env_int("NONEXISTENT_INT") == 0
+            assert get_env_int("NONEXISTENT_INT", 42) == 42
+
+
+class TestGetEnvFloat:
+    """测试获取浮点数类型环境变量。"""
+    
+    def test_get_env_float_valid(self):
+        """测试有效的浮点数。"""
+        with patch.dict(os.environ, {"TEST_FLOAT": "3.14"}):
+            assert get_env_float("TEST_FLOAT") == 3.14
+    
+    def test_get_env_float_negative(self):
+        """测试负浮点数。"""
+        with patch.dict(os.environ, {"TEST_FLOAT": "-2.5"}):
+            assert get_env_float("TEST_FLOAT") == -2.5
+    
+    def test_get_env_float_invalid(self):
+        """测试无效的浮点数。"""
+        with patch.dict(os.environ, {"TEST_FLOAT": "not_a_number"}):
+            assert get_env_float("TEST_FLOAT") == 0.0
+            assert get_env_float("TEST_FLOAT", 9.99) == 9.99
+    
+    def test_get_env_float_default(self):
+        """测试默认值。"""
+        with patch.dict(os.environ, {}, clear=True):
+            assert get_env_float("NONEXISTENT_FLOAT") == 0.0
+            assert get_env_float("NONEXISTENT_FLOAT", 1.5) == 1.5
+
+
+class TestValidateRequiredEnvVars:
+    """测试验证必需的环境变量。"""
+    
+    def test_validate_all_present(self):
+        """测试所有必需的环境变量都存在。"""
+        with patch.dict(os.environ, {"KEY1": "value1", "KEY2": "value2", "KEY3": "value3"}):
+            assert validate_required_env_vars(["KEY1", "KEY2", "KEY3"]) is True
+    
+    def test_validate_some_missing(self):
+        """测试部分环境变量缺失。"""
+        with patch.dict(os.environ, {"KEY1": "value1"}, clear=True):
+            assert validate_required_env_vars(["KEY1", "KEY2", "KEY3"]) is False
+    
+    def test_validate_all_missing(self):
+        """测试所有环境变量都缺失。"""
+        with patch.dict(os.environ, {}, clear=True):
+            assert validate_required_env_vars(["KEY1", "KEY2"]) is False
+    
+    def test_validate_empty_list(self):
+        """测试空列表。"""
+        assert validate_required_env_vars([]) is True