Files

33 lines
1.1 KiB
Python

"""pytest 全局配置。
在测试进程启动时创建独立临时目录,并通过环境变量把应用的数据目录、数据库、
存储和后台服务全部指向测试环境,避免污染本地开发数据。
"""
import atexit
import os
import shutil
import tempfile
from pathlib import Path
# 每个测试进程使用独立临时根目录,保证测试之间互不干扰。
TEST_ROOT = Path(tempfile.mkdtemp(prefix="wov-api-test-"))
os.environ["WOV_DATA_DIR"] = str(TEST_ROOT / "data")
os.environ["WOV_DB_PATH"] = str(TEST_ROOT / "data" / "wov.db")
os.environ["WOV_STORAGE_DIR"] = str(TEST_ROOT / "storage")
# 回收线程轮询调快,便于空闲回收测试尽快完成。
os.environ["WOV_REAP_INTERVAL_SECONDS"] = "0.1"
# 就绪超时缩短,避免节点启动失败用例等待过久。
os.environ["WOV_READY_TIMEOUT_SECONDS"] = "2"
# 默认关闭自动种子和后台调度,测试显式控制执行时机。
os.environ["WOV_AUTO_SEED"] = "0"
os.environ["WOV_SCHEDULER_ENABLED"] = "0"
def _cleanup() -> None:
"""进程退出时清理临时测试目录。"""
shutil.rmtree(TEST_ROOT, ignore_errors=True)
atexit.register(_cleanup)