X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/ebb2559b3a09636ff687316bbe512e0e8a86b168..53b24cf4fafa5c8ce76eb4abf5e61de944bb4a6a:/sdk/cwl/tests/test_job.py diff --git a/sdk/cwl/tests/test_job.py b/sdk/cwl/tests/test_job.py index 15da596eae..076514b1e9 100644 --- a/sdk/cwl/tests/test_job.py +++ b/sdk/cwl/tests/test_job.py @@ -11,6 +11,7 @@ import arvados import arvados_cwl import cwltool.process from schema_salad.ref_resolver import Loader +from schema_salad.sourceline import cmap from .mock_discovery import get_rootDesc from .matcher import JsonDiffMatcher @@ -34,12 +35,12 @@ class TestJob(unittest.TestCase): list_images_in_arv.return_value = [["zzzzz-4zz18-zzzzzzzzzzzzzzz"]] runner.api.collections().get().execute.return_vaulue = {"portable_data_hash": "99999999999999999999999999999993+99"} - tool = { + tool = cmap({ "inputs": [], "outputs": [], "baseCommand": "ls", "arguments": [{"valueFrom": "$(runtime.outdir)"}] - } + }) make_fs_access=functools.partial(arvados_cwl.CollectionFsAccess, api_client=runner.api) arvtool = arvados_cwl.ArvadosCommandTool(runner, tool, work_api="jobs", avsc_names=avsc_names, basedir="", make_fs_access=make_fs_access, loader=Loader({})) @@ -61,7 +62,7 @@ class TestJob(unittest.TestCase): 'repository': 'arvados', 'script': 'crunchrunner', 'runtime_constraints': { - 'docker_image': 'arvados/jobs:'+arvados_cwl.__version__, + 'docker_image': 'arvados/jobs', 'min_cores_per_node': 1, 'min_ram_mb_per_node': 1024, 'min_scratch_mb_per_node': 2048 # tmpdirSize + outdirSize @@ -71,7 +72,7 @@ class TestJob(unittest.TestCase): filters=[['repository', '=', 'arvados'], ['script', '=', 'crunchrunner'], ['script_version', 'in git', 'a3f2cb186e437bfce0031b024b2157b73ed2717d'], - ['docker_image_locator', 'in docker', 'arvados/jobs:'+arvados_cwl.__version__]] + ['docker_image_locator', 'in docker', 'arvados/jobs']] ) # The test passes some fields in builder.resources @@ -129,7 +130,7 @@ class TestJob(unittest.TestCase): 'repository': 'arvados', 'script': 'crunchrunner', 'runtime_constraints': { - 'docker_image': 'arvados/jobs:'+arvados_cwl.__version__, + 'docker_image': 'arvados/jobs', 'min_cores_per_node': 3, 'min_ram_mb_per_node': 3000, 'min_scratch_mb_per_node': 5024, # tmpdirSize + outdirSize @@ -140,7 +141,7 @@ class TestJob(unittest.TestCase): filters=[['repository', '=', 'arvados'], ['script', '=', 'crunchrunner'], ['script_version', 'in git', 'a3f2cb186e437bfce0031b024b2157b73ed2717d'], - ['docker_image_locator', 'in docker', 'arvados/jobs:'+arvados_cwl.__version__]]) + ['docker_image_locator', 'in docker', 'arvados/jobs']]) @mock.patch("arvados.collection.CollectionReader") def test_done(self, reader): @@ -294,18 +295,21 @@ class TestWorkflow(unittest.TestCase): 'runtime_constraints': { 'min_scratch_mb_per_node': 2048, 'min_cores_per_node': 1, - 'docker_image': 'arvados/jobs:'+arvados_cwl.__version__, + 'docker_image': 'arvados/jobs', 'min_ram_mb_per_node': 1024 }, 'owner_uuid': 'zzzzz-8i9sb-zzzzzzzzzzzzzzz'}), filters=[['repository', '=', 'arvados'], ['script', '=', 'crunchrunner'], ['script_version', 'in git', 'a3f2cb186e437bfce0031b024b2157b73ed2717d'], - ['docker_image_locator', 'in docker', 'arvados/jobs:'+arvados_cwl.__version__]], + ['docker_image_locator', 'in docker', 'arvados/jobs']], find_or_create=True) mockcollection().open().__enter__().write.assert_has_calls([mock.call(subwf)]) - mockcollection().open().__enter__().write.assert_has_calls([mock.call('{sleeptime: 5}')]) + mockcollection().open().__enter__().write.assert_has_calls([mock.call( +'''{ + "sleeptime": 5 +}''')]) def test_default_work_api(self): arvados_cwl.add_arv_hints()