def findJob(name): """Returns a Job object for the given job name. This will only return one or zero active job. @type name: str @param name: a job name @rtype: Job @return: a Job object""" return Job( Cuebot.getStub('job').FindJob(job_pb2.JobFindJobRequest(name=name), timeout=Cuebot.Timeout).job)
def testFindJob(self, getStubMock): stubMock = mock.Mock() stubMock.FindJob.return_value = job_pb2.JobFindJobResponse( job=job_pb2.Job(name=TEST_JOB_NAME)) getStubMock.return_value = stubMock job = opencue.api.findJob(TEST_JOB_NAME) stubMock.FindJob.assert_called_with( job_pb2.JobFindJobRequest(name=TEST_JOB_NAME), timeout=mock.ANY) self.assertEqual(TEST_JOB_NAME, job.name())