mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 21:59:37 +08:00
93c84d5bc0
This is necessary because our code now converts this to an integer. Mocks can't be converted to an integer, so the sub-magic-mock here can't be used. We need to provide something that can be cast to an integer.
58 lines
2.8 KiB
Python
58 lines
2.8 KiB
Python
from unittest.mock import patch, MagicMock
|
|
|
|
import pytest
|
|
|
|
from cura.Settings.SettingOverrideDecorator import SettingOverrideDecorator
|
|
|
|
|
|
extruder_manager = MagicMock(name= "ExtruderManager")
|
|
application = MagicMock(name="application")
|
|
container_registry = MagicMock(name="container_registry")
|
|
|
|
@pytest.fixture()
|
|
def setting_override_decorator():
|
|
# Ensure that all the call counts and the like are reset.
|
|
container_registry.reset_mock()
|
|
application.reset_mock()
|
|
extruder_manager.reset_mock()
|
|
|
|
# Actually create the decorator.
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
|
|
with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)):
|
|
with patch("cura.Settings.ExtruderManager.ExtruderManager.getInstance", MagicMock(return_value=extruder_manager)):
|
|
return SettingOverrideDecorator()
|
|
|
|
|
|
def test_onSettingValueChanged(setting_override_decorator):
|
|
def mock_getRawProperty(key, property_name, *args, **kwargs):
|
|
if property_name == "limit_to_extruder":
|
|
return "-1"
|
|
return MagicMock(name="rawProperty")
|
|
container_registry.findContainerStacks().__getitem__().getRawProperty = mock_getRawProperty
|
|
# On creation the needs slicing should be called once (as it being added should trigger a reslice)
|
|
assert application.getBackend().needsSlicing.call_count == 1
|
|
with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)):
|
|
setting_override_decorator._onSettingChanged("blarg", "value")
|
|
|
|
# Once we set a setting value, it should trigger again.
|
|
assert application.getBackend().needsSlicing.call_count == 2
|
|
|
|
|
|
def test_onSettingEnableChanged(setting_override_decorator):
|
|
# On creation the needs slicing should be called once (as it being added should trigger a reslice)
|
|
assert application.getBackend().needsSlicing.call_count == 1
|
|
with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)):
|
|
setting_override_decorator._onSettingChanged("blarg", "enabled")
|
|
|
|
# Once we set a property that is not a value, no re-slice should happen.
|
|
assert application.getBackend().needsSlicing.call_count == 1
|
|
|
|
|
|
def test_setActiveExtruder(setting_override_decorator):
|
|
setting_override_decorator.activeExtruderChanged.emit = MagicMock()
|
|
with patch("cura.Settings.ExtruderManager.ExtruderManager.getInstance", MagicMock(return_value=extruder_manager)):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
|
|
setting_override_decorator.setActiveExtruder("ZOMG")
|
|
setting_override_decorator.activeExtruderChanged.emit.assert_called_once_with()
|
|
assert setting_override_decorator.getActiveExtruder() == "ZOMG"
|