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