Let test_setVariantByIdExists use new container registry functionality

It sets the container it expects beforehand and tests for its ID.

Contributes to issue CURA-3497.
This commit is contained in:
Ghostkeeper 2017-04-12 09:48:52 +02:00
parent 4db7c6ae66
commit 0e82c84790
No known key found for this signature in database
GPG key ID: C5F96EE2BC0F7E75

View file

@ -402,15 +402,10 @@ def test_setQualityChangesByIdDoesntExist(extruder_stack):
extruder_stack.setQualityChangesById("some_quality_changes") #Container registry is empty now.
## Tests setting variants by specifying an ID of a variant that exists.
@pytest.mark.skip
def test_setVariantByIdExists(extruder_stack, container_registry):
original_container_registry = UM.Settings.ContainerStack._containerRegistry
UM.Settings.ContainerStack._containerRegistry = container_registry #Always has all the profiles you ask of.
extruder_stack.setVariantById("some_variant") #The container registry always has a container with the ID.
#Restore.
UM.Settings.ContainerStack._containerRegistry = original_container_registry
container_registry.return_value = getInstanceContainer(container_type = "variant")
extruder_stack.setVariantById("InstanceContainer")
assert extruder_stack.variant.getId() == "InstanceContainer"
## Tests setting variants by specifying an ID of a variant that doesn't exist.
@pytest.mark.skip