123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184 |
- import os
- import pytest
- import shutil
- import unittest.mock
- import urllib.parse
- from cura.Settings.CuraContainerRegistry import CuraContainerRegistry
- from cura.Settings.ExtruderStack import ExtruderStack
- from cura.Settings.GlobalStack import GlobalStack
- from UM.Resources import Resources
- import UM.Settings.InstanceContainer
- import UM.Settings.ContainerRegistry
- import UM.Settings.ContainerStack
- from UM.Settings.DefinitionContainer import DefinitionContainer
- @pytest.fixture()
- def container_registry():
- registry = CuraContainerRegistry()
- UM.Settings.InstanceContainer.setContainerRegistry(registry)
- UM.Settings.ContainerStack.setContainerRegistry(registry)
- return registry
- @pytest.fixture()
- def definition_container():
- return DefinitionContainer(container_id = "Test Definition")
- def teardown():
-
- temporary_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "stacks", "temporary.stack.cfg")
- if os.path.isfile(temporary_file):
- os.remove(temporary_file)
- def test_addContainerExtruderStack(container_registry, definition_container):
- container_registry.addContainer(definition_container)
- container_stack = UM.Settings.ContainerStack.ContainerStack(stack_id = "Test Container Stack")
- container_stack.addMetaDataEntry("type", "extruder_train")
- container_stack.insertContainer(0, definition_container)
- mock_super_add_container = unittest.mock.MagicMock()
- with unittest.mock.patch("UM.Settings.ContainerRegistry.ContainerRegistry.addContainer", mock_super_add_container):
- container_registry.addContainer(container_stack)
- assert len(mock_super_add_container.call_args_list) == 1
- assert len(mock_super_add_container.call_args_list[0][0]) == 1
- assert type(mock_super_add_container.call_args_list[0][0][0]) == ExtruderStack
- def test_addContainerGlobalStack(container_registry, definition_container):
- container_registry.addContainer(definition_container)
- container_stack = UM.Settings.ContainerStack.ContainerStack(stack_id = "Test Container Stack")
- container_stack.addMetaDataEntry("type", "machine")
- container_stack.insertContainer(0, definition_container)
- mock_super_add_container = unittest.mock.MagicMock()
- with unittest.mock.patch("UM.Settings.ContainerRegistry.ContainerRegistry.addContainer", mock_super_add_container):
- container_registry.addContainer(container_stack)
- assert len(mock_super_add_container.call_args_list) == 1
- assert len(mock_super_add_container.call_args_list[0][0]) == 1
- assert type(mock_super_add_container.call_args_list[0][0][0]) == GlobalStack
- def test_addContainerGoodSettingVersion(container_registry, definition_container):
- from cura.CuraApplication import CuraApplication
- definition_container.getMetaData()["setting_version"] = CuraApplication.SettingVersion
- container_registry.addContainer(definition_container)
- instance = UM.Settings.InstanceContainer.InstanceContainer(container_id = "Test Instance")
- instance.addMetaDataEntry("setting_version", CuraApplication.SettingVersion)
- instance.setDefinition(definition_container)
- mock_super_add_container = unittest.mock.MagicMock()
- with unittest.mock.patch("UM.Settings.ContainerRegistry.ContainerRegistry.addContainer", mock_super_add_container):
- container_registry.addContainer(instance)
- mock_super_add_container.assert_called_once_with(instance)
- def test_addContainerNoSettingVersion(container_registry, definition_container):
- from cura.CuraApplication import CuraApplication
- definition_container.getMetaData()["setting_version"] = CuraApplication.SettingVersion
- container_registry.addContainer(definition_container)
- instance = UM.Settings.InstanceContainer.InstanceContainer(container_id = "Test Instance")
-
- instance.setDefinition(definition_container)
- mock_super_add_container = unittest.mock.MagicMock()
- with unittest.mock.patch("UM.Settings.ContainerRegistry.ContainerRegistry.addContainer", mock_super_add_container):
- container_registry.addContainer(instance)
- mock_super_add_container.assert_not_called()
- def test_addContainerBadSettingVersion(container_registry, definition_container):
- from cura.CuraApplication import CuraApplication
- definition_container.getMetaData()["setting_version"] = CuraApplication.SettingVersion
- container_registry.addContainer(definition_container)
- instance = UM.Settings.InstanceContainer.InstanceContainer(container_id = "Test Instance")
- instance.addMetaDataEntry("setting_version", 9001)
- instance.setDefinition(definition_container)
- mock_super_add_container = unittest.mock.MagicMock()
- with unittest.mock.patch("UM.Settings.ContainerRegistry.ContainerRegistry.addContainer", mock_super_add_container):
- container_registry.addContainer(instance)
- mock_super_add_container.assert_not_called()
- @pytest.mark.parametrize("filename, output_class", [
- ("ExtruderLegacy.stack.cfg", ExtruderStack),
- ("MachineLegacy.stack.cfg", GlobalStack),
- ("Left.extruder.cfg", ExtruderStack),
- ("Global.global.cfg", GlobalStack),
- ("Global.stack.cfg", GlobalStack)
- ])
- def test_loadTypes(filename, output_class, container_registry):
-
- Resources.getAllResourcesOfType = unittest.mock.MagicMock(return_value = [os.path.join(os.path.dirname(os.path.abspath(__file__)), "stacks", filename)])
- def findContainers(container_type = 0, id = None):
- if id == "some_instance":
- return [UM.Settings.ContainerRegistry._EmptyInstanceContainer(id)]
- elif id == "some_definition":
- return [DefinitionContainer(container_id = id)]
- else:
- return []
- container_registry.findContainers = findContainers
- with unittest.mock.patch("cura.Settings.GlobalStack.GlobalStack.findContainer"):
- with unittest.mock.patch("os.remove"):
- container_registry.load()
-
- stack_id = filename.split(".")[0]
- for container in container_registry._containers:
- if container.getId() == stack_id:
- assert type(container) == output_class
- break
- else:
- assert False
- def test_loadLegacyFileRenamed(container_registry):
-
- stacks_folder = os.path.join(os.path.dirname(os.path.abspath(__file__)), "stacks")
- temp_file = os.path.join(stacks_folder, "temporary.stack.cfg")
- temp_file_source = os.path.join(stacks_folder, "MachineLegacy.stack.cfg")
- shutil.copyfile(temp_file_source, temp_file)
-
- UM.Settings.ContainerStack.setContainerRegistry(container_registry)
- Resources.getAllResourcesOfType = unittest.mock.MagicMock(return_value = [temp_file])
- def findContainers(container_type = 0, id = None):
- if id == "MachineLegacy":
- return None
- container = UM.Settings.ContainerRegistry._EmptyInstanceContainer(id)
- container.getNextStack = unittest.mock.MagicMock()
- return [container]
- old_find_containers = container_registry.findContainers
- container_registry.findContainers = findContainers
- with unittest.mock.patch("cura.Settings.GlobalStack.GlobalStack.findContainer"):
- container_registry.load()
- container_registry.findContainers = old_find_containers
- container_registry.saveAll()
- print("all containers in registry", container_registry._containers)
- assert not os.path.isfile(temp_file)
- mime_type = container_registry.getMimeTypeForContainer(GlobalStack)
- file_name = urllib.parse.quote_plus("MachineLegacy") + "." + mime_type.preferredSuffix
- path = Resources.getStoragePath(Resources.ContainerStacks, file_name)
- assert os.path.isfile(path)
|