Delete backend/test/test_chat_router_configs.py
This commit is contained in:
parent
fb05161f10
commit
f91e931a24
@ -1,70 +0,0 @@
|
|||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from fastapi import FastAPI
|
|
||||||
from fastapi.testclient import TestClient
|
|
||||||
|
|
||||||
from server.routers.chat_router import chat
|
|
||||||
from server.utils.auth_middleware import get_db, get_required_user
|
|
||||||
from yuxi.storage.postgres.models_business import User
|
|
||||||
|
|
||||||
|
|
||||||
def _build_app() -> FastAPI:
|
|
||||||
app = FastAPI()
|
|
||||||
app.include_router(chat, prefix="/api")
|
|
||||||
|
|
||||||
async def fake_db():
|
|
||||||
return None
|
|
||||||
|
|
||||||
async def fake_required_user():
|
|
||||||
return User(
|
|
||||||
username="admin",
|
|
||||||
user_id="admin",
|
|
||||||
password_hash="x",
|
|
||||||
role="admin",
|
|
||||||
department_id=1,
|
|
||||||
id=1,
|
|
||||||
)
|
|
||||||
|
|
||||||
app.dependency_overrides[get_db] = fake_db
|
|
||||||
app.dependency_overrides[get_required_user] = fake_required_user
|
|
||||||
return app
|
|
||||||
|
|
||||||
|
|
||||||
def test_list_agent_configs_creates_default_with_agent_id(monkeypatch):
|
|
||||||
agent_id = "demo-agent"
|
|
||||||
captured: dict[str, object] = {}
|
|
||||||
|
|
||||||
class DummyConfig:
|
|
||||||
id = 1
|
|
||||||
name = "初始配置"
|
|
||||||
description = None
|
|
||||||
icon = None
|
|
||||||
pics = []
|
|
||||||
examples = []
|
|
||||||
is_default = True
|
|
||||||
|
|
||||||
class DummyRepo:
|
|
||||||
def __init__(self, _db):
|
|
||||||
self.calls = 0
|
|
||||||
|
|
||||||
async def list_by_department_agent(self, *, department_id, agent_id):
|
|
||||||
self.calls += 1
|
|
||||||
if self.calls == 1:
|
|
||||||
return []
|
|
||||||
return [DummyConfig()]
|
|
||||||
|
|
||||||
async def get_or_create_default(self, *, department_id, agent_id, created_by=None):
|
|
||||||
captured["department_id"] = department_id
|
|
||||||
captured["agent_id"] = agent_id
|
|
||||||
captured["created_by"] = created_by
|
|
||||||
return None
|
|
||||||
|
|
||||||
monkeypatch.setattr("server.routers.chat_router.agent_manager.get_agent", lambda _agent_id: object())
|
|
||||||
monkeypatch.setattr("server.routers.chat_router.AgentConfigRepository", DummyRepo)
|
|
||||||
|
|
||||||
client = TestClient(_build_app())
|
|
||||||
resp = client.get(f"/api/chat/agent/{agent_id}/configs")
|
|
||||||
assert resp.status_code == 200, resp.text
|
|
||||||
assert captured["agent_id"] == agent_id
|
|
||||||
assert captured["department_id"] == 1
|
|
||||||
assert captured["created_by"] == "1"
|
|
||||||
Loading…
Reference in New Issue
Block a user