From ef4fae5b8008dbc9c26976b66a9edf5ec0b473ed Mon Sep 17 00:00:00 2001 From: Jaime van Kessel Date: Fri, 21 Aug 2020 15:39:03 +0200 Subject: [PATCH] Expand tests for hasUserSettings --- tests/TestMachineManager.py | 35 ++++++++++++++++++++++++++++++++++- 1 file changed, 34 insertions(+), 1 deletion(-) diff --git a/tests/TestMachineManager.py b/tests/TestMachineManager.py index bf23a997cc..35afa7c481 100644 --- a/tests/TestMachineManager.py +++ b/tests/TestMachineManager.py @@ -1,4 +1,4 @@ -from unittest.mock import MagicMock, patch +from unittest.mock import MagicMock, patch, PropertyMock import pytest from cura.Settings.MachineManager import MachineManager @@ -62,6 +62,39 @@ def test_hasUserSettings(machine_manager, application): assert machine_manager.hasUserSettings +def test_hasUserSettingsExtruder(machine_manager, application): + mocked_stack = application.getGlobalContainerStack() + extruder = createMockedExtruder("extruder") + + mocked_instance_container_global = MagicMock(name="UserSettingContainerGlobal") + mocked_instance_container_global.getNumInstances = MagicMock(return_value=0) + mocked_stack.getTop = MagicMock(return_value=mocked_instance_container_global) + mocked_stack.extruderList = [extruder] + + mocked_instance_container = MagicMock(name="UserSettingContainer") + mocked_instance_container.getNumInstances = MagicMock(return_value=200) + extruder.getTop = MagicMock(return_value = mocked_instance_container) + + assert machine_manager.hasUserSettings + assert machine_manager.numUserSettings == 200 + + +def test_hasUserSettingsEmptyUserChanges(machine_manager, application): + mocked_stack = application.getGlobalContainerStack() + extruder = createMockedExtruder("extruder") + + mocked_instance_container_global = MagicMock(name="UserSettingContainerGlobal") + mocked_instance_container_global.getNumInstances = MagicMock(return_value=0) + mocked_stack.getTop = MagicMock(return_value=mocked_instance_container_global) + mocked_stack.extruderList = [extruder] + + mocked_instance_container = MagicMock(name="UserSettingContainer") + mocked_instance_container.getNumInstances = MagicMock(return_value=0) + extruder.getTop = MagicMock(return_value = mocked_instance_container) + + assert not machine_manager.hasUserSettings + + def test_totalNumberOfSettings(machine_manager): registry = MagicMock() mocked_definition = MagicMock()