def createDependencyOnJob(self, job):
"""Create and return a layer on job dependency
@type job: Job
@param job: the job you want this job to depend on
@rtype: opencue.depend.Depend
@return: the new dependency"""
response = self.stub.CreateDependOnJob(
job_pb2.LayerCreateDependOnJobRequest(layer=self.data, job=job),
timeout=Cuebot.Timeout)
return opencue.wrappers.depend.Depend(response.depend)
def createDependencyOnJob(self, job):
"""Creates and returns a layer-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.LayerCreateDependOnJobRequest(layer=self.data,
job=job.data),
timeout=Cuebot.Timeout)
return opencue.wrappers.depend.Depend(response.depend)
def testCreateDependencyOnJob(self, getStubMock):
dependId = 'dddd-ddd-dddd'
jobId = 'jjjj-jjj-jjjj'
stubMock = mock.Mock()
stubMock.CreateDependOnJob.return_value = job_pb2.LayerCreateDependOnJobResponse(
depend=depend_pb2.Depend(id=dependId))
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
job = job_pb2.Job(id=jobId)
depend = layer.createDependencyOnJob(job)
stubMock.CreateDependOnJob.assert_called_with(
job_pb2.LayerCreateDependOnJobRequest(layer=layer.data, job=job),
timeout=mock.ANY)
self.assertEqual(depend.id(), dependId)