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,16 +55,13 @@ def container_registry():
registry.typeMetaData = "registry_mock"
def findInstanceContainers(registry, **kwargs):
if "id" not in kwargs:
return [MockContainer("test_container", registry.typeMetaData)]
else:
return [MockContainer(id, registry.typeMetaData)]
container_id = kwargs.get("id", default = "test_container")
return [MockContainer(container_id, registry.typeMetaData)]
registry.findInstanceContainers = functools.partial(findInstanceContainers, registry)
def findContainers(registry, container_type = None, id = None):
if not id:
return [MockContainer("test_container", registry.typeMetaData)]
else:
id = "test_container"
return [MockContainer(id, registry.typeMetaData)]
registry.findContainers = functools.partial(findContainers, registry)