mirror of
https://github.com/rbreu/beeref.git
synced 2026-03-11 08:54:28 +00:00
146 lines
4.7 KiB
Python
146 lines
4.7 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from PyQt6 import QtGui
|
|
|
|
from beeref import commands
|
|
from beeref.items import BeePixmapItem
|
|
from beeref.scene import BeeGraphicsScene
|
|
from .base import BeeTestCase
|
|
|
|
|
|
class InsertItemsTestCase(BeeTestCase):
|
|
|
|
@patch('beeref.scene.BeeGraphicsScene.views')
|
|
def test_redo_undo(self, views_mock):
|
|
scene = BeeGraphicsScene(None)
|
|
view = MagicMock(get_scale=MagicMock(return_value=1))
|
|
views_mock.return_value = [view]
|
|
scene.update_selection = MagicMock()
|
|
scene.max_z = 5
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
scene.addItem(item1)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
command = commands.InsertItems(scene, [item2])
|
|
command.redo()
|
|
assert list(scene.items_for_save()) == [item1, item2]
|
|
assert item1.isSelected() is False
|
|
assert item2.isSelected() is True
|
|
item2.zValue() > 5
|
|
command.undo()
|
|
assert list(scene.items_for_save()) == [item1]
|
|
assert item1.isSelected() is False
|
|
|
|
|
|
class DeleteItemsTestCase(BeeTestCase):
|
|
|
|
def test_redo_undo(self):
|
|
scene = BeeGraphicsScene(None)
|
|
scene.update_selection = MagicMock()
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
scene.addItem(item1)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
scene.addItem(item2)
|
|
item2.setSelected(True)
|
|
command = commands.DeleteItems(scene, [item2])
|
|
command.redo()
|
|
assert list(scene.items_for_save()) == [item1]
|
|
command.undo()
|
|
assert list(scene.items_for_save()) == [item1, item2]
|
|
assert item1.isSelected() is False
|
|
assert item2.isSelected() is True
|
|
|
|
|
|
class MoveItemsByTestCase(BeeTestCase):
|
|
|
|
def test_redo_undo(self):
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
item1.setPos(0, 0)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
item2.setPos(30, 40)
|
|
command = commands.MoveItemsBy([item1, item2], 50, 100)
|
|
command.redo()
|
|
assert item1.pos().x() == 50
|
|
assert item1.pos().y() == 100
|
|
assert item2.pos().x() == 80
|
|
assert item2.pos().y() == 140
|
|
|
|
command.undo()
|
|
assert item1.pos().x() == 0
|
|
assert item1.pos().y() == 0
|
|
assert item2.pos().x() == 30
|
|
assert item2.pos().y() == 40
|
|
|
|
def test_ignore_first_redo(self):
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
item1.setPos(0, 0)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
item2.setPos(30, 40)
|
|
command = commands.MoveItemsBy([item1, item2], 50, 100, True)
|
|
command.redo()
|
|
assert item1.pos().x() == 0
|
|
assert item1.pos().y() == 0
|
|
assert item2.pos().x() == 30
|
|
assert item2.pos().y() == 40
|
|
command.redo()
|
|
assert item1.pos().x() == 50
|
|
assert item1.pos().y() == 100
|
|
assert item2.pos().x() == 80
|
|
assert item2.pos().y() == 140
|
|
|
|
|
|
class ScaleItemsByTestCase(BeeTestCase):
|
|
|
|
def test_redo_undo(self):
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
item1.setScale(1)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
item2.setScale(3)
|
|
command = commands.ScaleItemsByDelta([item1, item2], 2, (100, 100))
|
|
command.redo()
|
|
assert item1.scale() == 3
|
|
assert item1.pos().x() == -200
|
|
assert item1.pos().y() == -200
|
|
assert item2.scale() == 5
|
|
assert item2.pos().x() == -200
|
|
assert item2.pos().y() == -200
|
|
command.undo()
|
|
assert item1.scale() == 1
|
|
assert item1.pos().x() == 0
|
|
assert item1.pos().y() == 0
|
|
assert item2.scale() == 3
|
|
assert item2.pos().x() == 0
|
|
assert item2.pos().y() == 0
|
|
|
|
def test_ignore_first_redo(self):
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
item1.setScale(1)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
item2.setScale(3)
|
|
command = commands.ScaleItemsByDelta([item1, item2], 2, (100, 100),
|
|
ignore_first_redo=True)
|
|
command.redo()
|
|
assert item1.scale() == 1
|
|
assert item2.scale() == 3
|
|
assert item1.pos().x() == 0
|
|
assert item1.pos().y() == 0
|
|
command.redo()
|
|
assert item1.scale() == 3
|
|
assert item2.scale() == 5
|
|
assert item1.pos().x() == -200
|
|
assert item1.pos().y() == -200
|
|
|
|
|
|
class NormalizeItemsTestCase(BeeTestCase):
|
|
|
|
def test_redo_undo(self):
|
|
item1 = BeePixmapItem(QtGui.QImage())
|
|
item1.setScale(1)
|
|
item2 = BeePixmapItem(QtGui.QImage())
|
|
item2.setScale(3)
|
|
command = commands.NormalizeItems([item1, item2], [2, 0.5])
|
|
command.redo()
|
|
assert item1.scale() == 2
|
|
assert item2.scale() == 0.5
|
|
command.undo()
|
|
assert item1.scale() == 1
|
|
assert item2.scale() == 3
|