def test_alreadyKnownMachineAdded(container_registry):
mocked_definition_container = MagicMock(spec=DefinitionContainer)
mocked_definition_container.getId = MagicMock(return_value="machine_2")
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance",
MagicMock(return_value=container_registry)):
container_tree = ContainerTree()
assert len(container_tree.machines) == 2
# The ID is already there, so no machine should be added.
container_tree._machineAdded(mocked_definition_container)
assert len(container_tree.machines) == 2
def test_newMachineAdded(container_registry):
mocked_definition_container = MagicMock(spec=DefinitionContainer)
mocked_definition_container.getId = MagicMock(return_value="machine_3")
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance",
MagicMock(return_value=container_registry)):
container_tree = ContainerTree()
# machine_3 shouldn't be there, as on init it wasn't in the registry
assert "machine_3" not in container_tree.machines
# It should only react when a globalStack is added.
container_tree._machineAdded(mocked_definition_container)
assert "machine_3" not in container_tree.machines
container_tree._machineAdded(createMockedStack("machine_3"))
assert "machine_3" in container_tree.machines