def createDependencyOnLayer(self, layer): """Create and return a frame on layer dependency @type layer: opencue.wrappers.layer.Layer @param layer: the layer you want this frame to depend on @rtype: opencue.wrappers.depend.Depend @return: The new dependency""" response = self.stub.CreateDependencyOnLayer( job_pb2.FrameCreateDependencyOnLayerRequest(frame=self.data, layer=layer.data), timeout=Cuebot.Timeout) return opencue.wrappers.depend.Depend(response.depend)
def testCreateDependencyOnLayer(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.CreateDependencyOnLayer.return_value = \ job_pb2.FrameCreateDependencyOnLayerResponse(depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock dependLayerName = 'layerDependTest' frame = opencue.wrappers.frame.Frame( job_pb2.Frame(name=TEST_FRAME_NAME, state=job_pb2.RUNNING)) dependOnLayer = job_pb2.Layer(name=dependLayerName) depend = frame.createDependencyOnLayer(dependOnLayer) stubMock.CreateDependencyOnLayer.assert_called_with( job_pb2.FrameCreateDependencyOnLayerRequest(frame=frame.data, layer=dependOnLayer), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)