37d282c0a2
- 添加 pytest 配置和测试依赖到 requirements.txt - 创建测试包结构和 fixtures (conftest.py) - 添加数据库模块的 CRUD 操作测试 (test_database.py) - 添加 Provider 插件系统测试 (test_providers.py) - 添加调度器模块测试 (test_scheduler.py) - 添加 API 路由测试 (test_api.py) - 添加回归测试覆盖边界条件和错误处理 (test_regressions.py) - 添加健康检查端点用于容器监控 - 修复调度器中的日历计算逻辑和任务执行参数处理 - 更新数据库函数以返回操作结果状态
78 lines
2.0 KiB
Python
78 lines
2.0 KiB
Python
"""测试配置和 fixtures"""
|
|
|
|
import asyncio
|
|
import tempfile
|
|
import uuid
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from app.config import AppConfig, load_config
|
|
from app.database import close_db, get_db
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def event_loop():
|
|
"""创建事件循环"""
|
|
loop = asyncio.get_event_loop_policy().new_event_loop()
|
|
yield loop
|
|
loop.close()
|
|
|
|
|
|
@pytest.fixture
|
|
async def temp_db():
|
|
"""临时数据库 fixture"""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
# 使用唯一的文件名确保每个测试都使用新数据库
|
|
db_path = Path(tmpdir) / f"test_{uuid.uuid4().hex}.db"
|
|
# 修改配置使用临时数据库
|
|
import app.config as config_module
|
|
original_db_path = config_module.settings.database.path
|
|
config_module.settings.database.path = str(db_path)
|
|
|
|
# 确保使用新的数据库连接
|
|
import app.database as db_module
|
|
db_module._db = None
|
|
|
|
yield db_path
|
|
|
|
# 清理
|
|
await close_db()
|
|
config_module.settings.database.path = original_db_path
|
|
db_module._db = None
|
|
|
|
|
|
@pytest.fixture
|
|
async def temp_storage():
|
|
"""临时存储目录 fixture"""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
import app.config as config_module
|
|
original_storage_path = config_module.settings.storage.path
|
|
config_module.settings.storage.path = tmpdir
|
|
|
|
yield tmpdir
|
|
|
|
config_module.settings.storage.path = original_storage_path
|
|
|
|
|
|
@pytest.fixture
|
|
async def db(temp_db):
|
|
"""初始化数据库 fixture"""
|
|
from app import database as db_module
|
|
await get_db()
|
|
return db_module
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_plan():
|
|
"""示例 Plan 数据"""
|
|
return {
|
|
"name": "Test Plan",
|
|
"provider_name": "openai",
|
|
"api_key": "sk-test-key",
|
|
"api_base": "https://api.openai.com/v1",
|
|
"plan_type": "coding",
|
|
"supported_models": ["gpt-4", "gpt-3.5-turbo"],
|
|
"enabled": True,
|
|
}
|