def getDepends(self):
"""Returns a list of all depends this job is involved with
@rtype: list<Depend>
@return: all depends involved with this job"""
response = self.stub.GetDepends(
job_pb2.JobGetDependsRequest(job=self.data),
timeout=Cuebot.Timeout)
dependSeq = response.depends
return [depend.Depend(depend) for depend in dependSeq.depends]
def testGetDepends(self, getStubMock):
dependId = 'ddd-dddd-ddd'
stubMock = mock.Mock()
stubMock.GetDepends.return_value = job_pb2.JobGetDependsResponse(
depends=depend_pb2.DependSeq(
depends=[depend_pb2.Depend(id=dependId)]))
getStubMock.return_value = stubMock
job = opencue.wrappers.job.Job(job_pb2.Job(name=TEST_JOB_NAME))
depends = job.getDepends()
stubMock.GetDepends.assert_called_with(
job_pb2.JobGetDependsRequest(job=job.data), timeout=mock.ANY)
self.assertEqual(len(depends), 1)
self.assertEqual(depends[0].id(), dependId)