mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-18 13:33:53 +08:00
初始化:独立 oclaw 仓库首提交
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore,避免提交本地运行态数据(_local、node_modules、logs 等) - 提交当前工程代码与配置 Made-with: Cursor
This commit is contained in:
commit
ba3836f00f
579 changed files with 83112 additions and 0 deletions
91
tests/test_chat_session_full_smoke.py
Normal file
91
tests/test_chat_session_full_smoke.py
Normal file
|
|
@ -0,0 +1,91 @@
|
|||
"""端到端:管理台登录 → 创建聊天会话 → 发送一条消息(不依赖外网 LLM)。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import hashlib
|
||||
import os
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from oclaw.interfaces.http.fastapi_app import create_app
|
||||
from oclaw.platform.persistence.sqlite_store import SqliteStore
|
||||
|
||||
|
||||
class ChatSessionFullSmokeTests(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self._tmp = tempfile.TemporaryDirectory(ignore_cleanup_errors=True)
|
||||
self.db = Path(self._tmp.name) / "ops.sqlite"
|
||||
os.environ["OPS_ASSISTANT_DB_PATH"] = str(self.db)
|
||||
os.environ["OPS_ASSISTANT_PASSWORD"] = "test-admin-pass"
|
||||
os.environ["OPS_ASSISTANT_MODE"] = "rule"
|
||||
store = SqliteStore(str(self.db))
|
||||
t = store.create_tenant("Team")
|
||||
self.tenant_id = str(t["id"])
|
||||
store.create_user_account(
|
||||
tenant_id=self.tenant_id,
|
||||
username="administrator",
|
||||
display_name="Admin",
|
||||
role="owner",
|
||||
password_hash=hashlib.sha256("test-admin-pass".encode("utf-8")).hexdigest(),
|
||||
is_active=True,
|
||||
)
|
||||
self.client = TestClient(create_app())
|
||||
self.client.post("/admin/api/auth/bootstrap", json={})
|
||||
self.token = self._login()
|
||||
|
||||
def tearDown(self) -> None:
|
||||
for k in ("OPS_ASSISTANT_DB_PATH", "OPS_ASSISTANT_MODE"):
|
||||
os.environ.pop(k, None)
|
||||
self._tmp.cleanup()
|
||||
|
||||
def _headers(self) -> dict[str, str]:
|
||||
return {"authorization": f"Bearer {self.token}"}
|
||||
|
||||
def _login(self) -> str:
|
||||
resp = self.client.post(
|
||||
"/admin/api/auth/login",
|
||||
json={
|
||||
"tenant_id": self.tenant_id,
|
||||
"username": "administrator",
|
||||
"password": "test-admin-pass",
|
||||
"purpose": "console",
|
||||
},
|
||||
)
|
||||
data = resp.json()
|
||||
self.assertTrue(data.get("ok"), data)
|
||||
return str(data.get("token") or "")
|
||||
|
||||
def test_create_session_send_message_list_messages(self) -> None:
|
||||
cr = self.client.post("/admin/api/chat/sessions", json={"title": "smoke"}, headers=self._headers())
|
||||
self.assertEqual(cr.status_code, 200, cr.text)
|
||||
cj = cr.json()
|
||||
self.assertTrue(cj.get("ok"), cj)
|
||||
sid = str((cj.get("session") or {}).get("id") or "")
|
||||
self.assertTrue(sid, cj)
|
||||
|
||||
mr = self.client.post(
|
||||
f"/admin/api/chat/sessions/{sid}/messages",
|
||||
json={"text": "ping"},
|
||||
headers=self._headers(),
|
||||
)
|
||||
self.assertEqual(mr.status_code, 200, mr.text)
|
||||
mj = mr.json()
|
||||
self.assertTrue(mj.get("ok"), mj)
|
||||
reply = str(mj.get("reply") or "")
|
||||
self.assertTrue(len(reply) > 0, mj)
|
||||
|
||||
lr = self.client.get(f"/admin/api/chat/sessions/{sid}/messages", headers=self._headers())
|
||||
self.assertEqual(lr.status_code, 200, lr.text)
|
||||
lj = lr.json()
|
||||
self.assertTrue(lj.get("ok"), lj)
|
||||
msgs = lj.get("messages") or []
|
||||
roles = [str(m.get("role") or "") for m in msgs if isinstance(m, dict)]
|
||||
self.assertIn("user", roles)
|
||||
self.assertIn("assistant", roles)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue