def getFrame(self):
"""Return the frame this proc is running.
@rtype: Frame
@return: The fame this proc is running."""
response = self.stub.GetFrame(
host_pb2.ProcGetFrameRequest(proc=self.data),
timeout=Cuebot.Timeout)
return opencue.wrappers.frame.Frame(response.frame)
def testGetFrame(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetFrame.return_value = host_pb2.ProcGetFrameResponse(
frame=job_pb2.Frame(layer_name=TEST_LAYER_NAME))
getStubMock.return_value = stubMock
proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME))
frame = proc.getFrame()
stubMock.GetFrame.assert_called_with(
host_pb2.ProcGetFrameRequest(proc=proc.data), timeout=mock.ANY)
self.assertEqual(frame.layer(), TEST_LAYER_NAME)
