3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-07-20 13:17:51 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fix tests

Browse source
This commit is contained in:
Jaime van Kessel 2019-08-29 17:07:36 +02:00
parent 784ab6e903
commit d9e94f5019
5 changed files with 31 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/Machines/TestContainerTree.py
View file

@ -2,10 +2,11 @@ from unittest.mock import patch, MagicMock
import pytest
from UM.Settings.DefinitionContainer import DefinitionContainer
from cura.Machines.ContainerTree import ContainerTree
from cura.Settings.GlobalStack import GlobalStack
def createMockedStack(definition_id: str):
result = MagicMock()
result = MagicMock(spec = GlobalStack)
result.definition.getId = MagicMock(return_value = definition_id)
return result
@ -34,8 +35,11 @@ def test_newMachineAdded(container_registry):
# machine_3 shouldn't be there, as on init it wasn't in the registry
assert "machine_3" not in container_tree.machines
# But when it does get added (by manually triggering the _machineAdded), it should be there.
# It should only react when a globalStack is added.
container_tree._machineAdded(mocked_definition_container)
assert "machine_3" not in container_tree.machines
container_tree._machineAdded(createMockedStack("machine_3"))
assert "machine_3" in container_tree.machines