708ff29515
- 新增: test_knowledge_base_backend.py, test_skills_middleware.py - 更新: 所有现有测试适配 db_id→kb_id, get_subagents_from_names→slugs 等接口变更
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
import yuxi.agents.backends.knowledge_base_backend as knowledge_base_backend
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_visible_knowledge_bases_filters_by_slug(monkeypatch):
|
|
async def fake_get_databases_by_uid(_uid):
|
|
return {
|
|
"databases": [
|
|
{"slug": "kb-a", "name": "Same Name"},
|
|
{"slug": "kb-b", "name": "Same Name"},
|
|
]
|
|
}
|
|
|
|
monkeypatch.setattr(knowledge_base_backend.knowledge_base, "get_databases_by_uid", fake_get_databases_by_uid)
|
|
|
|
context = SimpleNamespace(uid="u1", knowledges=["kb-b"])
|
|
|
|
databases = await knowledge_base_backend.resolve_visible_knowledge_bases_for_context(context)
|
|
|
|
assert databases == [{"slug": "kb-b", "name": "Same Name"}]
|
|
assert context._visible_knowledge_bases == databases
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_visible_knowledge_bases_requires_slug(monkeypatch):
|
|
async def fake_get_databases_by_uid(_uid):
|
|
return {"databases": [{"id": "legacy-id", "name": "Legacy"}]}
|
|
|
|
monkeypatch.setattr(knowledge_base_backend.knowledge_base, "get_databases_by_uid", fake_get_databases_by_uid)
|
|
|
|
context = SimpleNamespace(uid="u1", knowledges=["legacy-id"])
|
|
|
|
databases = await knowledge_base_backend.resolve_visible_knowledge_bases_for_context(context)
|
|
|
|
assert databases == []
|