def test_conversion_classes():
clear_entity_classes()
support = Support(
"mymodule",
[
Definition("port", [IdArg("name", "the name", "id")]),
Definition("device", [ObjectArg("port", "the port", "object")]),
],
)
namespace = make_entity_classes(support)
assert {"device", "port"}.issubset(dir(namespace))
assert namespace.port.__definition__ == support.defs[0]
assert namespace.device.__definition__ == support.defs[1]
d = dict(
ioc_name="",
description="",
entities=[
dict(type="mymodule.port", name="PORT"),
dict(type="mymodule.device", port="PORT"),
],
generic_ioc_image="",
)
ioc = IOC.deserialize(d)
port, device = ioc.entities
assert port.type == "mymodule.port"
assert id_to_entity == {"PORT": port}
assert device.type == "mymodule.device"
assert device.port is port
def epics_classes(epics_support):
# clear the entity classes to make sure there's nothing left
clear_entity_classes()
# make entity subclasses for everything defined in it
namespace = make_entity_classes(epics_support)
# return the namespace so that callers have access to all of the
# generated dataclasses
return namespace
def test_asyn_schema(tmp_path: Path, samples: Path):
"""generate schema from the asyn support module definition yaml"""
clear_entity_classes()
schema_path = tmp_path / "asyn.ibek.entities.schema.json"
yaml_path = samples / "yaml" / "asyn.ibek.defs.yaml"
run_cli("ioc-schema", yaml_path, schema_path)
expected = json.loads(
(samples / "schemas" / "asyn.ibek.entities.schema.json").read_text())
actual = json.loads((schema_path).read_text())
assert expected == actual
def test_container_schema(tmp_path: Path, samples: Path):
clear_entity_classes()
"""generate schema for a container with two support modules"""
schema_combined = tmp_path / "container.ibek.entities.schema.json"
yaml_path1 = samples / "yaml" / "asyn.ibek.defs.yaml"
yaml_path2 = samples / "yaml" / "pmac.ibek.defs.yaml"
run_cli("ioc-schema", yaml_path1, yaml_path2, schema_combined)
expected = json.loads((samples / "schemas" /
"container.ibek.entities.schema.json").read_text())
actual = json.loads((schema_combined).read_text())
assert expected == actual
def test_loading_module_twice(tmp_path: Path, samples: Path):
"""
regression test to demonstrate that clear_entity_classes works and
allows us to call make_entity_classes more than once
"""
clear_entity_classes()
# When we deserialize the same yaml twice as we do in the full test suite
# we may get clashes in the namespace of generated Entity classes.
# This tests that we get a sensible error when we do
definition_file = samples / "yaml" / "pmac.ibek.defs.yaml"
support = Support.deserialize(YAML(typ="safe").load(definition_file))
make_entity_classes(support)
with pytest.raises(AssertionError) as ctx:
make_entity_classes(support)
assert str(ctx.value) == "Entity classes already created for pmac"
def test_build_startup_single(tmp_path: Path, samples: Path):
"""
build an ioc startup script from an IOC instance entity file
and a single support module definition file
"""
clear_entity_classes()
entity_file = samples / "yaml" / "bl45p-mo-ioc-02.ibek.entities.yaml"
definition_file = samples / "yaml" / "pmac.ibek.defs.yaml"
out_file = tmp_path / "ioc.boot"
run_cli("build-startup", entity_file, definition_file, out_file)
example_boot = (samples / "helm" / "ioc.boot").read_text()
actual_boot = out_file.read_text()
assert example_boot == actual_boot
def test_build_helm(tmp_path: Path, samples: Path):
"""build an ioc helm chart from an IOC instance entity file"""
clear_entity_classes()
entity_file = samples / "yaml" / "bl45p-mo-ioc-02.ibek.entities.yaml"
os.chdir(str(samples))
run_cli("build-helm", entity_file, tmp_path)
example_entity = entity_file.read_text()
actual_file = tmp_path / "bl45p-mo-ioc-02" / "config" / "ioc.boot.yaml"
actual_entity = actual_file.read_text()
assert example_entity == actual_entity
for test_file in ["Chart.yaml", "values.yaml"]:
example = (samples / "helm" / test_file).read_text()
actual_file = tmp_path / "bl45p-mo-ioc-02" / test_file
actual = actual_file.read_text()
assert example == actual
def test_build_startup_env_vars_and_post_ioc_init(tmp_path: Path,
samples: Path):
"""
build an ioc startup script from an IOC instance entity file, multiple
support module definition files which include environment variables and
post iocInit() entries
"""
clear_entity_classes()
entity_file = samples / "yaml" / "bl45p-mo-ioc-04.ibek.entities.yaml"
definition_file1 = samples / "yaml" / "epics.ibek.defs.yaml"
definition_file2 = samples / "yaml" / "pmac.ibek.defs.yaml"
out_file = tmp_path / "ioc.boot"
run_cli("build-startup", entity_file, definition_file1, definition_file2,
out_file)
example_boot = (samples / "helm" / "bl45p-mo-ioc-04.boot").read_text()
actual_boot = out_file.read_text()
assert example_boot == actual_boot