From b6de376b02e4fb29b59187efd64e7472d5fdd7fa Mon Sep 17 00:00:00 2001 From: Jaime van Kessel Date: Fri, 19 Jul 2019 12:48:36 +0200 Subject: [PATCH] Add test for getMachine function --- tests/TestMachineManager.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/TestMachineManager.py b/tests/TestMachineManager.py index b1e155aa4f..7bdede08e0 100644 --- a/tests/TestMachineManager.py +++ b/tests/TestMachineManager.py @@ -53,6 +53,16 @@ def test_setActiveMachine(machine_manager): machine_manager._application.setGlobalContainerStack.assert_called_with(mocked_global_stack) +def test_getMachine(): + registry = MagicMock() + mocked_global_stack = MagicMock() + mocked_global_stack.getId = MagicMock(return_value="test_machine") + mocked_global_stack.definition.getId = MagicMock(return_value = "test") + registry.findContainerStacks = MagicMock(return_value=[mocked_global_stack]) + with patch("cura.Settings.CuraContainerRegistry.CuraContainerRegistry.getInstance", MagicMock(return_value=registry)): + assert MachineManager.getMachine("test") == mocked_global_stack + + def test_hasUserSettings(machine_manager, application): mocked_stack = application.getGlobalContainerStack()