Simplify defaults for container IDs in fixtures

Contributes to issue CURA-3497.
This commit is contained in:
Ghostkeeper 2017-04-07 14:02:39 +02:00
parent 46f3b5bd97
commit 7ce89f5d80
No known key found for this signature in database
GPG key ID: C5F96EE2BC0F7E75

View file

@ -55,17 +55,14 @@ def container_registry():
registry.typeMetaData = "registry_mock" registry.typeMetaData = "registry_mock"
def findInstanceContainers(registry, **kwargs): def findInstanceContainers(registry, **kwargs):
if "id" not in kwargs: container_id = kwargs.get("id", default = "test_container")
return [MockContainer("test_container", registry.typeMetaData)] return [MockContainer(container_id, registry.typeMetaData)]
else:
return [MockContainer(id, registry.typeMetaData)]
registry.findInstanceContainers = functools.partial(findInstanceContainers, registry) registry.findInstanceContainers = functools.partial(findInstanceContainers, registry)
def findContainers(registry, container_type = None, id = None): def findContainers(registry, container_type = None, id = None):
if not id: if not id:
return [MockContainer("test_container", registry.typeMetaData)] id = "test_container"
else: return [MockContainer(id, registry.typeMetaData)]
return [MockContainer(id, registry.typeMetaData)]
registry.findContainers = functools.partial(findContainers, registry) registry.findContainers = functools.partial(findContainers, registry)
UM.Settings.ContainerRegistry.ContainerRegistry._ContainerRegistry__instance = registry UM.Settings.ContainerRegistry.ContainerRegistry._ContainerRegistry__instance = registry