GitHub-Proxy/Cura
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura synced 2025-08-12 18:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update extrudermanager in tests to mock Application.

Browse Source 
CURA-9793
This commit is contained in:
Joey de l'Arago 2022-12-12 14:06:30 +01:00
parent c6cc9ee6ec
commit f52638ca75
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/conftest.py
View File

@ -6,6 +6,7 @@
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import pytest import pytest
from UM.Application import Application
from UM.Qt.QtApplication import QtApplication # QtApplication import is required, even though it isn't used. from UM.Qt.QtApplication import QtApplication # QtApplication import is required, even though it isn't used.
from cura.CuraApplication import CuraApplication from cura.CuraApplication import CuraApplication
@ -22,6 +23,12 @@ def application() -> CuraApplication:
app = MagicMock() app = MagicMock()
return app return app
@pytest.fixture()
def um_application() -> Application:
app = MagicMock()
app.getInstance = MagicMock(return_value=app)
return app
# Returns a MachineActionManager instance. # Returns a MachineActionManager instance.
@pytest.fixture() @pytest.fixture()
@ -43,13 +50,14 @@ def container_registry(application, global_stack) -> ContainerRegistry:
@pytest.fixture() @pytest.fixture()
def extruder_manager(application, container_registry) -> ExtruderManager: def extruder_manager(application, um_application, container_registry) -> ExtruderManager:
if ExtruderManager.getInstance() is not None: if ExtruderManager.getInstance() is not None:
# Reset the data # Reset the data
ExtruderManager._ExtruderManager__instance = None ExtruderManager._ExtruderManager__instance = None
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)): with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)):
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)): with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
with patch("UM.Application.Application.getInstance", um_application):
manager = ExtruderManager() manager = ExtruderManager()
return manager return manager