File: TestGroupedOperation.py

package info (click to toggle)
uranium 5.0.0-7
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 5,304 kB
  • sloc: python: 31,765; sh: 132; makefile: 12
file content (53 lines) | stat: -rw-r--r-- 1,864 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from unittest.mock import MagicMock

from UM.Math.Vector import Vector
from UM.Operations.GroupedOperation import GroupedOperation
import pytest

from UM.Operations.Operation import Operation
from UM.Operations.TranslateOperation import TranslateOperation
from UM.Scene.SceneNode import SceneNode


def test_addOperationFinalised():
    operation_1 = GroupedOperation()
    operation_2 = GroupedOperation()

    operation_1.redo()  # The operation is now finalized, so it shouldn't be possible to add operations to it now.
    with pytest.raises(Exception):
        operation_1.addOperation(operation_2)


def test_addAndMergeOperations():
    group_operation_1 = GroupedOperation()
    group_operation_2 = GroupedOperation()
    group_operation_3 = GroupedOperation()

    scene_node = SceneNode()

    operation_1 = TranslateOperation(scene_node, Vector(10, 10))
    operation_2 = TranslateOperation(scene_node, Vector(10, 20))
    operation_3 = Operation()

    operation_1.mergeWith = MagicMock()

    group_operation_1.addOperation(operation_1)
    assert group_operation_1.getNumChildrenOperations() == 1

    # Length of the grouped operations must be the same.
    assert not group_operation_1.mergeWith(group_operation_2)

    group_operation_3.addOperation(operation_3)

    # The base operation always says it can't be merged (so one child says it can't be merged). This should result
    # in the parent (grouped) operation also failing.
    assert not group_operation_3.mergeWith(group_operation_1)

    group_operation_2.addOperation(operation_2)
    merged_operation = group_operation_1.mergeWith(group_operation_2)

    # The merge of the nested operation should have been called once.
    assert operation_1.mergeWith.call_count == 1

    # Number of operations should still be the same.
    assert merged_operation.getNumChildrenOperations() == 1