def testLaunchSpecAndWait(self, getStubMock):
spec = 'arbitrary-spec'
stubMock = mock.Mock()
stubMock.LaunchSpecAndWait.return_value = job_pb2.JobLaunchSpecAndWaitResponse(
jobs=job_pb2.JobSeq(jobs=[job_pb2.Job(name=TEST_JOB_NAME)]))
getStubMock.return_value = stubMock
jobs = opencue.api.launchSpecAndWait(spec)
stubMock.LaunchSpecAndWait.assert_called_with(
job_pb2.JobLaunchSpecAndWaitRequest(spec=spec), timeout=mock.ANY)
self.assertEqual([TEST_JOB_NAME], [job.name() for job in jobs])
def launchSpecAndWait(spec):
"""Launch a new job with the given spec xml data.
This call waits on the server until the job is committed
in the database.
@type spec: str
@param spec: XML string containing job spec
@rtype: List<Job>
@return: List of Jobs that were submitted
"""
jobSeq = Cuebot.getStub('job').LaunchSpecAndWait(
job_pb2.JobLaunchSpecAndWaitRequest(spec=spec),
timeout=Cuebot.Timeout).jobs
return [Job(j) for j in jobSeq.jobs]
