def test_containers_provider(
container_provider: ContainerProvider,
upgrade_manager: VersionUpgradeManager) -> ContainerProvider:
my_folder = os.path.dirname(os.path.abspath(__file__))
definition_ids = {
"basic_definition", "children", "functions", "inherits",
"metadata_definition", "multiple_settings", "single_setting"
}
for definition_id in definition_ids:
container = DefinitionContainer(definition_id)
container.deserialize(
open(
os.path.join(my_folder, "definitions",
definition_id + ".def.json")).read())
container_provider._containers[definition_id] = container
container_provider.addMetadata(container.getMetaData())
instance_ids = {"basic_instance", "metadata_instance", "setting_values"}
for instance_id in instance_ids:
container = UM.Settings.InstanceContainer.InstanceContainer(
instance_id)
container.deserialize(
open(
os.path.join(my_folder, "instances",
instance_id + ".inst.cfg")).read())
container_provider._containers[instance_id] = container
container_provider.addMetadata(container.getMetaData())
return container_provider
def test_addContainer(container_registry):
definition_container_0 = DefinitionContainer("a")
assert definition_container_0.getMetaData() not in container_registry.findDefinitionContainersMetadata() # Sanity check.
assert definition_container_0 not in container_registry.findDefinitionContainers()
container_registry.addContainer(definition_container_0)
assert definition_container_0.getMetaData() in container_registry.findDefinitionContainersMetadata()
assert definition_container_0 in container_registry.findDefinitionContainers()
# Add a second one of the same type.
definition_container_1 = DefinitionContainer("b")
assert definition_container_1.getMetaData() not in container_registry.findDefinitionContainersMetadata() # Sanity check.
assert definition_container_1 not in container_registry.findDefinitionContainers() # Sanity check.
container_registry.addContainer(definition_container_1)
assert definition_container_1.getMetaData() in container_registry.findDefinitionContainersMetadata()
assert definition_container_1 in container_registry.findDefinitionContainers()
assert definition_container_0.getMetaData() in container_registry.findDefinitionContainersMetadata()
assert definition_container_0 in container_registry.findDefinitionContainers()
# Add a container with the same type and same ID.
definition_container_1_clone = DefinitionContainer("b")
container_registry.addContainer(definition_container_1_clone)
#Since comparing metadata is a deep comparison, you'll find that the metadata of the clone got in there. But it's not, it's just exactly the same as the original metadata so it appears as if it's in there.
assert definition_container_1_clone not in container_registry.findDefinitionContainers()
# For good measure, add a container with a different type too.
instance_container_1 = InstanceContainer("a")
assert instance_container_1.getMetaData() not in container_registry.findDefinitionContainersMetadata() # Sanity check.
assert instance_container_1 not in container_registry.findDefinitionContainers()
container_registry.addContainer(instance_container_1)
assert instance_container_1.getMetaData() not in container_registry.findDefinitionContainersMetadata()
assert instance_container_1 not in container_registry.findDefinitionContainers()
def test_containers_provider(container_provider: ContainerProvider, upgrade_manager: VersionUpgradeManager) -> ContainerProvider:
my_folder = os.path.dirname(os.path.abspath(__file__))
definition_ids = {"basic_definition", "children", "functions", "inherits", "metadata_definition", "multiple_settings", "single_setting"}
for definition_id in definition_ids:
container = DefinitionContainer(definition_id)
container.deserialize(open(os.path.join(my_folder, "definitions", definition_id + ".def.json"), encoding = "utf-8").read())
container_provider._containers[definition_id] = container
container_provider.addMetadata(container.getMetaData())
instance_ids = {"basic_instance", "metadata_instance", "setting_values"}
for instance_id in instance_ids:
container = UM.Settings.InstanceContainer.InstanceContainer(instance_id)
container.deserialize(open(os.path.join(my_folder, "instances", instance_id + ".inst.cfg"), encoding = "utf-8").read())
container_provider._containers[instance_id] = container
container_provider.addMetadata(container.getMetaData())
return container_provider
def test_findDefinitionContainers(container_registry, data):
for container in data["containers"]: # Fill the registry with mock containers.
definition_container = DefinitionContainer(container["id"])
for key, value in container.items(): # Copy data into metadata.
definition_container.getMetaData()[key] = value
container_registry.addContainer(definition_container)
results = container_registry.findDefinitionContainers(**data["filter"]) # The actual function call we're testing.
_verifyMetaDataMatches(results, data["result"])
def test_findDefinitionContainers(container_registry, data):
for container in data["containers"]: # Fill the registry with mock containers.
container = container.copy()
container_id = container["id"]
del container["id"]
definition_container = DefinitionContainer(container_id)
for key, value in container.items(): # Copy data into metadata.
definition_container.getMetaData()[key] = value
container_registry.addContainer(definition_container)
results = container_registry.findDefinitionContainers(**data["filter"]) # The actual function call we're testing.
_verifyMetaDataMatches(results, data["result"])
def test_roundtrip_definition(tmpdir, process_count, loaded_container_registry):
definition = loaded_container_registry.findDefinitionContainers(id = "multiple_settings")[0]
temp_file = tmpdir.join("container_stack_test")
mp_run(process_count, write_data, temp_file, definition)
assert len(list(tmpdir.listdir())) == 1
results = mp_run(process_count, read_data, temp_file)
for result in results:
deserialized_definition = DefinitionContainer("multiple_settings")
deserialized_definition.deserialize(result)
assert deserialized_definition.getName() == definition.getName()
assert deserialized_definition.getMetaData() == definition.getMetaData()
def test_roundtrip_stack(tmpdir, process_count, loaded_container_registry):
definition = loaded_container_registry.findDefinitionContainers(id = "multiple_settings")[0]
temp_file = tmpdir.join("container_stack_test")
mp_run(process_count, write_data, temp_file, definition)
assert len(list(tmpdir.listdir())) == 1
results = mp_run(process_count, read_data, temp_file)
for result in results:
deserialized_definition = DefinitionContainer("test_definition")
deserialized_definition.deserialize(result)
assert deserialized_definition.getName() == definition.getName()
assert deserialized_definition.getMetaData() == definition.getMetaData()
