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)
