Add test for registering global stacks

This test tests adding container stacks and seeing if they convert well.

Contributes to issue CURA-3427.
This commit is contained in:
Ghostkeeper 2017-05-12 16:11:51 +02:00
parent e50b0884f0
commit ab7b9913c7
No known key found for this signature in database
GPG key ID: C5F96EE2BC0F7E75

View file

@ -30,7 +30,7 @@ def teardown():
if os.path.isfile(temporary_file):
os.remove(temporary_file)
## Tests whether the addContainer function properly converts ContainerStacks.
## Tests whether addContainer properly converts to ExtruderStack.
def test_addContainerExtruderStack(container_registry):
definition = DefinitionContainer(container_id = "Test Definition") #Need some definition first to be able to register stacks.
container_registry.addContainer(definition)
@ -39,7 +39,7 @@ def test_addContainerExtruderStack(container_registry):
container_stack.addMetaDataEntry("type", "extruder_train") #This is now an extruder train.
container_stack.insertContainer(0, definition) #Add a definition to it so it doesn't complain.
mock_super_add_container = unittest.mock.MagicMock()
mock_super_add_container = unittest.mock.MagicMock() #Takes the role of the Uranium-ContainerRegistry where the resulting containers get registered.
with unittest.mock.patch("UM.Settings.ContainerRegistry.ContainerRegistry.addContainer", mock_super_add_container):
container_registry.addContainer(container_stack)
@ -47,6 +47,23 @@ def test_addContainerExtruderStack(container_registry):
assert len(mock_super_add_container.call_args_list[0][0]) == 1 #Called with one parameter.
assert type(mock_super_add_container.call_args_list[0][0][0]) == ExtruderStack
## Tests whether addContainer properly converts to GlobalStack.
def test_addContainerGlobalStack(container_registry):
definition = DefinitionContainer(container_id = "Test Definition") #Need some definition first to be able to register stacks.
container_registry.addContainer(definition)
container_stack = UM.Settings.ContainerStack.ContainerStack(stack_id = "Test Container Stack") #A container we're going to convert.
container_stack.addMetaDataEntry("type", "machine") #This is now a global stack.
container_stack.insertContainer(0, definition) #Must have a definition.
mock_super_add_container = unittest.mock.MagicMock() #Takes the role of the Uranium-ContainerRegistry where the resulting containers get registered.
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 #Called only once.
assert len(mock_super_add_container.call_args_list[0][0]) == 1 #Called with one parameter.
assert type(mock_super_add_container.call_args_list[0][0][0]) == GlobalStack
## Tests whether loading gives objects of the correct type.
@pytest.mark.parametrize("filename, output_class", [
("ExtruderLegacy.stack.cfg", ExtruderStack),