121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
"""Offscreen tests for image canvas interaction controls."""
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
|
|
|
|
from PIL import Image
|
|
|
|
|
|
class TestImageCanvasControls(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tmp = Path(tempfile.mkdtemp())
|
|
self.app = self._app()
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(str(self.tmp), ignore_errors=True)
|
|
|
|
def _app(self):
|
|
from PySide6.QtWidgets import QApplication
|
|
|
|
app = QApplication.instance()
|
|
if app is None:
|
|
app = QApplication([])
|
|
return app
|
|
|
|
def _write_image(self, name, size, color):
|
|
path = self.tmp / name
|
|
Image.new("RGBA", size, color).save(str(path), "PNG")
|
|
return path
|
|
|
|
def _canvas(self):
|
|
from app.widgets.image_canvas import ImageCanvas
|
|
|
|
garment = self._write_image("garment.png", (320, 240), (240, 240, 240, 255))
|
|
print_path = self._write_image("print.png", (80, 40), (0, 120, 255, 180))
|
|
canvas = ImageCanvas()
|
|
canvas.resize(640, 480)
|
|
canvas.show()
|
|
canvas.load_garment(garment)
|
|
canvas.load_print(print_path)
|
|
self.app.processEvents()
|
|
return canvas
|
|
|
|
def _controls_visible(self, canvas):
|
|
return all(item.isVisible() for item in canvas._control_items())
|
|
|
|
def _state_tuple(self, canvas):
|
|
state = canvas.get_transform()
|
|
return (
|
|
state.x,
|
|
state.y,
|
|
state.width,
|
|
state.height,
|
|
state.rotation,
|
|
state.keep_aspect_ratio,
|
|
)
|
|
|
|
def test_escape_hides_controls_without_changing_transform(self):
|
|
from PySide6.QtCore import Qt
|
|
from PySide6.QtTest import QTest
|
|
|
|
canvas = self._canvas()
|
|
before = self._state_tuple(canvas)
|
|
|
|
self.assertEqual(canvas._view.focusPolicy(), Qt.StrongFocus)
|
|
self.assertTrue(self._controls_visible(canvas))
|
|
canvas._view.setFocus()
|
|
QTest.keyClick(canvas._view, Qt.Key_Escape)
|
|
self.app.processEvents()
|
|
|
|
self.assertFalse(canvas._controls_visible)
|
|
self.assertFalse(self._controls_visible(canvas))
|
|
self.assertTrue(canvas._print_item.isVisible())
|
|
self.assertEqual(self._state_tuple(canvas), before)
|
|
|
|
def test_clicking_print_restores_controls_and_keeps_drag_behavior(self):
|
|
from PySide6.QtCore import QPointF
|
|
|
|
canvas = self._canvas()
|
|
canvas._hide_controls()
|
|
state = canvas.get_transform()
|
|
|
|
canvas._on_press(QPointF(state.x + state.width / 2.0, state.y + state.height / 2.0))
|
|
|
|
self.assertTrue(canvas._controls_visible)
|
|
self.assertTrue(self._controls_visible(canvas))
|
|
self.assertEqual(canvas._drag_tag, "print")
|
|
|
|
def test_clicking_empty_canvas_hides_controls_without_changing_transform(self):
|
|
from PySide6.QtCore import QPointF
|
|
|
|
canvas = self._canvas()
|
|
before = self._state_tuple(canvas)
|
|
|
|
canvas._on_press(QPointF(-20.0, -20.0))
|
|
|
|
self.assertFalse(canvas._controls_visible)
|
|
self.assertFalse(self._controls_visible(canvas))
|
|
self.assertTrue(canvas._print_item.isVisible())
|
|
self.assertIsNone(canvas._drag_tag)
|
|
self.assertEqual(self._state_tuple(canvas), before)
|
|
|
|
def test_loading_print_restores_controls_visible_by_default(self):
|
|
canvas = self._canvas()
|
|
canvas._hide_controls()
|
|
self.assertFalse(canvas._controls_visible)
|
|
|
|
next_print = self._write_image("print2.png", (48, 48), (255, 80, 0, 200))
|
|
canvas.load_print(next_print)
|
|
|
|
self.assertTrue(canvas._controls_visible)
|
|
self.assertTrue(self._controls_visible(canvas))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |