Serendipity/md-vector-db
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: test_config 用 monkeypatch 替代 importlib.reload

Browse Source
This commit is contained in:
刘航宇
2026-07-10 15:28:22 +08:00
parent 788f610d3f
commit 76ae81c772
1 changed files with 4 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_config.py
+4 -11
View File
@@ -87,20 +87,13 @@ class TestEmbedConfigEnvVar:
"""api_key 从环境变量读取.""" """api_key 从环境变量读取."""
def test_api_key_from_env(self, monkeypatch): def test_api_key_from_env(self, monkeypatch):
"""从环境变量读取 API Key."""
monkeypatch.setenv("EMBED_API_KEY", "sk-env-test") monkeypatch.setenv("EMBED_API_KEY", "sk-env-test")
import importlib cfg = EmbedConfig(mode="api")
from src.core import config
importlib.reload(config)
cfg = config.EmbedConfig(mode="api")
assert cfg.api_key == "sk-env-test" assert cfg.api_key == "sk-env-test"
# 清理
importlib.reload(config)
def test_api_key_empty_when_not_set(self, monkeypatch): def test_api_key_empty_when_not_set(self, monkeypatch):
"""未设置时返回空字符串."""
monkeypatch.delenv("EMBED_API_KEY", raising=False) monkeypatch.delenv("EMBED_API_KEY", raising=False)
import importlib cfg = EmbedConfig(mode="api")
from src.core import config
importlib.reload(config)
cfg = config.EmbedConfig(mode="api")
assert cfg.api_key == "" assert cfg.api_key == ""
importlib.reload(config)