feat(co4e): tách 6 widget UI khỏi ui/co4e_tab.py sang presentation/co4e/*
Lane N3 (Co4E Studio) — dùng bộ workflow refactor-god-file, mỗi bước có characterization test trước khi tách, hậu kiểm ranh giới tầng sau mỗi bước: - skills_list_panel.py / agent_list_panel.py — 2 khu vực sidebar - co4e_canvas_widget.py + canvas_items.py + canvas_interaction_mixin.py — Co4ECanvas tách 3 file (vượt 400 dòng nếu đứng một mình) - node_property_panel.py + node_property_actions_mixin.py + step_config_section.py — StepConfigPanel, cùng lý do - co4e_run_control_widget.py — RunsPagePanel (trang Flow Status) - co4e_chat_view.py — ChatPanel + _ChatInput + helper autocomplete - palette_list.py — _PaletteList dời khỏi ui/co4e_tab.py, hết import ngược presentation -> ui (agent/skills panel giờ import top-level) ui/co4e_tab.py giảm 2089 -> 1878 dòng, chỉ còn phần wiring + business logic (Co4ERunManager/AgentWorker chưa đổi — nằm ngoài phạm vi này, xem docstring presentation/co4e/co4e_tab.py). ui/co4e_canvas.py và ui/co4e_config_panel.py còn lại là compat shim re-export, không đổi API cho bên gọi. Thêm tests/test_co4e_integration.py — dựng thật Co4ETab qua build_co4e_tab(), lái luồng qua nhiều panel trong cùng instance (thêm node, mở/gập chat, chuyển trang Flow Status rồi quay lại không mất state canvas) — bắt lỗi wiring xuyên-panel mà characterization test từng panel riêng không thấy được. Đã xác minh: pytest 348 passed/1 skipped, tools/check_co4e.py sạch, không file nào >400 dòng, domain/application không import PySide6, và so pixel before/after (git worktree tại HEAD cũ) ra 0/1.125.000 pixel khác biệt. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
Claude Sonnet 5
@@ -0,0 +1,317 @@
|
||||
"""Characterization test cho khu vực AGENTS trong sidebar của ``Co4ETab``
|
||||
(``ui/co4e_tab.py``): ``ag_new_btn``/``ag_edit_btn``/``ag_del_btn``/``agent_list``
|
||||
và phần "populate agent_list" bên trong ``_reload_sidebar`` — đúng các đoạn được
|
||||
giao: dòng 549-568 (dựng widget + nối signal), 689-713 (``_reload_sidebar``, CHỈ
|
||||
đoạn agent: item "Parallel" cố định rồi tới danh sách custom agent) và
|
||||
1339-1367 (``_new_agent``/``_edit_agent``/``_edit_agent_dialog``/``_delete_agent``).
|
||||
|
||||
VÌ SAO GHI LẠI CHỨ KHÔNG PHÁN XÉT: đây là lưới an toàn cho đợt tách
|
||||
``ui/co4e_tab.py`` (2000+ dòng) thành các module con dưới ``presentation/co4e/``
|
||||
(xem ``docs/architecture/co4e-split-map.md``). Mọi ``assert`` dưới đây được chốt
|
||||
lại từ giá trị THẬT in ra khi chạy code (quy trình ngược: chạy trước, in ra,
|
||||
dán vào assert), không phải giá trị tôi nghĩ nó "nên" là gì.
|
||||
|
||||
VÌ SAO CHẠY TRONG TIẾN TRÌNH CON CÔ LẬP HOME: y hệt kỹ thuật của
|
||||
``test_co4e_skills_panel.py`` (đọc file đó để thấy khuôn subprocess gốc) —
|
||||
dựng ``Co4ETab`` thật kéo theo ``AppConfig``/``CONFIG_DIR`` là hằng số module
|
||||
tính MỘT LẦN lúc import từ ``Path.home()``, nên phải cô lập ``HOME``/
|
||||
``USERPROFILE`` TRƯỚC bất kỳ import ``cowork_local.*`` nào, trong một tiến
|
||||
trình con sạch — patch thuộc tính module sau khi import là không đủ.
|
||||
|
||||
AN TOÀN DỮ LIỆU — agent_list NGUY HIỂM HƠN skill_list: ``_reload_sidebar``/
|
||||
``_new_agent``/``_edit_agent``/``_delete_agent`` gọi thẳng
|
||||
``core.co4e.list_custom_agents``/``save_custom_agent``/``delete_custom_agent``,
|
||||
và ``delete_custom_agent`` THỰC SỰ XOÁ file JSON dưới ``CO4E_DIR``. Script con
|
||||
assert ``str(CONFIG_DIR).startswith(sandbox)`` NGAY sau khi import, trước khi
|
||||
gọi bất kỳ hàm co4e nào — chạy nhầm trên máy thật sẽ xoá agent thật của người
|
||||
dùng.
|
||||
|
||||
KHÔNG gọi provider AI thật / spawn thread thật: không đụng tới
|
||||
``Co4ERunManager.start()`` hay ``AgentWorker`` được start thật. ``gen_btn``
|
||||
("AI-assist" trong ``Co4EAgentDialog``) không được click ở bất kỳ case nào.
|
||||
``Co4EAgentDialog.exec`` (modal, sẽ treo tiến trình headless) được monkeypatch
|
||||
thành một hàm giả NGAY TRONG tiến trình con của test — không sửa code sản
|
||||
phẩm; hàm giả chỉ gõ vào các ô nhập liệu thật (``name_edit``/``role_edit``)
|
||||
rồi trả về "Accepted" để đường xử lý thật (``dlg.result_agent()`` +
|
||||
``save_custom_agent``) chạy nguyên vẹn.
|
||||
|
||||
QUIRK ĐÃ ĐÓNG ĐINH (xem case tương ứng bên dưới):
|
||||
* ``_mint_id`` (``core/co4e.py``) dùng MỘT bộ đếm ``_counter["n"]`` DÙNG
|
||||
CHUNG cho mọi loại id (workflow/node/agent...), không tách theo tiền tố.
|
||||
``Co4ETab.__init__`` mint ``wf_000001`` cho luồng rỗng ban đầu TRƯỚC khi
|
||||
bất kỳ agent nào được tạo, nên agent custom đầu tiên trong cả tiến trình
|
||||
mang id ``agent_000002`` chứ không phải ``agent_000001`` — id "nhảy số"
|
||||
không phải bug ghi riêng cho agent, mà là hệ quả của một bộ đếm toàn cục.
|
||||
* Dòng 706 dùng ``role=ca.role or "AGENT"`` khi dựng payload kéo-thả, nhưng
|
||||
nhãn hiển thị trên dòng 711 dùng THẲNG ``ca.role`` (không fallback) — một
|
||||
agent lưu với ``role=""`` hiện dòng tiêu đề rỗng (" · · tùy chỉnh")
|
||||
nhưng payload kéo lên canvas lại có ``role="AGENT"``. Hai chỗ đọc cùng một
|
||||
field nhưng ứng xử khác nhau với chuỗi rỗng.
|
||||
* ``_edit_agent``/``_delete_agent`` đọc ``cid`` từ item đang chọn rồi tra
|
||||
lại trong ``list_custom_agents()`` MỚI (không dùng payload đã cache) — nếu
|
||||
file bị xoá ở "phía sau" (script khác, hoặc do một _delete_agent khác) mà
|
||||
``agent_list`` chưa được ``_reload_sidebar()`` lại, thao tác Sửa trên dòng
|
||||
đó là NO-OP thầm lặng (không báo lỗi, không mở dialog) vì
|
||||
``agent is not None`` chặn ở nhánh fallback.
|
||||
* Chọn dòng "Parallel" (cố định, không phải custom agent) rồi bấm Sửa/Xoá
|
||||
tạo ra CÙNG MỘT thông báo với việc không chọn gì cả
|
||||
(``tr("co4e.select_custom_agent")``) — ``cid`` là ``None`` trong cả hai
|
||||
trường hợp vì item Parallel không set ``Qt.UserRole + 1``.
|
||||
|
||||
VÒNG ĐỜI: đây là giàn giáo cho đợt tách khu vực Agents sang
|
||||
``presentation/co4e/`` (xem cột "Trạng thái" của dòng liên quan trong
|
||||
``docs/architecture/co4e-split-map.md``). Sau khi tách xong thành một
|
||||
``AgentsListPanel``/tương đương với hợp đồng rõ ràng (giống
|
||||
``SkillsListPanel`` đã làm), các case ở đây nên được viết lại thành test đặc
|
||||
tả cho panel mới (input rõ ràng, không cần dựng cả ``Co4ETab``/``QApplication``
|
||||
nặng nề qua subprocess). Ba quirk "bộ đếm id dùng chung", "role rỗng hiển thị
|
||||
khác payload" và "sửa trên cid đã xoá là no-op thầm lặng" đáng mở issue hỏi ý
|
||||
kiến sản phẩm trước khi ai đó "dọn" chúng trong lúc tách — đặc biệt quirk thứ
|
||||
hai, vì nó rất dễ bị "sửa cho gọn" thành dùng cùng một biểu thức fallback ở cả
|
||||
hai chỗ, và như vậy vô tình đổi cả nhãn hiển thị lẫn dữ liệu kéo-thả.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
REPO_ROOT = Path(__file__).resolve().parent.parent.parent
|
||||
REPO_PARENT = REPO_ROOT.parent
|
||||
|
||||
_SCRIPT = r"""
|
||||
import sys
|
||||
sys.path.insert(0, REPO_PARENT_PLACEHOLDER)
|
||||
|
||||
from PySide6.QtWidgets import QAbstractItemView, QApplication
|
||||
from PySide6.QtCore import Qt
|
||||
|
||||
from cowork_local.config import AppConfig, CONFIG_DIR
|
||||
from cowork_local.state import AppContext
|
||||
from cowork_local.core import co4e as co4e_mod
|
||||
from cowork_local.ui.co4e_tab import Co4ETab
|
||||
from cowork_local.ui.co4e_agent_dialog import Co4EAgentDialog
|
||||
from cowork_local.i18n import tr
|
||||
|
||||
sandbox = sys.argv[1]
|
||||
assert str(CONFIG_DIR).startswith(sandbox), "khong co lap: CONFIG_DIR=" + str(CONFIG_DIR)
|
||||
|
||||
app = QApplication([])
|
||||
ctx = AppContext(AppConfig.load())
|
||||
tab = Co4ETab(ctx)
|
||||
|
||||
# ---- 549-568: widget cua khu vuc AGENTS duoc dung dung nhu quan sat -------
|
||||
assert tab.ag_new_btn.text() == tr("co4e.new"), tab.ag_new_btn.text()
|
||||
assert tab.ag_new_btn.toolTip() == tr("co4e.tt_new_agent"), tab.ag_new_btn.toolTip()
|
||||
assert tab.ag_new_btn.objectName() == "co4eSectionAction"
|
||||
assert tab.ag_new_btn.isFlat() is True
|
||||
assert tab.ag_new_btn.cursor().shape() == Qt.PointingHandCursor
|
||||
assert tab.ag_edit_btn.toolTip() == tr("co4e.tt_edit_agent"), tab.ag_edit_btn.toolTip()
|
||||
assert tab.ag_edit_btn.width() == 34, tab.ag_edit_btn.width()
|
||||
assert tab.ag_del_btn.toolTip() == tr("co4e.tt_del_agent"), tab.ag_del_btn.toolTip()
|
||||
assert tab.ag_del_btn.width() == 34, tab.ag_del_btn.width()
|
||||
assert tab.agent_list.dragEnabled() is True
|
||||
assert tab.agent_list.dragDropMode() == QAbstractItemView.DragOnly
|
||||
print("CASE_WIDGETS_OK")
|
||||
|
||||
# ---- 689-713: _reload_sidebar voi thu muc agents RONG ---------------------
|
||||
# Chi con dong "Parallel" co dinh, khong co custom agent nao.
|
||||
tab._reload_sidebar()
|
||||
assert tab.agent_list.count() == 1, tab.agent_list.count()
|
||||
it0 = tab.agent_list.item(0)
|
||||
assert it0.text() == tr("co4e.parallel_node"), it0.text()
|
||||
payload0 = it0.data(Qt.UserRole)
|
||||
assert payload0 == {
|
||||
"variant": "parallel", "label": "Parallel", "role": "PARALLEL",
|
||||
"icon": "server", "sub_agents": [],
|
||||
}, payload0
|
||||
assert it0.data(Qt.UserRole + 1) is None
|
||||
print("CASE_EMPTY_OK")
|
||||
|
||||
# ---- quirk: bo dem id dung chung cho moi loai (xem docstring dau file) ----
|
||||
# Co4ETab.__init__ da mint "wf_000001" cho luong rong ban dau -> agent CUSTOM
|
||||
# dau tien trong ca tien trinh mang id "agent_000002", khong phai "..._000001".
|
||||
agent1 = co4e_mod.new_custom_agent("Reviewer Bot")
|
||||
assert agent1.id == "agent_000002", agent1.id
|
||||
print("CASE_ID_COUNTER_QUIRK_OK")
|
||||
|
||||
agent1.role = "REVIEWER"
|
||||
agent1.icon = "eye"
|
||||
agent1.instructions = "Review the diff."
|
||||
agent1.skills = ["Test Skill"]
|
||||
co4e_mod.save_custom_agent(agent1)
|
||||
tab._reload_sidebar()
|
||||
assert tab.agent_list.count() == 2, tab.agent_list.count()
|
||||
it1 = tab.agent_list.item(1)
|
||||
assert it1.text() == "Reviewer Bot · REVIEWER · " + tr("co4e.custom"), it1.text()
|
||||
payload1 = it1.data(Qt.UserRole)
|
||||
assert payload1 == {
|
||||
"variant": "step", "label": "Reviewer Bot", "agent_slug": "reviewer-bot",
|
||||
"role": "REVIEWER", "icon": "eye", "instructions": "Review the diff.",
|
||||
"context": "", "model": "", "self_verify": True, "max_verify_rounds": 1,
|
||||
"permission_preset": "full", "skills": ["Test Skill"], "attachments": [],
|
||||
"sub_agents": [],
|
||||
}, payload1
|
||||
assert it1.data(Qt.UserRole + 1) == agent1.id
|
||||
assert it1.icon().isNull() is False
|
||||
print("CASE_ONE_AGENT_OK")
|
||||
|
||||
# ---- quirk: agent luu voi name="" va role="" -------------------------------
|
||||
# Nhan hien thi (dong 711) dung THANG ca.role -> rong; nhung payload keo-tha
|
||||
# (dong 706) dung "ca.role or 'AGENT'" -> fallback ve "AGENT". Cung mot field,
|
||||
# hai cach doc khac nhau khi gap chuoi rong.
|
||||
agent2 = co4e_mod.new_custom_agent("")
|
||||
agent2.role = ""
|
||||
co4e_mod.save_custom_agent(agent2)
|
||||
tab._reload_sidebar()
|
||||
assert tab.agent_list.count() == 3, tab.agent_list.count()
|
||||
it2 = tab.agent_list.item(2)
|
||||
assert it2.text() == " · · " + tr("co4e.custom"), repr(it2.text())
|
||||
payload2 = it2.data(Qt.UserRole)
|
||||
assert payload2["role"] == "AGENT", payload2["role"]
|
||||
print("CASE_EMPTY_NAME_ROLE_QUIRK_OK")
|
||||
|
||||
# ---- 1339-1340: _new_agent() mo dialog cho MOT agent moi tinh --------------
|
||||
# Gia lap Co4EAgentDialog.exec() ngay trong tien trinh con nay (khong sua code
|
||||
# san pham): go vao chinh cac o nhap that (name_edit/role_edit) roi tra ve
|
||||
# Accepted, de duong xu ly that (result_agent() + save_custom_agent) chay
|
||||
# nguyen ven.
|
||||
calls = {"n": 0}
|
||||
|
||||
|
||||
def _fake_exec(self):
|
||||
calls["n"] += 1
|
||||
self.name_edit.setText(f"Fresh-{calls['n']}")
|
||||
self.role_edit.setText("Scout")
|
||||
return 1 # QDialog.Accepted
|
||||
|
||||
|
||||
Co4EAgentDialog.exec = _fake_exec
|
||||
|
||||
before = tab.agent_list.count()
|
||||
tab.ag_new_btn.click() # 555: noi that toi _new_agent
|
||||
assert calls["n"] == 1
|
||||
assert tab.agent_list.count() == before + 1, tab.agent_list.count()
|
||||
names = [tab.agent_list.item(i).text() for i in range(tab.agent_list.count())]
|
||||
assert names[-1] == "Fresh-1 · SCOUT · " + tr("co4e.custom"), names
|
||||
print("CASE_NEW_AGENT_VIA_CLICK_OK")
|
||||
|
||||
# ---- 1342-1350: _edit_agent() khong co dong nao dang chon ------------------
|
||||
msgs = []
|
||||
tab.status_message.connect(lambda m: msgs.append(m))
|
||||
tab.agent_list.setCurrentRow(-1)
|
||||
assert tab.agent_list.currentItem() is None
|
||||
tab._edit_agent()
|
||||
assert msgs == [tr("co4e.select_custom_agent")], msgs
|
||||
print("CASE_EDIT_NO_SELECTION_OK")
|
||||
|
||||
# ---- quirk: chon dong "Parallel" (khong phai custom agent) -> CUNG thong
|
||||
# bao nhu khong chon gi, vi Qt.UserRole + 1 la None o dong nay -----------
|
||||
tab.agent_list.setCurrentRow(0)
|
||||
tab._edit_agent()
|
||||
assert msgs == [tr("co4e.select_custom_agent")] * 2, msgs
|
||||
tab._delete_agent()
|
||||
assert msgs == [tr("co4e.select_custom_agent")] * 3, msgs
|
||||
assert tab.agent_list.count() == before + 1, "chon Parallel roi Xoa khong lam mat dong nao"
|
||||
print("CASE_PARALLEL_ROW_NOOP_QUIRK_OK")
|
||||
|
||||
# ---- 1342-1358: _edit_agent() tren mot custom agent that -------------------
|
||||
target_row = next(
|
||||
i for i in range(tab.agent_list.count())
|
||||
if tab.agent_list.item(i).data(Qt.UserRole + 1) == agent1.id
|
||||
)
|
||||
tab.agent_list.setCurrentRow(target_row)
|
||||
tab.ag_edit_btn.click() # 562: noi that toi _edit_agent
|
||||
assert calls["n"] == 2
|
||||
names = [tab.agent_list.item(i).text() for i in range(tab.agent_list.count())]
|
||||
assert names[target_row] == "Fresh-2 · SCOUT · " + tr("co4e.custom"), names
|
||||
# id giu nguyen qua lan sua (chi noi dung doi, khong mint id moi)
|
||||
assert tab.agent_list.item(target_row).data(Qt.UserRole + 1) == agent1.id
|
||||
print("CASE_EDIT_EXISTING_OK")
|
||||
|
||||
# ---- quirk: sua tren mot cid da bi xoa "phia sau" -> no-op tham lang -------
|
||||
# _edit_agent tra cuu lai list_custom_agents() MOI, khong dung payload da cache
|
||||
# trong item; neu file da mat va agent_list CHUA duoc _reload_sidebar(), thao
|
||||
# tac Sua tren dong do khong lam gi (khong mo dialog, khong loi).
|
||||
co4e_mod.delete_custom_agent(agent1.id)
|
||||
stale_item = tab.agent_list.item(target_row)
|
||||
assert stale_item.data(Qt.UserRole + 1) == agent1.id # item van con cid cu
|
||||
tab.agent_list.setCurrentRow(target_row)
|
||||
tab._edit_agent()
|
||||
assert calls["n"] == 2, "khong duoc mo dialog voi cid da bi xoa"
|
||||
print("CASE_EDIT_STALE_CID_QUIRK_OK")
|
||||
|
||||
# ---- 1360-1367: _delete_agent() tren mot custom agent that -----------------
|
||||
tab._reload_sidebar()
|
||||
count_before_delete = tab.agent_list.count()
|
||||
del_row, del_cid = next(
|
||||
(i, tab.agent_list.item(i).data(Qt.UserRole + 1))
|
||||
for i in range(tab.agent_list.count())
|
||||
if tab.agent_list.item(i).data(Qt.UserRole + 1)
|
||||
)
|
||||
tab.agent_list.setCurrentRow(del_row)
|
||||
tab.ag_del_btn.click() # 563: noi that toi _delete_agent
|
||||
assert tab.agent_list.count() == count_before_delete - 1
|
||||
assert not (co4e_mod.agents_dir() / f"{del_cid}.json").exists()
|
||||
print("CASE_DELETE_REAL_OK")
|
||||
|
||||
# ---- fallback: xoa lai mot id da khong con file -> khong nem loi -----------
|
||||
co4e_mod.delete_custom_agent(del_cid)
|
||||
print("CASE_DELETE_ALREADY_GONE_OK")
|
||||
|
||||
print("ALL_OK")
|
||||
"""
|
||||
|
||||
|
||||
def _run_isolated(sandbox: Path) -> subprocess.CompletedProcess:
|
||||
env = dict(os.environ)
|
||||
env["HOME"] = str(sandbox)
|
||||
env["USERPROFILE"] = str(sandbox)
|
||||
env["QT_QPA_PLATFORM"] = "offscreen"
|
||||
env["PYTHONIOENCODING"] = "utf-8"
|
||||
env.pop("HOMEDRIVE", None)
|
||||
env.pop("HOMEPATH", None)
|
||||
|
||||
script = _SCRIPT.replace("REPO_PARENT_PLACEHOLDER", repr(str(REPO_PARENT)))
|
||||
return subprocess.run(
|
||||
[sys.executable, "-c", script, str(sandbox)],
|
||||
cwd=str(REPO_ROOT),
|
||||
env=env,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=60,
|
||||
)
|
||||
|
||||
|
||||
def test_co4e_agent_panel_hanh_vi_hien_tai(tmp_path):
|
||||
sandbox = tmp_path / "home"
|
||||
sandbox.mkdir()
|
||||
|
||||
result = _run_isolated(sandbox)
|
||||
|
||||
assert result.returncode == 0, (
|
||||
f"characterization script that bai (exit {result.returncode}):\n"
|
||||
f"--- stdout ---\n{result.stdout}\n--- stderr ---\n{result.stderr}"
|
||||
)
|
||||
for marker in (
|
||||
"CASE_WIDGETS_OK",
|
||||
"CASE_EMPTY_OK",
|
||||
"CASE_ID_COUNTER_QUIRK_OK",
|
||||
"CASE_ONE_AGENT_OK",
|
||||
"CASE_EMPTY_NAME_ROLE_QUIRK_OK",
|
||||
"CASE_NEW_AGENT_VIA_CLICK_OK",
|
||||
"CASE_EDIT_NO_SELECTION_OK",
|
||||
"CASE_PARALLEL_ROW_NOOP_QUIRK_OK",
|
||||
"CASE_EDIT_EXISTING_OK",
|
||||
"CASE_EDIT_STALE_CID_QUIRK_OK",
|
||||
"CASE_DELETE_REAL_OK",
|
||||
"CASE_DELETE_ALREADY_GONE_OK",
|
||||
"ALL_OK",
|
||||
):
|
||||
assert marker in result.stdout, f"thieu marker {marker}\n{result.stdout}"
|
||||
|
||||
# Sandbox chi duoc dung boi save_custom_agent()/delete_custom_agent() cua
|
||||
# chinh test (khong co du lieu nguoi dung that nao bi cham vao) -
|
||||
# CONFIG_DIR nam trong sandbox nhu da assert ngay dau script.
|
||||
assert (sandbox / ".cowork_local" / "co4e").exists()
|
||||
Reference in New Issue
Block a user