lixi
4762358f54
CI / ai-gates (push) Failing after 29s
`api_key=volcengine.api_key` 这类**依赖注入传参**被 KEY-ASSIGN 判为疑似凭证。 这是规则的真缺口而非代码问题：行里没有任何秘密，而做 DI 的代码到处都是这种 写法——绕过一次就得在每个调用点绕过，那等于把规则废掉。故改规则。 新增 ALLOW_REFERENCE，刻意收得很紧以免开后门： - 值必须是**未加引号**的小写起首标识符引用（真密钥要么带引号，要么不是标识符形状） - 只允许**恰好一个点且随即行尾** - 连字符不在字符集内 对抗性夹具验证（三真三假）： - 拦下 `api_key = "sk-T3Blbk…"`（OpenAI 形态） - 拦下 `secret_key: "AKID…"`（腾讯云形态） - 拦下 `auth_token=eyJ….eyJ….Sfl…`（**三段式 JWT——两个点，故不落进放行**） - 放行 `api_key=lowercase.reference` / `api_key = settings.api_key` / `api_token: config.authToken` 原 5 形态泄漏夹具复验仍 8 条命中，一条不少。 另更正口径：脚本头与 git-workflow.md 仍写「三仓同构」，实际已是四仓—— patbond-ai 是唯一持有 AI 服务商凭证的仓，漏同步它风险最大。四仓 md5 已一致 （4945940093b4b8413b7d3eec8bf6a5ec），四仓 --all 全绿。 Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
"""装配根与注册表的测试。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from app.assembly import build_registry
|
|
from app.config import Settings, VolcengineSettings
|
|
from app.domain.generation import (
|
|
GenerationRequest,
|
|
GenerationResult,
|
|
MediaKind,
|
|
TaskState,
|
|
)
|
|
from app.providers.base import GenerationProvider
|
|
from app.providers.registry import ProviderRegistry
|
|
from app.providers.volcengine.video import PROVIDER_CODE
|
|
|
|
FAKE_API_KEY = "dummy-api-key-for-tests"
|
|
|
|
|
|
class StubProvider(GenerationProvider):
|
|
def __init__(self, code: str, kind: MediaKind = MediaKind.VIDEO):
|
|
self._code = code
|
|
self._kind = kind
|
|
|
|
@property
|
|
def provider_code(self) -> str:
|
|
return self._code
|
|
|
|
@property
|
|
def media_kind(self) -> MediaKind:
|
|
return self._kind
|
|
|
|
def submit(self, request: GenerationRequest) -> str:
|
|
return "stub-1"
|
|
|
|
def fetch(self, provider_request_id: str) -> GenerationResult:
|
|
return GenerationResult(provider_request_id, TaskState.RUNNING)
|
|
|
|
|
|
class TestRegistry:
|
|
def test_finds_by_code(self):
|
|
registry = ProviderRegistry([StubProvider("a"), StubProvider("b")])
|
|
assert registry.find("a").provider_code == "a"
|
|
assert registry.codes == frozenset({"a", "b"})
|
|
|
|
def test_unknown_code_returns_none(self):
|
|
"""对已持久化的任务，未知 code 意味着当初提交它的适配器已被移除。
|
|
|
|
调用方应当作不可重试的失败，而不是循环重试到预算耗尽。
|
|
"""
|
|
assert ProviderRegistry([StubProvider("a")]).find("gone") is None
|
|
|
|
def test_duplicate_registration_is_rejected(self):
|
|
# 静默覆盖会让「哪个实例在处理任务」取决于注册顺序，是最难查的一类问题。
|
|
with pytest.raises(ValueError, match="重复注册"):
|
|
ProviderRegistry([StubProvider("same"), StubProvider("same")])
|
|
|
|
|
|
class TestAssembly:
|
|
def test_seedance_is_registered_when_credentials_exist(self):
|
|
registry = build_registry(
|
|
Settings(volcengine=VolcengineSettings(api_key=FAKE_API_KEY))
|
|
)
|
|
provider = registry.find(PROVIDER_CODE)
|
|
assert provider is not None
|
|
assert provider.media_kind is MediaKind.VIDEO
|
|
|
|
def test_missing_credentials_skip_the_provider_instead_of_failing_startup(self):
|
|
"""本地只配了一家凭证时，服务仍应起得来。
|
|
|
|
真正的失败点在调用时：find() 返回 None，调用方按「适配器不可用」处理，
|
|
而不是拿到一个半死的实例。
|
|
"""
|
|
registry = build_registry(Settings(volcengine=VolcengineSettings(api_key="")))
|
|
assert registry.find(PROVIDER_CODE) is None
|
|
assert len(registry) == 0
|
|
|
|
def test_registered_code_matches_the_catalog_column(self):
|
|
"""注册表的键必须与 creation.generation_models.provider_code 同值。
|
|
|
|
对不上时，Java 侧按任务行的快照来查会一直查不到适配器——而症状是任务
|
|
永远卡在排队，离根因很远。
|
|
"""
|
|
registry = build_registry(
|
|
Settings(volcengine=VolcengineSettings(api_key=FAKE_API_KEY))
|
|
)
|
|
assert PROVIDER_CODE in registry.codes
|