378 lines
13 KiB
Python
378 lines
13 KiB
Python
"""Offline Chinese onboarding dialog regression; no Chrome or network access."""
|
|
|
|
import json
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
|
|
|
from PySide6.QtCore import Qt
|
|
from PySide6.QtWidgets import (
|
|
QApplication,
|
|
QDialog,
|
|
QLineEdit,
|
|
QMessageBox,
|
|
)
|
|
|
|
from account_store import DEFAULT_RULE
|
|
from accounts_app import Backend, Window
|
|
from get_current_user import parse_user_response
|
|
|
|
|
|
class OnboardingUiTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.app = QApplication.instance() or QApplication([])
|
|
|
|
def setUp(self):
|
|
self.temp = tempfile.TemporaryDirectory()
|
|
self.backend = Backend(Path(self.temp.name))
|
|
self.window = Window(self.backend)
|
|
|
|
def tearDown(self):
|
|
self.window.allow_close = True
|
|
self.window.close()
|
|
self.temp.cleanup()
|
|
|
|
def test_both_roles_only_ask_for_name(self):
|
|
def fill(dialog):
|
|
inputs = dialog.findChildren(QLineEdit)
|
|
self.assertEqual(len(inputs), 1, "UID must not be a form input")
|
|
inputs[0].setText("测试账号名称")
|
|
return QDialog.DialogCode.Accepted
|
|
|
|
for role in ("main", "worker"):
|
|
with (
|
|
patch.object(QDialog, "exec", fill),
|
|
patch.object(self.window, "send") as send,
|
|
):
|
|
self.window.add(role)
|
|
send.assert_called_once_with(
|
|
"add", {"role": role, "name": "测试账号名称", "owner": None}
|
|
)
|
|
|
|
def test_profile_and_zero_statistics_are_displayed(self):
|
|
account = {
|
|
"id": "a" * 32,
|
|
"name": "业务名称",
|
|
"role": "main",
|
|
"owner": None,
|
|
"uid": None,
|
|
"nickname": "",
|
|
"avatar": "",
|
|
"profile": "{}",
|
|
"rule": json.dumps(DEFAULT_RULE),
|
|
"login": "待登录",
|
|
"identity": "等待首次登录",
|
|
"state": "等待登录",
|
|
}
|
|
snapshot = {
|
|
"accounts": [account],
|
|
"tasks": [],
|
|
"counts": {},
|
|
"selected": [],
|
|
"active": [],
|
|
"browser": {},
|
|
"browser_overrides": {},
|
|
}
|
|
self.window.refresh(snapshot)
|
|
card = self.window.card_table.cards[account["id"]]
|
|
self.assertIn("待登录自动获取", card.meta_label.text())
|
|
self.assertIn("等待登录", card.status_label.text())
|
|
account.update(
|
|
uid="301",
|
|
nickname="平台昵称",
|
|
profile=json.dumps(
|
|
{
|
|
"follower_count": 0,
|
|
"following_count": 12,
|
|
"total_favorited": 123,
|
|
"aweme_count": 4,
|
|
}
|
|
),
|
|
)
|
|
self.window.refresh(snapshot)
|
|
card = self.window.card_table.cards[account["id"]]
|
|
self.assertIn("平台昵称", card.meta_label.text())
|
|
for text in ("粉丝 0", "关注 12", "获赞 123", "作品 4"):
|
|
self.assertIn(text, self.window.profile_summary.text())
|
|
|
|
def test_account_tree_expands_and_collapses_workers(self):
|
|
main = {
|
|
"id": "m" * 32,
|
|
"name": "主账号",
|
|
"role": "main",
|
|
"owner": None,
|
|
"uid": "100",
|
|
"nickname": "主号名称",
|
|
"avatar": "",
|
|
"profile": "{}",
|
|
"rule": json.dumps(DEFAULT_RULE),
|
|
"login": "已登录",
|
|
"identity": "已绑定",
|
|
"state": "未启动",
|
|
}
|
|
worker = dict(main)
|
|
worker.update(
|
|
id="w" * 32,
|
|
name="小号别名",
|
|
role="worker",
|
|
owner=main["id"],
|
|
uid="200",
|
|
nickname="小号名称",
|
|
)
|
|
snapshot = {
|
|
"accounts": [main, worker],
|
|
"tasks": [],
|
|
"counts": {},
|
|
"selected": [],
|
|
"active": [],
|
|
"browser": {},
|
|
"browser_overrides": {},
|
|
}
|
|
self.window.refresh(snapshot)
|
|
node = self.window.card_table.nodes[main["id"]]
|
|
self.assertTrue(node.expanded)
|
|
node.card.toggle_button.click()
|
|
self.assertFalse(node.expanded)
|
|
node.card.toggle_button.click()
|
|
self.assertTrue(node.expanded)
|
|
self.assertEqual(
|
|
self.window.card_table.cards[worker["id"]].meta_label.text(),
|
|
"名称：小号名称",
|
|
)
|
|
|
|
def test_log_stream_is_incremental_and_capped(self):
|
|
self.backend.logs_ready.emit([f"entry-{i}" for i in range(2502)])
|
|
self.assertEqual(self.window.log.document().blockCount(), 2000)
|
|
self.assertTrue(self.window.log.toPlainText().startswith("entry-502\n"))
|
|
self.backend.logs_ready.emit(["latest"])
|
|
self.assertEqual(self.window.log.document().blockCount(), 2000)
|
|
self.assertTrue(self.window.log.toPlainText().endswith("latest"))
|
|
self.assertEqual(self.window.log.document().maximumBlockCount(), 2000)
|
|
|
|
def test_worker_delete_requires_explicit_confirmation(self):
|
|
with (
|
|
patch.object(
|
|
self.window,
|
|
"selected",
|
|
return_value={"id": "worker", "name": "测试小号", "role": "worker"},
|
|
),
|
|
patch.object(self.window, "send") as send,
|
|
):
|
|
with patch.object(
|
|
QMessageBox, "question", return_value=QMessageBox.StandardButton.No
|
|
) as question:
|
|
self.window.delete_worker()
|
|
send.assert_not_called()
|
|
self.assertEqual(
|
|
question.call_args.args[-1], QMessageBox.StandardButton.No
|
|
)
|
|
with patch.object(
|
|
QMessageBox, "question", return_value=QMessageBox.StandardButton.Yes
|
|
):
|
|
self.window.delete_worker()
|
|
send.assert_called_once_with("delete_worker", {"id": "worker"})
|
|
with (
|
|
patch.object(
|
|
self.window, "selected", return_value={"id": "main", "role": "main"}
|
|
),
|
|
patch.object(QMessageBox, "information"),
|
|
patch.object(self.window, "send") as send,
|
|
):
|
|
self.window.delete_worker()
|
|
send.assert_not_called()
|
|
|
|
def test_history_and_works_buttons_require_bound_main(self):
|
|
main = {"id": "main", "role": "main", "uid": "101"}
|
|
with (
|
|
patch.object(self.window, "selected", return_value=main),
|
|
patch.object(self.window, "send") as send,
|
|
):
|
|
self.window.history_events()
|
|
self.window.fetch_works()
|
|
self.assertEqual(
|
|
[call.args for call in send.call_args_list],
|
|
[
|
|
("history_fetch", {"id": "main"}),
|
|
("works_open", {"id": "main"}),
|
|
],
|
|
)
|
|
|
|
def test_history_preview_requires_selection_and_explicit_confirmation(self):
|
|
payload = {
|
|
"account": "main",
|
|
"baseline": "rule-and-works-fingerprint",
|
|
"items": [
|
|
{
|
|
"nid": "9007199254740993",
|
|
"create_time": 123,
|
|
"kind": "comment",
|
|
"work_id": "700",
|
|
"work_desc": "完整作品描述",
|
|
"actor_uids": ["999"],
|
|
"actor_names": ["完整昵称"],
|
|
"comment": "完整评论正文",
|
|
"origin": "history",
|
|
"business": {
|
|
"nid_str": "9007199254740993",
|
|
"comment": "完整评论正文",
|
|
},
|
|
}
|
|
],
|
|
}
|
|
|
|
with (
|
|
patch.object(QMessageBox, "information"),
|
|
patch.object(
|
|
QMessageBox, "question", return_value=QMessageBox.StandardButton.Yes
|
|
),
|
|
patch.object(self.window, "send") as send,
|
|
):
|
|
self.window.history_dialog(payload)
|
|
table = self.window.history_table
|
|
content = table.item(0, 6)
|
|
assert content is not None
|
|
self.assertIn("完整评论正文", content.text())
|
|
values = []
|
|
for column in range(table.columnCount()):
|
|
cell = table.item(0, column)
|
|
assert cell is not None
|
|
values.append(cell.text())
|
|
self.assertNotIn("{", " ".join(values))
|
|
check = table.item(0, 0)
|
|
assert check is not None
|
|
check.setCheckState(Qt.CheckState.Checked)
|
|
self.window.confirm_history()
|
|
send.assert_called_once_with(
|
|
"history_enqueue",
|
|
{
|
|
"id": "main",
|
|
"ids": ["9007199254740993"],
|
|
"baseline": "rule-and-works-fingerprint",
|
|
},
|
|
)
|
|
|
|
def test_works_preview_saves_selected_mode_and_full_business_data(self):
|
|
payload = {
|
|
"account": "main",
|
|
"mode": "selected",
|
|
"selected": [],
|
|
"refresh_interval": 3600,
|
|
"items": [
|
|
{
|
|
"aweme_id": "700",
|
|
"desc": "完整作品描述",
|
|
"create_time": 123,
|
|
"statistics": {"collect_count": 8},
|
|
"cover": "https://example.invalid/full.jpg",
|
|
"updated": 456,
|
|
"business": {"aweme_id": "700", "desc": "完整作品描述"},
|
|
}
|
|
],
|
|
}
|
|
|
|
with (
|
|
patch.object(
|
|
self.window,
|
|
"selected",
|
|
return_value={"id": "main", "role": "main"},
|
|
),
|
|
patch.object(self.window, "send") as send,
|
|
):
|
|
self.window.works_dialog(payload)
|
|
table = self.window.works_table
|
|
description = table.item(0, 2)
|
|
assert description is not None
|
|
self.assertIn("完整作品描述", description.text())
|
|
values = []
|
|
for column in range(table.columnCount()):
|
|
cell = table.item(0, column)
|
|
assert cell is not None
|
|
values.append(cell.text())
|
|
self.assertNotIn("{", " ".join(values))
|
|
check = table.item(0, 0)
|
|
assert check is not None
|
|
check.setCheckState(Qt.CheckState.Checked)
|
|
self.window.works_all_mode.setChecked(False)
|
|
self.window.save_work_filter()
|
|
send.assert_called_once_with(
|
|
"work_filter",
|
|
{
|
|
"id": "main",
|
|
"mode": "selected",
|
|
"ids": ["700"],
|
|
"refresh_interval": 3600,
|
|
},
|
|
)
|
|
|
|
def test_lists_scroll_and_copy_rows(self):
|
|
self.window.resize(1260, 860)
|
|
self.window.show()
|
|
items = [
|
|
{
|
|
"aweme_id": str(i),
|
|
"desc": f"作品 {i}",
|
|
"create_time": 123,
|
|
"statistics": {},
|
|
"cover": "",
|
|
"updated": 456,
|
|
}
|
|
for i in range(30)
|
|
]
|
|
self.window.works_dialog(
|
|
{
|
|
"account": "main",
|
|
"items": items,
|
|
"selected": [],
|
|
"mode": "selected",
|
|
"refresh_interval": 3600,
|
|
}
|
|
)
|
|
self.app.processEvents()
|
|
self.assertGreater(self.window.works_card_area.verticalScrollBar().maximum(), 0)
|
|
|
|
tasks = [
|
|
{
|
|
"id": str(i),
|
|
"event": "通知",
|
|
"source": "大号",
|
|
"worker": "小号",
|
|
"target": f"用户{i}",
|
|
"action": "follow",
|
|
"status": "pending",
|
|
"result": "",
|
|
"created_at": 123,
|
|
"started_at": 123,
|
|
"finished_at": 123,
|
|
}
|
|
for i in range(100)
|
|
]
|
|
self.window._fill_task_table(self.window.current_task_table, tasks, {}, True)
|
|
self.app.processEvents()
|
|
self.assertGreater(self.window.current_task_table.verticalScrollBar().maximum(), 0)
|
|
|
|
self.window.copy_table_row(self.window.current_task_table, 0)
|
|
copied = QApplication.clipboard().text()
|
|
self.assertIn("选择: 未选择", copied)
|
|
self.assertIn("任务: 0", copied)
|
|
self.window._copy_text(self.window._work_row_text(items[0]))
|
|
self.assertIn("作品ID: 0", QApplication.clipboard().text())
|
|
|
|
def test_anonymous_profile_is_never_bound(self):
|
|
for user in ({}, {"uid": "0"}, {"uid": True}, {"uid": "not-a-uid"}):
|
|
with self.assertRaises(RuntimeError):
|
|
parse_user_response(
|
|
{
|
|
"status": 200,
|
|
"body": json.dumps({"status_code": 0, "user": user}),
|
|
}
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|