135 lines
5 KiB
Python
135 lines
5 KiB
Python
import os
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
|
|
|
from PIL import Image
|
|
from PyQt5.QtCore import QItemSelectionModel, QSettings, Qt
|
|
from PyQt5.QtTest import QTest
|
|
from PyQt5.QtWidgets import QApplication, QAbstractItemView
|
|
|
|
from gui.rpgmaker_image_manager import RPGMakerImageManager
|
|
|
|
|
|
class RPGMakerImageManagerSelectionTests(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.app = QApplication.instance() or QApplication([])
|
|
|
|
def setUp(self):
|
|
self.temp = tempfile.TemporaryDirectory()
|
|
root = Path(self.temp.name)
|
|
self.game_root = root / "Game"
|
|
image_dir = self.game_root / "img" / "pictures"
|
|
data_dir = self.game_root / "data"
|
|
image_dir.mkdir(parents=True)
|
|
data_dir.mkdir(parents=True)
|
|
data_dir.joinpath("System.json").write_text(
|
|
'{"encryptionKey":"00112233445566778899aabbccddeeff"}',
|
|
encoding="utf-8",
|
|
)
|
|
for index in range(4):
|
|
Image.new("RGBA", (24, 24), (index * 40, 10, 20, 255)).save(
|
|
image_dir / f"image{index}.png"
|
|
)
|
|
settings = QSettings(str(root / "settings.ini"), QSettings.IniFormat)
|
|
self.manager = RPGMakerImageManager(self.game_root, settings=settings)
|
|
self.manager.resize(1100, 760)
|
|
self.manager.show()
|
|
self.manager._scan_worker.wait(5000)
|
|
self.app.processEvents()
|
|
|
|
def tearDown(self):
|
|
for worker in list(self.manager._thumbnail_workers):
|
|
worker.wait(5000)
|
|
self.manager.close()
|
|
self.app.processEvents()
|
|
self.temp.cleanup()
|
|
|
|
def _click(self, index: int, modifiers=Qt.NoModifier):
|
|
item = self.manager.image_list.item(index)
|
|
rect = self.manager.image_list.visualItemRect(item)
|
|
QTest.mouseClick(
|
|
self.manager.image_list.viewport(),
|
|
Qt.LeftButton,
|
|
modifiers,
|
|
rect.center(),
|
|
)
|
|
self.app.processEvents()
|
|
|
|
def test_ctrl_shift_and_ctrl_a_use_standard_extended_selection(self):
|
|
image_list = self.manager.image_list
|
|
self.assertEqual(
|
|
image_list.selectionMode(), QAbstractItemView.ExtendedSelection
|
|
)
|
|
|
|
self._click(0)
|
|
self._click(2, Qt.ControlModifier)
|
|
self.assertEqual(len(image_list.selectedItems()), 2)
|
|
self.assertEqual(len(self.manager.selected_ids), 2)
|
|
|
|
self._click(3, Qt.ShiftModifier)
|
|
selected_names = {item.text() for item in image_list.selectedItems()}
|
|
self.assertIn("image2.png", selected_names)
|
|
self.assertIn("image3.png", selected_names)
|
|
|
|
image_list.clearSelection()
|
|
image_list.setFocus()
|
|
QTest.keyClick(image_list, Qt.Key_A, Qt.ControlModifier)
|
|
self.app.processEvents()
|
|
self.assertEqual(len(image_list.selectedItems()), 4)
|
|
self.assertEqual(len(self.manager.selected_ids), 4)
|
|
|
|
def test_selected_tiles_are_restored_after_filtering(self):
|
|
selection = self.manager.image_list.selectionModel()
|
|
selection.select(
|
|
self.manager.image_list.model().index(1, 0),
|
|
QItemSelectionModel.Select,
|
|
)
|
|
selection.select(
|
|
self.manager.image_list.model().index(3, 0),
|
|
QItemSelectionModel.Select,
|
|
)
|
|
self.app.processEvents()
|
|
selected_before = set(self.manager.selected_ids)
|
|
self.assertEqual(len(selected_before), 2)
|
|
|
|
self.manager.search_edit.setText("image0")
|
|
self.app.processEvents()
|
|
self.manager.search_edit.clear()
|
|
self.app.processEvents()
|
|
|
|
self.assertEqual(self.manager.selected_ids, selected_before)
|
|
self.assertEqual(len(self.manager.image_list.selectedItems()), 2)
|
|
|
|
def test_thumbnail_batch_keeps_one_stable_tile_per_asset(self):
|
|
for worker in list(self.manager._thumbnail_workers):
|
|
worker.wait(5000)
|
|
self.app.processEvents()
|
|
|
|
image_list = self.manager.image_list
|
|
self.assertTrue(image_list.uniformItemSizes())
|
|
self.assertNotIn("border-bottom", image_list.styleSheet())
|
|
self.assertEqual(image_list.count(), 4)
|
|
self.assertTrue(
|
|
all(not image_list.item(index).icon().isNull() for index in range(4))
|
|
)
|
|
rects = [image_list.visualItemRect(image_list.item(index)) for index in range(4)]
|
|
self.assertEqual(len({(rect.x(), rect.y()) for rect in rects}), 4)
|
|
|
|
|
|
class RPGMakerImageManagerNavigationTests(unittest.TestCase):
|
|
def test_image_manager_is_a_dedicated_sidebar_page(self):
|
|
root = Path(__file__).resolve().parents[1]
|
|
main_source = (root / "gui" / "main.py").read_text(encoding="utf-8")
|
|
workflow_source = (root / "gui" / "workflow_tab.py").read_text(encoding="utf-8")
|
|
self.assertIn("PAGE_IMAGES = 2", main_source)
|
|
self.assertIn("self.image_manager_tab = RPGMakerImageManager", main_source)
|
|
self.assertIn('create_nav_button("🖼", "Images")', main_source)
|
|
self.assertNotIn("Open Image Manager", workflow_source)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|