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]