Add tests for ExtruderManager

This commit is contained in:
Jaime van Kessel 2019-07-19 13:55:16 +02:00
parent 1c93dffc22
commit 9ad8d91ff9
3 changed files with 51 additions and 18 deletions

View file

@ -2,8 +2,6 @@ from unittest.mock import MagicMock, patch
import pytest
from UM.Settings.ContainerRegistry import ContainerRegistry
from cura.Settings.ExtruderManager import ExtruderManager
from cura.Settings.MachineManager import MachineManager
@ -13,22 +11,6 @@ def global_stack():
stack.getId = MagicMock(return_value ="GlobalStack")
return stack
@pytest.fixture()
def container_registry() -> ContainerRegistry:
return MagicMock(name = "ContainerRegistry")
@pytest.fixture()
def extruder_manager(application, container_registry) -> ExtruderManager:
if ExtruderManager.getInstance() is not None:
# Reset the data
ExtruderManager._ExtruderManager__instance = None
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)):
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
manager = ExtruderManager()
return manager
@pytest.fixture()
def machine_manager(application, extruder_manager, container_registry, global_stack) -> MachineManager: