def testMarkdone(self, getStubMock):
stubMock = mock.Mock()
stubMock.MarkdoneFrames.return_value = job_pb2.LayerMarkdoneFramesResponse()
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.markdone()
stubMock.MarkdoneFrames.assert_called_with(
job_pb2.LayerMarkdoneFramesRequest(layer=layer.data), timeout=mock.ANY)
def markdone(self):
"""Drops any dependency that requires this layer or requires any frame
in the layer"""
return self.stub.MarkdoneFrames(
job_pb2.LayerMarkdoneFramesRequest(layer=self.data),
timeout=Cuebot.Timeout)
