def getWhatThisDependsOn(self): """Returns a list of dependencies that this frame depends on @rtype: list<Depend> @return: List of dependencies that this frame depends on""" response = self.stub.GetWhatThisDependsOn( job_pb2.FrameGetWhatThisDependsOnRequest(frame=self.data), timeout=Cuebot.Timeout) return [opencue.wrappers.depend.Depend(dep) for dep in response.depends.depends]
def testGetWhatThisDependsOn(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.GetWhatThisDependsOn.return_value = job_pb2.FrameGetWhatThisDependsOnResponse( depends=depend_pb2.DependSeq(depends=[depend_pb2.Depend(id=dependId)])) getStubMock.return_value = stubMock frame = opencue.wrappers.frame.Frame( job_pb2.Frame(name=TEST_FRAME_NAME, state=job_pb2.RUNNING)) depends = frame.getWhatThisDependsOn() stubMock.GetWhatThisDependsOn.assert_called_with( job_pb2.FrameGetWhatThisDependsOnRequest(frame=frame.data), timeout=mock.ANY) self.assertEqual(len(depends), 1) self.assertEqual(depends[0].id(), dependId)