Python LayerRetryFramesRequestの例

コード例 #1

    def testRetry(self, getStubMock):
        stubMock = mock.Mock()
        stubMock.RetryFrames.return_value = job_pb2.LayerRetryFramesResponse()
        getStubMock.return_value = stubMock

        layer = opencue.wrappers.layer.Layer(
            job_pb2.Layer(name=TEST_LAYER_NAME))
        layer.retry()

        stubMock.RetryFrames.assert_called_with(
            job_pb2.LayerRetryFramesRequest(layer=layer.data), timeout=mock.ANY)

コード例 #2

ファイル: layer.py プロジェクト: lai3d/OpenCue

 def retry(self):
     """Retry entire layer"""
     return self.stub.RetryFrames(
         job_pb2.LayerRetryFramesRequest(layer=self.data),
         timeout=Cuebot.Timeout)