def retryFrames(self, **request):
"""Retries all frames that match the FrameSearch
@type request: Dict
@param request: FrameSearch parameters"""
criteria = opencue.search.FrameSearch.criteriaFromOptions(**request)
return self.stub.RetryFrames(job_pb2.JobRetryFramesRequest(job=self.data, req=criteria),
timeout=Cuebot.Timeout)
def testRetryFrames(self, getStubMock):
stubMock = mock.Mock()
stubMock.RetryFrames.return_value = job_pb2.JobRetryFramesResponse()
getStubMock.return_value = stubMock
frameRange = '1-10'
criteria = opencue.search.FrameSearch.criteriaFromOptions(range=frameRange)
job = opencue.wrappers.job.Job(
job_pb2.Job(name=TEST_JOB_NAME))
job.retryFrames(range=frameRange)
stubMock.RetryFrames.assert_called_with(
job_pb2.JobRetryFramesRequest(job=job.data, req=criteria), timeout=mock.ANY)
