mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-28 23:04:34 +08:00
6abfa38838
One of the weirdest errors out there, and as far as I know we're the only ones suffering from it, due to having multiple packages of Python bindings created with Sip as well as having PyQt and different threads.
65 lines
2.8 KiB
Python
65 lines
2.8 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
|
|
|
|
# Prevents error: "PyCapsule_GetPointer called with incorrect name" with conflicting SIP configurations between Arcus and PyQt: Import custom Sip bindings first!
|
|
import Savitar # Dont remove this line
|
|
import Arcus # No really. Don't. It needs to be there!
|
|
import pynest2d # Really!
|
|
from UM.Qt.QtApplication import QtApplication # QtApplication import is required, even though it isn't used.
|
|
# Even though your IDE says these files are not used, don't believe it. It's lying. They need to be there.
|
|
|
|
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
|