def getJobs(self):
"""Returns the jobs in this group
@rtype: list<opencue.wrappers.job.Job>
@return: List of jobs in this group"""
response = self.stub.GetJobs(job_pb2.GroupGetJobsRequest(group=self.data),
timeout=Cuebot.Timeout)
return [opencue.wrappers.job.Job(j) for j in response.jobs.jobs]
def testGetJobs(self, getStubMock):
jobName = 'testJob'
stubMock = mock.Mock()
stubMock.GetJobs.return_value = job_pb2.GroupGetJobsResponse(
jobs=job_pb2.JobSeq(jobs=[job_pb2.Job(name=jobName)]))
getStubMock.return_value = stubMock
group = opencue.wrappers.group.Group(
job_pb2.Group(name=TEST_GROUP_NAME))
jobs = group.getJobs()
stubMock.GetJobs.assert_called_with(
job_pb2.GroupGetJobsRequest(group=group.data), timeout=mock.ANY)
self.assertEqual(len(jobs), 1)
self.assertEqual(jobs[0].name(), jobName)
