From 99400238161af8e5852bea404bf942b42632d918 Mon Sep 17 00:00:00 2001 From: Jaime van Kessel Date: Fri, 19 Jul 2019 12:56:54 +0200 Subject: [PATCH] Add test for adding a new machine --- tests/TestMachineManager.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/TestMachineManager.py b/tests/TestMachineManager.py index 7bdede08e0..ab89fdcc29 100644 --- a/tests/TestMachineManager.py +++ b/tests/TestMachineManager.py @@ -63,6 +63,19 @@ def test_getMachine(): assert MachineManager.getMachine("test") == mocked_global_stack +def test_addMachine(machine_manager): + registry = MagicMock() + + mocked_stack = MagicMock() + mocked_stack.getId = MagicMock(return_value="newlyCreatedStack") + mocked_create_machine = MagicMock(name="createMachine", return_value = mocked_stack) + machine_manager.setActiveMachine = MagicMock() + with patch("cura.Settings.CuraStackBuilder.CuraStackBuilder.createMachine", mocked_create_machine): + with patch("cura.Settings.CuraContainerRegistry.CuraContainerRegistry.getInstance", MagicMock(return_value=registry)): + machine_manager.addMachine("derp") + machine_manager.setActiveMachine.assert_called_with("newlyCreatedStack") + + def test_hasUserSettings(machine_manager, application): mocked_stack = application.getGlobalContainerStack()