22 lines
687 B
Python
22 lines
687 B
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_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 == []
|