86 lines
3.0 KiB
Python
86 lines
3.0 KiB
Python
"""PDD 任务表勾选模型的离屏测试。"""
|
|
|
|
import os
|
|
import unittest
|
|
|
|
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
|
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtWidgets import QApplication
|
|
|
|
from src.pdd_ui import CHECK_COLUMN, PDDTaskPage, TaskRow, TaskTableModel
|
|
|
|
|
|
class TaskTableModelCheckTests(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.app = QApplication.instance() or QApplication([])
|
|
|
|
def test_check_uses_stable_task_id_and_survives_reorder(self):
|
|
model = TaskTableModel()
|
|
model.set_tasks(
|
|
[TaskRow("TASK-1", "采集"), TaskRow("TASK-2", "采集")]
|
|
)
|
|
|
|
self.assertTrue(
|
|
model.setData(
|
|
model.index(1, CHECK_COLUMN), Qt.Checked, Qt.CheckStateRole
|
|
)
|
|
)
|
|
model.set_tasks(
|
|
[TaskRow("TASK-2", "采集"), TaskRow("TASK-1", "采集")]
|
|
)
|
|
|
|
self.assertEqual(model.checked_task_ids(), ("TASK-2",))
|
|
self.assertEqual(
|
|
model.data(model.index(0, CHECK_COLUMN), Qt.CheckStateRole),
|
|
Qt.Checked,
|
|
)
|
|
|
|
def test_placeholder_is_not_checkable_and_removed_task_is_pruned(self):
|
|
model = TaskTableModel()
|
|
self.assertEqual(model.flags(model.index(0, CHECK_COLUMN)), Qt.NoItemFlags)
|
|
model.set_tasks([TaskRow("TASK-1", "采集")])
|
|
model.setData(
|
|
model.index(0, CHECK_COLUMN), Qt.Checked, Qt.CheckStateRole
|
|
)
|
|
|
|
model.set_tasks([])
|
|
|
|
self.assertEqual(model.checked_task_ids(), ())
|
|
|
|
def test_search_clears_checks_and_refresh_is_next_to_search(self):
|
|
page = PDDTaskPage()
|
|
page.set_tasks([TaskRow("TASK-1", "采集")])
|
|
page.taskModel.setData(
|
|
page.taskModel.index(0, CHECK_COLUMN), Qt.Checked, Qt.CheckStateRole
|
|
)
|
|
|
|
page.resize(1300, 650)
|
|
self.app.processEvents()
|
|
layout = page.commandCard.layout()
|
|
search_position = layout.getItemPosition(layout.indexOf(page.searchButton))
|
|
refresh_position = layout.getItemPosition(layout.indexOf(page.refreshButton))
|
|
rerun_position = layout.getItemPosition(layout.indexOf(page.rerunButton))
|
|
resubmit_position = layout.getItemPosition(layout.indexOf(page.resubmitButton))
|
|
self.assertEqual(refresh_position[1], search_position[1] + 1)
|
|
self.assertEqual(resubmit_position[1], rerun_position[1] + 1)
|
|
self.assertEqual(page.checkedCountLabel.text(), "已选 1 条")
|
|
|
|
page.resize(1000, 650)
|
|
self.app.processEvents()
|
|
compact_search = layout.getItemPosition(layout.indexOf(page.searchButton))
|
|
compact_refresh = layout.getItemPosition(layout.indexOf(page.refreshButton))
|
|
self.assertEqual(compact_search[0], 1)
|
|
self.assertEqual(compact_refresh[1], compact_search[1] + 1)
|
|
|
|
page.searchButton.click()
|
|
|
|
self.assertEqual(page.taskModel.checked_task_ids(), ())
|
|
self.assertEqual(page.checkedCountLabel.text(), "已选 0 条")
|
|
page.deleteLater()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|