def eatFrames(self, **request):
"""Eats all frames that match the FrameSearch
@type request: Dict
@param request: FrameSearch parameters"""
criteria = opencue.search.FrameSearch.criteriaFromOptions(**request)
return self.stub.EatFrames(job_pb2.JobEatFramesRequest(job=self.data, req=criteria),
timeout=Cuebot.Timeout)
def testEatFrames(self, getStubMock):
stubMock = mock.Mock()
stubMock.EatFrames.return_value = job_pb2.JobEatFramesResponse()
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.eatFrames(range=frameRange)
stubMock.EatFrames.assert_called_with(
job_pb2.JobEatFramesRequest(job=job.data, req=criteria), timeout=mock.ANY)
