def test_tojson_without_profiles_hooks_metadata(
self, convert_yaml_schemas_to_json, load_schema
):
t = Transformation(
"test",
namespace="pegasus",
site="local",
pfn="/pfn",
is_stageable=True,
bypass_staging=True,
checksum={"sha256": "abc123"},
)
t.add_requirement("required")
result = json.loads(json.dumps(t, cls=_CustomEncoder))
expected = {
"name": "test",
"namespace": "pegasus",
"checksum": {"sha256": "abc123"},
"requires": ["required"],
"sites": [
{"name": "local", "pfn": "/pfn", "type": "stageable", "bypass": True}
],
}
transformation_schema = load_schema("tc-5.0.json")["$defs"]["transformation"]
validate(instance=result, schema=transformation_schema)
assert result == expected
def test_write(self):
tc = TransformationCatalog()
(tc.add_transformations(Transformation("t1")).add_transformations(
Transformation("t2")))
expected = {
"pegasus":
"5.0",
"transformations": [
{
"name": "t1",
"sites": []
},
{
"name": "t2",
"sites": []
},
],
}
expected["transformations"] = sorted(expected["transformations"],
key=lambda t: t["name"])
with NamedTemporaryFile("r+") as f:
tc.write(f, _format="json")
f.seek(0)
result = json.load(f)
result["transformations"] = sorted(expected["transformations"],
key=lambda t: t["name"])
assert result == expected
def test_write(self):
tc = TransformationCatalog()
(
tc.add_transformations(Transformation("t1")).add_transformations(
Transformation("t2")
)
)
expected = {
"pegasus": "5.0",
"transformations": [
{"name": "t1", "sites": []},
{"name": "t2", "sites": []},
],
}
expected["transformations"] = sorted(
expected["transformations"], key=lambda t: t["name"]
)
with NamedTemporaryFile("r+") as f:
tc.write(f, _format="json")
f.seek(0)
result = json.load(f)
result["transformations"] = sorted(
expected["transformations"], key=lambda t: t["name"]
)
assert "createdOn" in result["x-pegasus"]
assert result["x-pegasus"]["createdBy"] == getpass.getuser()
assert result["x-pegasus"]["apiLang"] == "python"
del result["x-pegasus"]
assert result == expected
def test_add_invalid_requirement_as_str(self, namespace, name, version,
bad_field):
t = Transformation("test")
with pytest.raises(ValueError) as e:
t.add_requirement(name, namespace, version)
assert "invalid {bad_field}".format(bad_field=bad_field) in str(e)
def test_example_transformation_catalog(
self, convert_yaml_schemas_to_json, load_schema, _format, loader
):
# validates the sample tc in pegasus/etc/sample-5.0-data/tc.yml
tc = TransformationCatalog()
foo = (
Transformation("foo")
.add_globus_profile(max_time=2)
.add_dagman_profile(retry=2)
.add_metadata(size=2048)
.add_sites(
TransformationSite(
"local",
"/nfs/u2/ryan/bin/foo",
True,
arch=Arch.X86_64,
os_type=OS.LINUX,
)
.add_env(JAVA_HOME="/usr/bin/java")
.add_metadata(size=2048)
)
.add_requirement("bar")
.add_shell_hook(EventType.START, "/bin/echo 'starting'")
)
bar = Transformation("bar").add_sites(
TransformationSite(
"local",
"/nfs/u2/ryan/bin/bar",
True,
arch=Arch.X86_64,
os_type=OS.LINUX,
)
)
centos_pegasus_container = Container(
"centos-pegasus",
Container.DOCKER,
"docker:///ryan/centos-pegasus:latest",
arguments="--shm-size 123",
mounts=["/Volumes/Work/lfs1:/shared-data/:ro"],
).add_env(JAVA_HOME="/usr/bin/java")
(tc.add_transformations(foo, bar).add_containers(centos_pegasus_container))
with NamedTemporaryFile(mode="r+") as f:
tc.write(f, _format=_format)
f.seek(0)
tc_json = loader(f)
tc_schema = load_schema("tc-5.0.json")
validate(instance=tc_json, schema=tc_schema)
def test_add_multiple_transformations(self):
tc = TransformationCatalog()
t1 = Transformation("name")
t2 = Transformation("name", namespace="namespace")
t3 = Transformation("name", namespace="namespace", version="version")
tc.add_transformations(t1, t2, t3)
assert "None::name::None" in tc.transformations
assert "namespace::name::None" in tc.transformations
assert "namespace::name::version" in tc.transformations
assert len(tc.transformations) == 3
def test_chaining(self):
tc = TransformationCatalog()
(tc.add_transformations(Transformation("t1")).add_transformations(
Transformation("t2")).add_containers(
Container("container1", Container.DOCKER, "image",
["mount1", "mount2"])).add_containers(
Container("container2", Container.DOCKER,
"image", ["mount1", "mount2"])))
assert "None::t1::None" in tc.transformations
assert "None::t2::None" in tc.transformations
assert "container1" in tc.containers
assert "container2" in tc.containers
def test_tojson_no_containers(self, convert_yaml_schemas_to_json,
load_schema):
tc = TransformationCatalog()
(tc.add_transformations(
Transformation("t1").add_sites(
TransformationSite("local", "/pfn",
False))).add_transformations(
Transformation("t2").add_sites(
TransformationSite(
"local2", "/pfn", True))))
expected = {
"pegasus":
PEGASUS_VERSION,
"transformations": [
{
"name":
"t1",
"sites": [{
"name": "local",
"pfn": "/pfn",
"type": "installed"
}],
},
{
"name":
"t2",
"sites": [{
"name": "local2",
"pfn": "/pfn",
"type": "stageable"
}],
},
],
}
expected["transformations"] = sorted(expected["transformations"],
key=lambda t: t["name"])
result = json.loads(json.dumps(tc, cls=_CustomEncoder))
result["transformations"] = sorted(result["transformations"],
key=lambda t: t["name"])
tc_schema = load_schema("tc-5.0.json")
validate(instance=result, schema=tc_schema)
assert expected == result
def test_invalid_pfn(self):
with pytest.raises(ValueError) as e:
Transformation(
"executable", site="local", pfn=Path("."), is_stageable=False
)
assert "invalid pfn" in str(e)
def test_add_duplicate_site(self):
with pytest.raises(DuplicateError) as e:
t = Transformation("test")
t.add_sites(TransformationSite("local", "/pfn", True))
t.add_sites(TransformationSite("isi", "/pfn", True))
t.add_sites(TransformationSite("local", "/pfn", True))
assert "local" in str(e)
def test_chaining(self):
t = (Transformation("test").add_sites(
TransformationSite("local", "/pfn", True).add_env(
JAVA_HOME="/java/home")).add_requirement("required"))
assert "local" in t.sites
assert t.sites["local"].profiles["env"]["JAVA_HOME"] == "/java/home"
assert "required" in t.requires
def test_add_single_tranformation_site_constructor(self):
t1 = Transformation("t1")
assert len(t1.sites) == 0
# pfn is not provided, TransformationSite should not be added
t2 = Transformation("t2", site="local")
assert len(t2.sites) == 0
# site and pfn provided, TransformationSite should be added
t3 = Transformation("t3", site="local", pfn="/t3")
assert len(t3.sites) == 1
assert t3.sites["local"].transformation_type == "installed"
# site and pfn provided, TransformationSite should be added
t4 = Transformation("t4", site="local", pfn="/t4", is_stageable=True)
assert len(t4.sites) == 1
assert t4.sites["local"].transformation_type == "stageable"
def test_add_duplicate_requirement(self, transformation):
t = Transformation("test")
t.add_requirement(transformation)
with pytest.raises(DuplicateError) as e:
t.add_requirement(transformation)
assert "transformation: required" in str(e)
def tc2():
return (TransformationCatalog().add_transformations(
Transformation("t1", namespace="test", version="1.0").add_sites(
TransformationSite(
"local",
"/pfn",
True,
))).add_containers(
Container(
"cont",
Container.DOCKER,
"docker:///ryan/centos-pegasus:latest",
mounts=["/Volumes/Work/lfs1:/shared-data/:ro"],
image_site="local",
)))
def test_transformation_catalog_ordering_on_yml_write(self):
tc = TransformationCatalog()
tc.add_transformations(Transformation("t1"))
tc.add_containers(Container("c1", Container.DOCKER, "img"))
tc.write()
EXPECTED_FILE = Path("transformations.yml")
with EXPECTED_FILE.open() as f:
result = f.read()
EXPECTED_FILE.unlink()
"""
Check that tc keys have been ordered as follows:
- pegasus
- transformations
- containers
"""
p = re.compile(
r"pegasus: '5.0'[\w\W]+transformations:[\w\W]+containers[\w\W]+")
assert p.match(result) is not None
def tc1():
return (TransformationCatalog().add_transformations(
Transformation("t1", namespace="test", version="1.0").add_sites(
TransformationSite(
"local",
"/pfn",
True,
arch=Arch.X86_64,
os_type=OS.LINUX,
os_release="1",
os_version="1",
container="cont",
).add_dagman_profile(retry="3").add_metadata(
JAVA_HOME="/usr/bin/java")).add_requirement(
"t2", namespace="test", version="1.0").add_shell_hook(
EventType.START, "echo hello")).add_containers(
Container(
"cont",
Container.DOCKER,
"docker:///ryan/centos-pegasus:latest",
mounts=["/Volumes/Work/lfs1:/shared-data/:ro"],
image_site="local",
).add_env(JAVA_HOME="/usr/bin/java")))
def test_tojson_without_profiles_hooks_metadata(
self, convert_yaml_schemas_to_json, load_schema):
t = Transformation("test", namespace="pegasus")
t.add_sites(TransformationSite("local", "/pfn", True))
t.add_requirement("required")
result = json.loads(json.dumps(t, cls=_CustomEncoder))
expected = {
"name": "test",
"namespace": "pegasus",
"requires": ["required"],
"sites": [{
"name": "local",
"pfn": "/pfn",
"type": "stageable"
}],
}
transformation_schema = load_schema(
"tc-5.0.json")["$defs"]["transformation"]
validate(instance=result, schema=transformation_schema)
assert result == expected
def test_add_single_transformation(self):
tc = TransformationCatalog()
tc.add_transformations(Transformation("test"))
assert "None::test::None" in tc.transformations
assert len(tc.transformations) == 1
def test_tojson_with_profiles_hooks_metadata(
self, convert_yaml_schemas_to_json, load_schema
):
t = Transformation("test", namespace="pegasus")
t.add_sites(
TransformationSite("local", "/pfn", True).add_env(JAVA_HOME="/java/home")
)
t.add_requirement("required")
t.add_env(JAVA_HOME="/java/home")
t.add_shell_hook(EventType.START, "/bin/echo hi")
t.add_metadata(key="value")
result = json.loads(json.dumps(t, cls=_CustomEncoder))
expected = {
"name": "test",
"namespace": "pegasus",
"requires": ["required"],
"sites": [
{
"name": "local",
"pfn": "/pfn",
"type": "stageable",
"profiles": {"env": {"JAVA_HOME": "/java/home"}},
}
],
"metadata": {"key": "value"},
"profiles": {Namespace.ENV.value: {"JAVA_HOME": "/java/home"}},
"hooks": {"shell": [{"_on": EventType.START.value, "cmd": "/bin/echo hi"}]},
}
transformation_schema = load_schema("tc-5.0.json")["$defs"]["transformation"]
validate(instance=result, schema=transformation_schema)
assert result == expected
def test_add_invalid_requirement(self):
t = Transformation("test")
with pytest.raises(TypeError) as e:
t.add_requirement(1)
assert "invalid required_transformation: {tr}".format(tr=1) in str(e)
def test_add_single_transformation_site_constructor_with_valid_container(
self, container
):
assert Transformation("t", site="local", pfn="/t1", container=container)
def test_add_requirement_as_str(self, namespace, name, version, expected):
t = Transformation("test")
t.add_requirement(name, namespace=namespace, version=version)
assert expected in t.requires
def test_eq_invalid(self):
with pytest.raises(ValueError) as e:
Transformation("name") == "tr"
assert "Transformation cannot be compared with" in str(e)
def test_hash(self):
assert hash(Transformation("name")) == hash("None::name::None")
def test_add_single_transformation_site_constructor_with_invalid_container(self):
with pytest.raises(TypeError) as e:
Transformation("t", site="local", pfn="/t1", container=1)
assert "invalid container: 1" in str(e)
def test_add_duplicate_transformation(self):
tc = TransformationCatalog()
tc.add_transformations(Transformation("name"))
with pytest.raises(DuplicateError):
tc.add_transformations(Transformation("name", namespace=None, version=None))
def test_tojson(self, convert_yaml_schemas_to_json, load_schema):
tc = TransformationCatalog()
(
tc.add_transformations(
Transformation("t1").add_sites(
TransformationSite("local", "/pfn", False)
)
)
.add_transformations(
Transformation("t2").add_sites(
TransformationSite("local", "/pfn", False)
)
)
.add_containers(
Container(
"container1",
Container.DOCKER,
"image",
arguments="--shm-size 123",
mounts=["mount1"],
bypass_staging=True,
)
)
.add_containers(
Container("container2", Container.DOCKER, "image", mounts=["mount1"])
)
)
expected = {
"pegasus": PEGASUS_VERSION,
"transformations": [
{
"name": "t1",
"sites": [{"name": "local", "pfn": "/pfn", "type": "installed"}],
},
{
"name": "t2",
"sites": [{"name": "local", "pfn": "/pfn", "type": "installed"}],
},
],
"containers": [
{
"name": "container1",
"type": "docker",
"image": "image",
"mounts": ["mount1"],
"bypass": True,
"profiles": {"pegasus": {"container.arguments": "--shm-size 123"}},
},
{
"name": "container2",
"type": "docker",
"image": "image",
"mounts": ["mount1"],
},
],
}
expected["transformations"] = sorted(
expected["transformations"], key=lambda t: t["name"]
)
expected["containers"] = sorted(expected["containers"], key=lambda c: c["name"])
result = json.loads(json.dumps(tc, cls=_CustomEncoder))
result["transformations"] = sorted(
result["transformations"], key=lambda t: t["name"]
)
result["containers"] = sorted(result["containers"], key=lambda c: c["name"])
tc_schema = load_schema("tc-5.0.json")
validate(instance=result, schema=tc_schema)
assert result == expected
def test_add_requirement_as_transformation_object(self, transformation, expected):
t = Transformation("test")
t.add_requirement(transformation)
assert expected in t.requires
def test_add_site(self):
t = Transformation("test")
t.add_sites(TransformationSite("local", "/pfn", True))
assert "local" in t.sites
def test_add_invalid_site(self):
with pytest.raises(TypeError) as e:
t = Transformation("test")
t.add_sites("badsite")
assert "badsite" in str(e)
