mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
oliver
ba3836f00f
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore，避免提交本地运行态数据（_local、node_modules、logs 等） - 提交当前工程代码与配置 Made-with: Cursor
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
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
|
|
|
|
MINI_PNG = base64.b64decode(
|
|
"iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg=="
|
|
)
|
|
|
|
|
|
class ChatProfileApiTests(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"
|
|
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={})
|
|
resp = self.client.post(
|
|
"/admin/api/auth/login",
|
|
json={
|
|
"tenant_id": self.tenant_id,
|
|
"username": "administrator",
|
|
"password": "test-admin-pass",
|
|
"purpose": "console",
|
|
},
|
|
)
|
|
self.assertEqual(resp.status_code, 200, resp.text)
|
|
self.token = str(resp.json().get("token") or "")
|
|
|
|
def tearDown(self) -> None:
|
|
self._tmp.cleanup()
|
|
|
|
def _h(self) -> dict[str, str]:
|
|
return {"authorization": f"Bearer {self.token}"}
|
|
|
|
def test_profile_get_patch_avatar_delete(self) -> None:
|
|
g = self.client.get("/admin/api/chat/profile", headers=self._h())
|
|
self.assertEqual(g.status_code, 200, g.text)
|
|
data = g.json()
|
|
self.assertTrue(data.get("ok"))
|
|
prof = data.get("profile") or {}
|
|
self.assertEqual(prof.get("username"), "administrator")
|
|
self.assertFalse(prof.get("avatar_attachment_id"))
|
|
|
|
p = self.client.patch(
|
|
"/admin/api/chat/profile",
|
|
json={"display_name": "Admin Renamed"},
|
|
headers=self._h(),
|
|
)
|
|
self.assertEqual(p.status_code, 200, p.text)
|
|
self.assertEqual((p.json().get("profile") or {}).get("display_name"), "Admin Renamed")
|
|
|
|
up = self.client.post(
|
|
"/admin/api/chat/profile/avatar",
|
|
files={"file": ("x.png", MINI_PNG, "image/png")},
|
|
headers=self._h(),
|
|
)
|
|
self.assertEqual(up.status_code, 200, up.text)
|
|
aid = str((up.json() or {}).get("avatar_attachment_id") or "").strip()
|
|
self.assertTrue(aid)
|
|
|
|
g2 = self.client.get("/admin/api/chat/profile", headers=self._h())
|
|
self.assertEqual((g2.json().get("profile") or {}).get("avatar_attachment_id"), aid)
|
|
|
|
att = self.client.get(f"/admin/api/chat/attachments/{aid}", headers=self._h())
|
|
self.assertEqual(att.status_code, 200)
|
|
self.assertTrue(len(att.content) > 10)
|
|
|
|
d = self.client.delete("/admin/api/chat/profile/avatar", headers=self._h())
|
|
self.assertEqual(d.status_code, 200, d.text)
|
|
g3 = self.client.get("/admin/api/chat/profile", headers=self._h())
|
|
self.assertFalse((g3.json().get("profile") or {}).get("avatar_attachment_id"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|