Browse Source

Add test for getMachine function

Jaime van Kessel 5 years ago
parent
commit
b6de376b02
1 changed files with 10 additions and 0 deletions
  1. 10 0
      tests/TestMachineManager.py

+ 10 - 0
tests/TestMachineManager.py

@@ -53,6 +53,16 @@ def test_setActiveMachine(machine_manager):
             machine_manager._application.setGlobalContainerStack.assert_called_with(mocked_global_stack)
             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):
 def test_hasUserSettings(machine_manager, application):
     mocked_stack = application.getGlobalContainerStack()
     mocked_stack = application.getGlobalContainerStack()