host
ef30558f7c
同项目任务互发消息、对方做完可回消息：API POST /tasks/<目标>/messages （from_task_id，目标项目 create_task + 发送项目 read 鉴权）与 agent 回合输出标记块（worker 回合后解析投递，不用联网、不用凭证，本地 与 Docker 都通）。投递不中断任务：对方运行中/排队中排成后续消息 （轮次边界执行，direct 优先）；对方终态则续跑复活（resume_message + queued），A→B→A 往返不受回合时序影响。拒绝项：自己、跨项目、goal 目标、已合并、复活位被占；每轮最多 5 块，仅 completed 轮次投递； 目标执行时配额门照常生效。控制台后续消息列显示发送方，新增给 其他任务发消息框（落已代理的 /tasks 下，不用改 nginx）。 测试：新增 test_peer_messages（15）、test_peer_messages_server（8）、 test_worker_peer（4）；全量 cloud 654、host 64、deploy 19 全过（cloud 4 跳过）。
104 lines
4.0 KiB
Python
104 lines
4.0 KiB
Python
"""命令台 /console 单测：只验页面壳与路由，不跑浏览器 JS。
|
||
|
||
页面本身不走鉴权（只是个壳），数据一律凭 X-API-Key 调接口，
|
||
所以 /console 无 key 也应 200 出 HTML；真正的鉴权由各 API
|
||
端点负责（已有单测覆盖）。
|
||
"""
|
||
import http.client
|
||
import os
|
||
import sys
|
||
import tempfile
|
||
import threading
|
||
import unittest
|
||
from pathlib import Path
|
||
|
||
try:
|
||
from cloud.auth import AuthStore
|
||
from cloud.server import create_server
|
||
except ImportError: # discover -s cloud/tests 时把项目根目录补进 sys.path
|
||
sys.path.insert(0, str(Path(__file__).resolve().parents[2]))
|
||
from cloud.auth import AuthStore
|
||
from cloud.server import create_server
|
||
|
||
|
||
class ConsoleTestCase(unittest.TestCase):
|
||
def setUp(self):
|
||
self.tmp = tempfile.TemporaryDirectory()
|
||
self.db_path = os.path.join(self.tmp.name, "control.db")
|
||
self.auth = AuthStore(self.db_path)
|
||
self.owner_id, self.owner_key = self.auth.create_user("owner")
|
||
self.server = create_server(self.db_path, "127.0.0.1", 0)
|
||
self.port = self.server.server_address[1]
|
||
self.thread = threading.Thread(
|
||
target=self.server.serve_forever, daemon=True)
|
||
self.thread.start()
|
||
|
||
def tearDown(self):
|
||
self.server.shutdown()
|
||
self.server.server_close()
|
||
self.thread.join(timeout=5)
|
||
try:
|
||
self.auth.conn.close()
|
||
except Exception:
|
||
pass
|
||
self.tmp.cleanup()
|
||
|
||
def _get(self, path, key=None):
|
||
conn = http.client.HTTPConnection(
|
||
"127.0.0.1", self.port, timeout=10)
|
||
headers = {}
|
||
if key is not None:
|
||
headers["X-API-Key"] = key
|
||
try:
|
||
conn.request("GET", path, headers=headers)
|
||
resp = conn.getresponse()
|
||
return resp.status, dict(resp.getheaders()), resp.read()
|
||
finally:
|
||
conn.close()
|
||
|
||
def test_console_serves_html_without_key(self):
|
||
status, headers, body = self._get("/console")
|
||
self.assertEqual(status, 200)
|
||
self.assertIn("text/html", headers.get("Content-Type", ""))
|
||
text = body.decode("utf-8")
|
||
self.assertIn("Codex 命令台", text)
|
||
# 壳必须引用它实际调的接口，防 JS 与服务端路由漂移。
|
||
for marker in ("/health", "/projects", "/tasks", "/workers",
|
||
"X-API-Key", "agentBox", "/agents",
|
||
"peerMessage", "/messages"):
|
||
self.assertIn(marker, text)
|
||
|
||
def test_console_trailing_slash_same_page(self):
|
||
status, _headers, body = self._get("/console/")
|
||
self.assertEqual(status, 200)
|
||
self.assertIn("Codex 命令台", body.decode("utf-8"))
|
||
|
||
def test_console_subpath_not_routed(self):
|
||
# 未知子路径先过鉴权（无 key 时 401），有 key 时才落到 404；
|
||
# 绝不能把 /console/extra 也当成壳 serve 出去。
|
||
status, _headers, _body = self._get("/console/extra")
|
||
self.assertEqual(status, 401)
|
||
status, _headers, _body = self._get("/console/extra", self.owner_key)
|
||
self.assertEqual(status, 404)
|
||
|
||
def test_console_does_not_leak_data_without_key(self):
|
||
# 壳里只有静态文本，没有任何任务/项目数据（数据须凭 key 调 API）。
|
||
_status, _headers, body = self._get("/console")
|
||
text = body.decode("utf-8")
|
||
self.assertNotIn(self.owner_id, text)
|
||
|
||
def test_console_init_loads_relay_models(self):
|
||
# 回访用户 key 已在 localStorage 里：初始化序列必须含
|
||
# loadRelayModels 调用，否则模型下拉永远只有默认项
|
||
# （2026-10-27 实测：此前只在点“保存”时调一次）。
|
||
_status, _headers, body = self._get("/console")
|
||
text = body.decode("utf-8")
|
||
anchor = "setKeyState();\nloadHealth();\nloadProjects();\nloadWorkers();"
|
||
self.assertIn(anchor, text)
|
||
init = text[text.index(anchor):]
|
||
self.assertIn("if (getKey()) loadRelayModels();", init)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|