Files
Profeto/tests/test_bzzoiro_source_registry.py
T
shangfangjian b9d7b6423e test(c2): 补齐导入次序回归,注明环状导入为预期且可自愈
test_bzzoiro_source_registry.py 增至 11 项:
- 新增 TestImportOrderSelfHeal:用独立子进程验证两种真实导入次序
  (sources 先 / bzzoiro 先)下注册表均可用。该缺陷只在真实模块初始化
  时序下成立,mock 照不出来,故不 mock。
- 新增未知名字仍抛 ValueError 的契约守卫。

src/data/sources.py:补全 _load_sources 文档,说明首次装载失败属正常的
环状导入(bzzoiro 仍在初始化),失败时保持 _loaded=False 使后续
get_source()/list_sources() 调用可重试装载(自愈)。

验证:
- tests/test_bzzoiro_source_registry.py:11 passed
- 两种导入次序实测均输出 bzzoiro ['bzzoiro']
2026-09-21 17:13:57 +08:00

145 lines
6.7 KiB
Python

"""回归测试:锁定 bzzoiro events 管线被删除 + 注册表静默失效的缺陷不再复发。
背景(真实数据丢失事故):
重构提交 6a49940 删除了 src/data/bzzoiro.py 中的 fetch_bzzoiro_events 与
BzzoiroSource,但 src/data/sources.py 的模块级预热把 ImportError 用
`try/except Exception: pass` 吞掉了。后果:
- _SOURCES 注册表恒为空
- get_source("bzzoiro") 恒抛 ValueError("未知数据源: bzzoiro")
- src/api/routes/ingest.py 与 schedules.py 在运行时全线失效,
且日志中看不到任何导入错误的痕迹 —— 这才是它长期漏网的原因。
本测试用静态结构断言 + 真实导入来锁死这两点,不 mock 网络:
- 注册表必须非空(直接守卫「静默吞异常」回归)
- get_source("bzzoiro") 必须返回真实实例
- ingest() 的签名必须保持 caller 依赖的 keyword-only 参数
"""
from __future__ import annotations
import inspect
from pathlib import Path
import pytest
from src.data.bzzoiro import BzzoiroSource, fetch_bzzoiro_events
from src.data.sources import get_source, list_sources
SRC = Path(__file__).resolve().parent.parent / "src"
class TestSourceRegistry:
"""P0: 注册表必须真的装载到 bzzoiro,而不是静默为空。"""
def test_registry_is_not_empty(self):
"""注册表恒非空 —— 直接守卫 `try/except: pass` 静默吞 ImportError。"""
assert list_sources(), (
"数据源注册表为空 —— _load_sources() 的导入失败了,"
"且(修复前)异常被静默吞掉。任何导入错误都必须被 logger.exception 记录。"
)
def test_get_source_returns_bzzoiro(self):
"""get_source('bzzoiro') 必须返回实例,而不是抛 ValueError。"""
source = get_source("bzzoiro")
assert source.name == "bzzoiro"
def test_list_sources_contains_bzzoiro(self):
assert "bzzoiro" in list_sources()
def test_register_decorator_still_exports_the_class(self):
"""@register 必须仍然返回原类(不能再被改成返回实例)。"""
assert isinstance(BzzoiroSource, type), "@register 不应把类替换成实例"
class TestIngestContract:
"""caller 依赖 ingest() 的签名,routes 传的就是这些参数。"""
def test_ingest_exists_and_is_async(self):
assert hasattr(BzzoiroSource, "ingest"), "BzzoiroSource.ingest 丢失(events 管线被删)"
assert inspect.iscoroutinefunction(BzzoiroSource.ingest), "ingest 必须是 async"
def test_ingest_accepts_caller_keyword_args(self):
"""ingest.py:65 与 schedules.py:41 依赖的 keyword-only 参数必须齐全。"""
params = inspect.signature(BzzoiroSource.ingest).parameters
for name in ("leagues", "date_from", "date_to", "status"):
assert name in params, f"ingest() 缺少 keyword 参数 {name!r} —— caller 会 TypeError"
# leagues 必须 keyword-only(src/api/routes/ingest.py 用 leagues=[code] 传)
assert params["leagues"].kind is inspect.Parameter.KEYWORD_ONLY
# 默认值契约:schedules.py 只传 leagues + status
assert params["date_from"].default is None
assert params["date_to"].default is None
assert params["status"].default == "finished"
def test_fetch_bzzoiro_events_signature(self):
"""抓取函数也必须存在,且 leagues 抓取走 keyword-only 的 status/日期。"""
assert callable(fetch_bzzoiro_events)
params = inspect.signature(fetch_bzzoiro_events).parameters
assert "league_code" in params
for name in ("status", "date_from", "date_to"):
assert name in params, f"fetch_bzzoiro_events() 缺少 {name!r}"
class TestImportOrderSelfHeal:
"""导入次序不得影响注册表 —— 这是本缺陷的第二个隐藏面。
背景:src.api.app 先 `import src.data.sources`,其模块级预热在
`src.data.bzzoiro` 尚未初始化时发起,导入链在 sources.py 内成环,
首次导入必然失败(bzzoiro 仍在加载中)。修复前 `except Exception: pass`
把这次失败同时变得「无声」且「不可恢复」,注册表就永久空掉了。
这里用独立子进程验证真实导入次序,不 mock —— 因为该缺陷只在
真实的模块初始化时序下才成立,单元测试里的 mock 反而照不出来。
"""
@staticmethod
def _run(import_lines: list[str]) -> str:
"""在全新解释器中按指定次序导入,返回 get_source/list_sources 结果。"""
import subprocess
import sys
root = Path(__file__).resolve().parent.parent
code = (
"import sys; sys.path.insert(0, r'%s')\n" % root
+ "\n".join(import_lines)
+ "\nfrom src.data.sources import get_source, list_sources\n"
"print('RESULT', get_source('bzzoiro').name, list_sources())\n"
)
proc = subprocess.run(
[sys.executable, "-c", code],
capture_output=True, text=True, timeout=120, cwd=str(root),
)
assert proc.returncode == 0, (
f"导入次序 {import_lines} 下 get_source('bzzoiro') 失败:\n{proc.stderr[-1500:]}"
)
return proc.stdout.strip()
def test_sources_first_then_bzzoiro(self):
"""次序 B:sources 先导入(正常应用路径)。"""
out = self._run(["import src.data.sources"])
assert out == "RESULT bzzoiro ['bzzoiro']", out
def test_bzzoiro_first_then_sources(self):
"""次序 A:bzzoiro 先导入 —— 注册表必须仍然可用(自愈)。"""
out = self._run(["import src.data.bzzoiro"])
assert out == "RESULT bzzoiro ['bzzoiro']", (
out + " —— 注册表为空说明首次装载失败后没有再重试(静默失效回归)"
)
def test_unknown_name_still_raises(self):
"""真正未知的名字仍须抛 ValueError —— 修复不应放宽这一契约。"""
with pytest.raises(ValueError, match="未知数据源"):
get_source("no_such_source_xyz")
class TestNoSilentImportSwallow:
"""sources.py 不许再用 `except Exception: pass` 吞掉导入失败。"""
def test_load_sources_logs_failure(self):
src = (SRC / "data" / "sources.py").read_text(encoding="utf-8")
assert "logger" in src, "sources.py 缺少模块 logger,无法记录导入失败"
body = src[src.index("def _load_sources"):]
assert "logger.exception" in body, (
"_load_sources() 失败时未记日志 —— 静默吞异常会让注册表恒为空,"
"把 ImportError 伪装成「未知数据源」(P0 事故根因)"
)
assert "pass" not in body.split("def ")[0], "不得再用裸 pass 吞掉导入异常"