def getFrames(self, **options): """Returns the list of up to 1000 frames from within the job. frames = job.getFrames(show=["edu","beo"],user="******") frames = job.getFrames(show="edu",shot="bs.012") Allowed: offset, limit, states+, layers+. frameset, changedate @rtype: list<Frame> @return: List of frames""" criteria = opencue.search.FrameSearch.criteriaFromOptions(**options) response = self.stub.GetFrames(job_pb2.JobGetFramesRequest(job=self.data, req=criteria), timeout=Cuebot.Timeout) frameSeq = response.frames return [opencue.wrappers.frame.Frame(frm) for frm in frameSeq.frames]
def testGetFrames(self, getStubMock): frameNames = ['testFrameA', 'testFrameB'] stubMock = mock.Mock() stubMock.GetFrames.return_value = job_pb2.JobGetFramesResponse( frames=job_pb2.FrameSeq(frames=[job_pb2.Frame(name=frameNames[0]), job_pb2.Frame(name=frameNames[1])])) 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)) frames = job.getFrames(range=frameRange) stubMock.GetFrames.assert_called_with( job_pb2.JobGetFramesRequest(job=job.data, req=criteria), timeout=mock.ANY) self.assertTrue(len(frames), 2) self.assertTrue(frames[0].name(), frameNames[0]) self.assertTrue(frames[1].name(), frameNames[1])