def test_cp_wasb_args(self):
assert (
cp_wasb_args(path_from="wasb://foo",
path_to="/local",
is_file=False) ==
"polyaxon initializer wasb --path_from=wasb://foo --path_to=/local "
)
def test_get_volume_args_az(self):
az_store = V1ConnectionType(
name="test_az",
kind=V1ConnectionKind.WASB,
schema=V1BucketConnection(bucket="Conwasb://[email protected]"),
)
path_to = "/path/to/"
path_from = os.path.join(az_store.store_path, "")
assert get_volume_args(az_store, path_to, None) == "; ".join(
[
get_or_create_args(path=path_to),
cp_wasb_args(path_from=path_from, path_to=path_to, is_file=False),
]
)
az_store = V1ConnectionType(
name="test_az",
kind=V1ConnectionKind.WASB,
schema=V1BucketConnection(bucket="Conwasb://[email protected]"),
)
base_path = "/path/to/"
path_to1 = "/path/to/path1"
path_to2 = "/path/to/path2"
path_from1 = os.path.join(az_store.store_path, "path1")
path_from2 = os.path.join(az_store.store_path, "path2")
assert get_volume_args(
az_store,
"/path/to",
artifacts=V1ArtifactsType(files=["path1"], dirs=["path2"]),
) == "; ".join(
[
get_or_create_args(path=base_path),
cp_wasb_args(path_from=path_from1, path_to=path_to1, is_file=True),
get_or_create_args(path=base_path),
cp_wasb_args(path_from=path_from2, path_to=path_to2, is_file=False),
]
)
def test_files_cp_wasb_args(self):
assert cp_wasb_args(
path_from="wasb://foo", path_to="/local", is_file=True
) == (
"polyaxon initializer wasb --path_from=wasb://foo --path_to=/local --is_file"
)
