loop-uh
d5a4715771
Some checks failed
Windows project source guards / test (push) Has been cancelled
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
"""Theme repaint tests for manual-paint widgets (AC6b).
|
|
|
|
Keep the ``test_app_*`` prefix: widget test modules must sort before
|
|
``tests/test_engine_process_stop.py`` which creates a bare QCoreApplication
|
|
at import time (see tests/test_app_nodes_page_view.py).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import unittest
|
|
|
|
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
|
|
|
from PyQt6.QtCore import Qt
|
|
from PyQt6.QtWidgets import QApplication
|
|
|
|
_existing = QApplication.instance()
|
|
if _existing is not None and not isinstance(_existing, QApplication):
|
|
raise RuntimeError(
|
|
"A bare QCoreApplication was created before test_app_theme_repaint "
|
|
"was imported; widget tests need a QApplication."
|
|
)
|
|
app = _existing or QApplication([])
|
|
|
|
from qfluentwidgets import Theme, qconfig, setTheme
|
|
|
|
from xray_fluent.profiles.models import Node
|
|
from xray_fluent.ui.nodes_table_model import COL_PING, NodesTableModel
|
|
|
|
|
|
class NodesModelThemeRepaintTest(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self._saved_theme = qconfig.themeMode.value
|
|
self.model = NodesTableModel()
|
|
self.model.set_nodes(
|
|
[
|
|
Node(name="a", server="1.1.1.1", port=443, scheme="vless", is_alive=True),
|
|
Node(name="b", server="2.2.2.2", port=443, scheme="vless", is_alive=False),
|
|
]
|
|
)
|
|
|
|
def tearDown(self) -> None:
|
|
setTheme(self._saved_theme)
|
|
QApplication.processEvents()
|
|
|
|
def test_theme_change_emits_foreground_role_for_all_rows(self) -> None:
|
|
emissions: list[tuple[int, int, list[int]]] = []
|
|
|
|
def on_data_changed(top_left, bottom_right, roles) -> None:
|
|
emissions.append((top_left.row(), bottom_right.row(), list(roles)))
|
|
|
|
self.model.dataChanged.connect(on_data_changed)
|
|
setTheme(Theme.DARK)
|
|
QApplication.processEvents()
|
|
emissions.clear()
|
|
|
|
setTheme(Theme.LIGHT)
|
|
QApplication.processEvents()
|
|
|
|
foreground_spans = [
|
|
(first, last)
|
|
for first, last, roles in emissions
|
|
if Qt.ItemDataRole.ForegroundRole in roles
|
|
]
|
|
self.assertTrue(foreground_spans, "no dataChanged with ForegroundRole on theme change")
|
|
first, last = foreground_spans[0]
|
|
self.assertEqual((first, last), (0, self.model.rowCount() - 1))
|
|
|
|
def test_status_brushes_follow_theme(self) -> None:
|
|
index = self.model.index(1, COL_PING) # dead node -> error brush
|
|
setTheme(Theme.DARK)
|
|
dark_brush = self.model.data(index, Qt.ItemDataRole.ForegroundRole)
|
|
setTheme(Theme.LIGHT)
|
|
light_brush = self.model.data(index, Qt.ItemDataRole.ForegroundRole)
|
|
self.assertIsNotNone(dark_brush)
|
|
self.assertIsNotNone(light_brush)
|
|
self.assertNotEqual(dark_brush.color().getRgb(), light_brush.color().getRgb())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|