def createDependencyOnJob(self, job):
"""Create and return a job on job dependency
@type job: Job
@param job: the job you want this job to depend on
@rtype: Depend
@return: The new dependency"""
response = self.stub.CreateDependencyOnJob(
job_pb2.JobCreateDependencyOnJobRequest(job=self.data, on_job=job),
timeout=Cuebot.Timeout)
return depend.Depend(response.depend)
def createDependencyOnJob(self, job):
"""Create and return a job on job dependency.
:type job: opencue.wrappers.job.Job
:param job: the job you want this job to depend on
:rtype: opencue.wrappers.depend.Depend
:return: The new dependency"""
response = self.stub.CreateDependencyOnJob(
job_pb2.JobCreateDependencyOnJobRequest(job=self.data, on_job=job.data),
timeout=Cuebot.Timeout)
return opencue.wrappers.depend.Depend(response.depend)
def testCreateDependencyOnJob(self, getStubMock):
dependId = 'ddd-dddd-ddd'
stubMock = mock.Mock()
stubMock.CreateDependencyOnJob.return_value = job_pb2.JobCreateDependencyOnJobResponse(
depend=depend_pb2.Depend(id=dependId))
getStubMock.return_value = stubMock
onJob = job_pb2.Job(name=TEST_JOB_NAME+"Depend")
job = opencue.wrappers.job.Job(
job_pb2.Job(name=TEST_JOB_NAME))
depend = job.createDependencyOnJob(onJob)
stubMock.CreateDependencyOnJob.assert_called_with(
job_pb2.JobCreateDependencyOnJobRequest(job=job.data, on_job=onJob),
timeout=mock.ANY)
self.assertEqual(depend.id(), dependId)
