def createDependencyOnLayer(self, layer):
"""Create and return a layer on layer dependency
@type layer: Layer
@param layer: the layer you want this layer to depend on
@rtype: opencue.depend.Depend
@return: the new dependency"""
response = self.stub.CreateDependOnLayer(
job_pb2.LayerCreateDependOnLayerRequest(layer=self.data, depend_on_layer=layer),
timeout=Cuebot.Timeout)
return opencue.wrappers.depend.Depend(response.depend)
def createDependencyOnLayer(self, layer):
"""Creates and returns a layer-on-layer dependency.
:type layer: opencue.wrappers.layer.Layer
:param layer: the layer you want this layer to depend on
:rtype: opencue.wrappers.depend.Depend
:return: the new dependency
"""
response = self.stub.CreateDependencyOnLayer(
job_pb2.LayerCreateDependOnLayerRequest(
layer=self.data, depend_on_layer=layer.data),
timeout=Cuebot.Timeout)
return opencue.wrappers.depend.Depend(response.depend)
def testCreateDependencyOnLayer(self, getStubMock):
dependId = 'dddd-ddd-dddd'
layerId = 'llll-lll-llll'
stubMock = mock.Mock()
stubMock.CreateDependOnLayer.return_value = job_pb2.LayerCreateDependOnLayerResponse(
depend=depend_pb2.Depend(id=dependId))
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
dependLayer = job_pb2.Layer(id=layerId)
depend = layer.createDependencyOnLayer(dependLayer)
stubMock.CreateDependOnLayer.assert_called_with(
job_pb2.LayerCreateDependOnLayerRequest(layer=layer.data, depend_on_layer=dependLayer),
timeout=mock.ANY)
self.assertEqual(depend.id(), dependId)
