mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 05:39:37 +08:00
65 lines
2.4 KiB
Python
65 lines
2.4 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
# The purpose of this class is to create fixtures or methods that can be shared among all tests.
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
import pytest
|
|
|
|
from UM.Qt.QtApplication import QtApplication # QtApplication import is required, even though it isn't used.
|
|
|
|
from cura.CuraApplication import CuraApplication
|
|
from cura.Settings.ExtruderManager import ExtruderManager
|
|
from cura.Settings.MachineManager import MachineManager
|
|
from cura.UI.MachineActionManager import MachineActionManager
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry
|
|
|
|
|
|
# Create a CuraApplication object that will be shared among all tests. It needs to be initialized.
|
|
# Since we need to use it more that once, we create the application the first time and use its instance afterwards.
|
|
@pytest.fixture()
|
|
def application() -> CuraApplication:
|
|
app = MagicMock()
|
|
return app
|
|
|
|
|
|
# Returns a MachineActionManager instance.
|
|
@pytest.fixture()
|
|
def machine_action_manager(application) -> MachineActionManager:
|
|
return MachineActionManager(application)
|
|
|
|
|
|
@pytest.fixture()
|
|
def global_stack():
|
|
return MagicMock(name="Global Stack")
|
|
|
|
|
|
@pytest.fixture()
|
|
def container_registry(application, global_stack) -> ContainerRegistry:
|
|
result = MagicMock()
|
|
result.findContainerStacks = MagicMock(return_value = [global_stack])
|
|
application.getContainerRegistry = MagicMock(return_value = result)
|
|
return result
|
|
|
|
|
|
@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:
|
|
application.getExtruderManager = MagicMock(return_value = extruder_manager)
|
|
application.getGlobalContainerStack = MagicMock(return_value = global_stack)
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
|
|
manager = MachineManager(application)
|
|
|
|
return manager
|