Adds a test that re-attaches a job, but to another executor instance than

the one used to create the job in the first place.

Author: hategan

tests/_test_tools.py

@@ -38,8 +38,9 @@ def assert_completed(job: Job, status: Optional[JobStatus]) -> None:
                              % (status.message, stdout, stderr))
 
 
-def _get_executor_instance(ep: ExecutorTestParams, job: Job) -> JobExecutor:
-    assert job.spec is not None
-    job.spec.launcher = ep.launcher
-    job.spec.attributes = JobAttributes(custom_attributes=ep.custom_attributes)
+def _get_executor_instance(ep: ExecutorTestParams, job: Optional[Job] = None) -> JobExecutor:
+    if job is not None:
+        assert job.spec is not None
+        job.spec.launcher = ep.launcher
+        job.spec.attributes = JobAttributes(custom_attributes=ep.custom_attributes)
     return JobExecutor.get_instance(ep.executor, url=ep.url)

tests/test_executor.py

@@ -45,6 +45,21 @@ def test_attach(execparams: ExecutorTestParams) -> None:
     assert_completed(job2, status)
 
 
+def test_attach2(execparams: ExecutorTestParams) -> None:
+    job = Job(JobSpec(executable='/bin/sleep', arguments=['1']))
+    ex = _get_executor_instance(execparams, job)
+    ex.submit(job)
+    job.wait(target_states=[JobState.ACTIVE, JobState.COMPLETED])
+    native_id = job.native_id
+
+    assert native_id is not None
+    job2 = Job()
+    ex2 = _get_executor_instance(execparams)
+    ex2.attach(job2, native_id)
+    status = job2.wait(timeout=_get_timeout(execparams))
+    assert_completed(job2, status)
+
+
 def test_cancel(execparams: ExecutorTestParams) -> None:
     job = Job(JobSpec(executable='/bin/sleep', arguments=['60']))
     ex = _get_executor_instance(execparams, job)