mirror of
https://github.com/happycola233/tchMaterial-parser.git
synced 2026-10-18 14:34:22 +08:00
肥宅水水呀
7d37b0e4de
- 提前登记整批下载任务，使用线程池并确保完成通知只触发一次 - 为共享 HTTP 会话设置默认超时，覆盖解析、目录及封面请求 - 保留搜索重建后的图片引用，避免封面与复选框消失 - 将分类勾选限定在当前筛选结果内 - 同步网址编辑与勾选状态，支持粘贴、删除和撤销 - 消除进度测试对执行顺序的依赖，补充下载、超时及界面回归测试
193 lines
8.1 KiB
Python
193 lines
8.1 KiB
Python
from contextlib import ExitStack
|
|
import io
|
|
from pathlib import Path
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import tkinter as tk
|
|
from tkinter import ttk
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from PIL import Image
|
|
import pytest
|
|
|
|
if __name__ == "__main__":
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
|
|
from src.tchmaterial_parser.ui import resource_tree, runtime, theme
|
|
|
|
|
|
RESOURCES = {"books": {"display_name": "电子教材", "children": {
|
|
"primary": {"display_name": "小学", "children": {
|
|
"a": {"display_name": "语文 一年级上册", "content_id": "a"},
|
|
"b": {"display_name": "语文 一年级下册", "content_id": "b", "custom_properties": {"thumbnails": ["https://example.com/cover.png"]}},
|
|
}},
|
|
}}}
|
|
|
|
|
|
class ResourceTreeUITest(unittest.TestCase):
|
|
__test__ = False # 由下方 pytest 入口在独立进程运行真实 Tk 交互测试
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
try:
|
|
cls.root = tk.Tk()
|
|
except tk.TclError as error:
|
|
if "no display name" in str(error) or "couldn't connect to display" in str(error):
|
|
raise unittest.SkipTest(f"当前环境没有图形显示服务：{error}") from error
|
|
raise
|
|
cls.root.withdraw()
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.root.destroy()
|
|
|
|
def setUp(self):
|
|
self.context = ExitStack()
|
|
self.addCleanup(self.context.close)
|
|
self.addCleanup(self.destroy_widgets)
|
|
enter = self.context.enter_context
|
|
enter(patch.object(runtime, "root", self.root, create=True))
|
|
enter(patch.object(runtime, "ui_scale", 1.0))
|
|
enter(patch.object(runtime, "app_closing", False))
|
|
enter(patch.object(theme, "ui_font_family", "TkDefaultFont", create=True))
|
|
for name, value in (("theme_actions", []), ("themed_widgets", set()), ("current_colors", {}), ("current_theme", "light"), ("switched_theme", "light")):
|
|
enter(patch.object(theme, name, value))
|
|
theme.apply_theme("light")
|
|
enter(patch.object(resource_tree, "thread_it", lambda fn, *args: fn(*args)))
|
|
# 隐藏测试窗口，单独模拟封面进入可视区域，仍执行真实的加载与图片合成代码。
|
|
enter(patch.object(ttk.Treeview, "bbox", return_value=(0, 0, 100, 38)))
|
|
cover = io.BytesIO()
|
|
Image.new("RGB", (80, 112), "#c4ded2").save(cover, format="PNG")
|
|
response = type("CoverResponse", (), {"ok": True, "content": cover.getvalue()})()
|
|
enter(patch.object(resource_tree.session, "get", return_value=response))
|
|
self.pane = ttk.Frame(self.root)
|
|
self.urls = tk.Text(self.root, undo=True)
|
|
self.errors = []
|
|
self.root.report_callback_exception = lambda _type, error, _traceback: self.errors.append(error)
|
|
resource_tree.build_resource_tree(self.pane, RESOURCES, self.urls)
|
|
widgets = list(self.descendants(self.pane))
|
|
self.tree = next(widget for widget in widgets if isinstance(widget, ttk.Treeview))
|
|
self.search = next(widget for widget in widgets if isinstance(widget, ttk.Entry))
|
|
self.count = next(widget for widget in widgets if isinstance(widget, ttk.Label) and widget.grid_info().get("column") == 1)
|
|
self.root.update()
|
|
|
|
def tearDown(self):
|
|
if hasattr(self, "errors"):
|
|
self.assertEqual(self.errors, [])
|
|
|
|
def destroy_widgets(self):
|
|
self.root.update()
|
|
for widget in self.root.winfo_children():
|
|
widget.destroy()
|
|
|
|
def descendants(self, widget):
|
|
for child in widget.winfo_children():
|
|
yield child
|
|
yield from self.descendants(child)
|
|
|
|
def toggle(self, item_id):
|
|
self.tree.focus(item_id)
|
|
# 执行已注册的空格事件回调，无需让隐藏窗口抢占键盘焦点。
|
|
command = re.search(r"\[([^\s]+)", self.tree.bind("<space>")).group(1)
|
|
self.root.tk.call(command, "unused")
|
|
self.root.update()
|
|
|
|
def filter(self, query):
|
|
self.search.delete(0, "end")
|
|
self.search.insert(0, query)
|
|
self.root.after(180, self.root.quit)
|
|
self.root.mainloop()
|
|
self.root.update()
|
|
|
|
def url(self, suffix):
|
|
return resource_tree.build_resource_url(f"books:primary:{suffix}", RESOURCES["books"]["children"]["primary"]["children"][suffix])
|
|
|
|
def lines(self):
|
|
return {line.strip() for line in self.urls.get("1.0", "end").splitlines() if line.strip()}
|
|
|
|
def image(self, item_id):
|
|
name = str(self.tree.item(item_id, "image")[0])
|
|
self.assertIn(name, self.root.tk.call("image", "names"))
|
|
return Image.open(io.BytesIO(self.root.tk.call(name, "data", "-format", "png")))
|
|
|
|
def assert_state(self, item_id, state):
|
|
image = self.image(item_id)
|
|
expected = resource_tree.draw_checkbox_image(18, state, theme.current_colors)
|
|
top = (image.height - expected.height) // 2
|
|
self.assertEqual(image.crop((0, top, 18, top + 18)).convert("RGBA").tobytes(), expected.tobytes())
|
|
|
|
def test_cover_and_checkbox_survive_search_clear_and_theme_changes(self):
|
|
width = self.image("books:primary:b").width
|
|
self.assertGreater(width, 24)
|
|
self.toggle("books:primary:b")
|
|
for name in ("dark", "light"):
|
|
theme.apply_theme(name)
|
|
for query in ("下册", ""):
|
|
self.filter(query)
|
|
self.assertEqual(self.image("books:primary:b").width, width)
|
|
self.assert_state("books:primary:b", "checked")
|
|
|
|
def test_filtered_category_only_toggles_visible_resources(self):
|
|
self.filter("下册")
|
|
self.toggle("books:primary")
|
|
self.assertEqual(self.lines(), {self.url("b")})
|
|
self.assert_state("books:primary", "checked")
|
|
self.filter("")
|
|
self.assert_state("books:primary", "partial")
|
|
self.assert_state("books:primary:a", "unchecked")
|
|
|
|
def test_filtered_uncheck_preserves_hidden_selection(self):
|
|
self.toggle("books:primary")
|
|
self.filter("下册")
|
|
self.toggle("books:primary")
|
|
self.assertEqual(self.lines(), {self.url("a")})
|
|
self.assert_state("books:primary", "unchecked")
|
|
self.filter("")
|
|
self.assert_state("books:primary", "partial")
|
|
|
|
def test_manual_deletion_then_parent_selection_restores_both_urls(self):
|
|
self.toggle("books:primary:b")
|
|
self.urls.delete("1.0", "end")
|
|
self.root.update()
|
|
self.assert_state("books:primary:b", "unchecked")
|
|
self.assertEqual(self.count.cget("text"), "")
|
|
self.toggle("books:primary")
|
|
self.assertEqual(self.lines(), {self.url("a"), self.url("b")})
|
|
self.assertEqual(self.count.cget("text"), "已选 2 项")
|
|
|
|
def test_paste_whitespace_and_undo_sync_without_changing_other_urls(self):
|
|
external = "https://example.com/manual"
|
|
self.urls.insert("1.0", f" {self.url('b')} \n{external}")
|
|
self.urls.edit_separator()
|
|
self.root.update()
|
|
self.assert_state("books:primary:b", "checked")
|
|
self.urls.delete("1.0", "end")
|
|
self.urls.edit_separator()
|
|
self.root.update()
|
|
self.assert_state("books:primary:b", "unchecked")
|
|
self.urls.edit_undo()
|
|
self.root.update()
|
|
self.assertIn(self.url("b"), self.lines())
|
|
self.assert_state("books:primary:b", "checked")
|
|
self.toggle("books:primary:b")
|
|
self.assertEqual(self.lines(), {external})
|
|
|
|
|
|
@pytest.mark.parametrize("case", unittest.defaultTestLoader.getTestCaseNames(ResourceTreeUITest))
|
|
def test_resource_tree_interaction(case):
|
|
# 隔离 Tk 全局状态，并避免 pytest 切换标准流文件描述符干扰 Windows Tcl 的文件读取。
|
|
result = subprocess.run(
|
|
[sys.executable, "-X", "utf8", str(Path(__file__).resolve()), case],
|
|
capture_output=True, text=True, encoding="utf-8", timeout=20,
|
|
)
|
|
if result.returncode == 77:
|
|
pytest.skip("当前环境没有图形显示服务，需在桌面环境或 Xvfb 中运行")
|
|
assert result.returncode == 0, result.stdout + result.stderr
|
|
|
|
|
|
if __name__ == "__main__":
|
|
result = unittest.TextTestRunner().run(unittest.TestSuite([ResourceTreeUITest(sys.argv[1])]))
|
|
raise SystemExit(77 if result.skipped else int(not result.wasSuccessful()))
|