修复了一些错误更新了AI.md(给ai看的)

2026-05-02 19:21:50 +08:00
parent 0783428f80
commit 70c531860b
240 changed files with 5626 additions and 10790 deletions
--- a/oss/tests/test_config.py
+++ b/oss/tests/test_config.py
@@ -0,0 +1,141 @@
+Tests for Configuration Management
+
+import os
+import json
+import tempfile
+import pytest
+from pathlib import Path
+
+from oss.config import Config, get_config, init_config
+
+
+class TestConfig:
+        temp_dir = tempfile.mkdtemp()
+        config_file = os.path.join(temp_dir, "config.json")
+        
+        config_data = {
+            "HTTP_API_PORT": 9000,
+            "HTTP_TCP_PORT": 9002,
+            "HOST": "127.0.0.1",
+            "DATA_DIR": "./test_data",
+            "STORE_DIR": "./test_store",
+            "LOG_LEVEL": "DEBUG",
+            "PERMISSION_CHECK": False,
+            "MAX_WORKERS": 8,
+            "API_KEY": "test-key",
+            "CORS_ALLOWED_ORIGINS": ["http://localhost:8080"]
+        }
+        
+        with open(config_file, 'w') as f:
+            json.dump(config_data, f)
+        
+        yield config_file
+        
+        os.remove(config_file)
+        os.rmdir(temp_dir)
+    
+    def test_config_initialization_defaults(self):
+        config = Config(temp_config_file)
+        
+        assert config.get("HTTP_API_PORT") == 9000
+        assert config.get("HTTP_TCP_PORT") == 9002
+        assert config.get("HOST") == "127.0.0.1"
+        assert config.get("DATA_DIR") == "./test_data"
+        assert config.get("STORE_DIR") == "./test_store"
+        assert config.get("LOG_LEVEL") == "DEBUG"
+        assert config.get("PERMISSION_CHECK") is False
+        assert config.get("MAX_WORKERS") == 8
+        assert config.get("API_KEY") == "test-key"
+        assert config.get("CORS_ALLOWED_ORIGINS") == ["http://localhost:8080"]
+    
+    def test_config_load_from_nonexistent_file(self):
+        temp_dir = tempfile.mkdtemp()
+        config_file = os.path.join(temp_dir, "invalid_config.json")
+        
+        with open(config_file, 'w') as f:
+            f.write("{ invalid json")
+        
+        config = Config(config_file)
+        
+        assert config.get("HTTP_API_PORT") == 8080
+        
+        os.remove(config_file)
+        os.rmdir(temp_dir)
+    
+    def test_config_load_from_env(self):
+        os.environ["HTTP_API_PORT"] = "7000"
+        os.environ["HOST"] = "192.168.1.1"
+        
+        try:
+            config = Config(temp_config_file)
+            
+            assert config.get("HTTP_TCP_PORT") == 9002
+            assert config.get("DATA_DIR") == "./test_data"
+            
+            assert config.get("HTTP_API_PORT") == 7000
+            assert config.get("HOST") == "192.168.1.1"
+        finally:
+            for key in ["HTTP_API_PORT", "HOST"]:
+                if key in os.environ:
+                    del os.environ[key]
+    
+    def test_config_env_type_conversion(self):
+        os.environ["HTTP_API_PORT"] = "not_a_number"
+        os.environ["PERMISSION_CHECK"] = "not_a_boolean"
+        
+        try:
+            config = Config()
+            
+            assert config.get("HTTP_API_PORT") == 8080
+            assert config.get("PERMISSION_CHECK") is True
+        finally:
+            for key in ["HTTP_API_PORT", "PERMISSION_CHECK"]:
+                if key in os.environ:
+                    del os.environ[key]
+    
+    def test_config_get_with_default(self):
+        config = Config()
+        
+        config.set("HTTP_API_PORT", 9000)
+        assert config.get("HTTP_API_PORT") == 9000
+        
+        config.set("NONEXISTENT_KEY", "value")
+        assert config.get("NONEXISTENT_KEY") is None
+    
+    def test_config_all(self):
+        config = Config()
+        
+        assert isinstance(config.http_api_port, int)
+        assert isinstance(config.http_tcp_port, int)
+        assert isinstance(config.host, str)
+        assert isinstance(config.data_dir, Path)
+        assert isinstance(config.store_dir, Path)
+        assert isinstance(config.log_level, str)
+        assert isinstance(config.permission_check, bool)
+        
+        assert config.http_api_port == 8080
+        assert config.http_tcp_port == 8082
+        assert config.host == "0.0.0.0"
+        assert config.data_dir == Path("./data")
+        assert config.store_dir == Path("./store")
+        assert config.log_level == "INFO"
+        assert config.permission_check is True
+
+
+class TestGlobalConfig:
+        config1 = get_config()
+        config2 = get_config()
+        
+        assert config1 is config2
+    
+    def test_init_config(self):
+        config = init_config(temp_config_file)
+        
+        assert isinstance(config, Config)
+        assert config.get("HTTP_API_PORT") == 9000
+        
+        assert config is get_config()
+
+
+if __name__ == '__main__':
+    pytest.main([__file__, '-v'])