def createDependencyOnLayer(self, layer): """Create and return a job on layer dependency @type layer: opencue.wrappers.layer.Layer @param layer: the layer you want this job to depend on @rtype: opencue.wrappers.Depend @return: the new dependency""" response = self.stub.CreateDependencyOnLayer( job_pb2.JobCreateDependencyOnLayerRequest(job=self.data, layer=layer.data), timeout=Cuebot.Timeout) return opencue.wrappers.depend.Depend(response.depend)
def testCreateDependencyOnLayer(self, getStubMock): dependId = 'ddd-dddd-ddd' dependLayer = 'testLayer' stubMock = mock.Mock() stubMock.CreateDependencyOnLayer.return_value = job_pb2.JobCreateDependencyOnLayerResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock onLayer = job_pb2.Layer(name=dependLayer) job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) depend = job.createDependencyOnLayer(onLayer) stubMock.CreateDependencyOnLayer.assert_called_with( job_pb2.JobCreateDependencyOnLayerRequest(job=job.data, layer=onLayer), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)