def reparentJobs(self, jobs):
"""Moves the given jobs into this group
@type jobs: list<Job>
@param jobs: The jobs to add to this group"""
jobSeq = job_pb2.JobSeq(jobs=[job.data for job in jobs])
self.stub.ReparentJobs(job_pb2.GroupReparentJobsRequest(
group=self.data, jobs=jobSeq),
timeout=Cuebot.Timeout)
def runFilterOnJobs(self, jobs):
"""Runs the filter on the list of jobs provided
@type jobs: list<JobInterfacePrx or Job or id or str jobname>
@param jobs: The jobs to add to this group"""
jobSeq = job_pb2.JobSeq(jobs=jobs)
self.stub.RunFilterOnJobs(filter_pb2.FilterRunFilterOnJobsRequest(
filter=self.data, jobs=jobSeq),
timeout=Cuebot.Timeout)
def runFilterOnJobs(self, jobs):
"""Runs the filter on the list of jobs provided
@type jobs: list<opencue.wrapper.job.Job>
@param jobs: The jobs to run the filter on"""
jobSeq = job_pb2.JobSeq(jobs=[job.data for job in jobs])
self.stub.RunFilterOnJobs(filter_pb2.FilterRunFilterOnJobsRequest(
filter=self.data, jobs=jobSeq),
timeout=Cuebot.Timeout)
def testLaunchSpecAndWait(self, getStubMock):
spec = 'arbitrary-spec'
stubMock = mock.Mock()
stubMock.LaunchSpecAndWait.return_value = job_pb2.JobLaunchSpecAndWaitResponse(
jobs=job_pb2.JobSeq(jobs=[job_pb2.Job(name=TEST_JOB_NAME)]))
getStubMock.return_value = stubMock
jobs = opencue.api.launchSpecAndWait(spec)
stubMock.LaunchSpecAndWait.assert_called_with(
job_pb2.JobLaunchSpecAndWaitRequest(spec=spec), timeout=mock.ANY)
self.assertEqual([TEST_JOB_NAME], [job.name() for job in jobs])
def reparentJobs(self, jobs):
"""Moves the given jobs into this group
@type jobs: list<opencue.wrappers.job.Job>
@param jobs: The jobs to add to this group"""
jobsToReparent = []
for job in jobs:
if isinstance(job, opencue.wrappers.job.NestedJob):
job = job.asJob()
jobsToReparent.append(job.data)
jobSeq = job_pb2.JobSeq(jobs=jobsToReparent)
self.stub.ReparentJobs(job_pb2.GroupReparentJobsRequest(group=self.data, jobs=jobSeq),
timeout=Cuebot.Timeout)
def testRunFilterOnJobs(self, getStubMock):
stubMock = mock.Mock()
stubMock.RunFilterOnJobs.return_value = filter_pb2.FilterRunFilterOnJobsResponse()
getStubMock.return_value = stubMock
jobs = [opencue.wrappers.job.Job(job_pb2.Job(name='testJob'))]
jobSeq = job_pb2.JobSeq(jobs=[job.data for job in jobs])
filterToRun = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToRun.runFilterOnJobs(jobs)
stubMock.RunFilterOnJobs.assert_called_with(
filter_pb2.FilterRunFilterOnJobsRequest(filter=filterToRun.data, jobs=jobSeq),
timeout=mock.ANY)
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)
def testReparentJobs(self, getStubMock):
stubMock = mock.Mock()
stubMock.ReparentJobs.return_value = job_pb2.GroupReparentJobsResponse()
getStubMock.return_value = stubMock
testJob = job_pb2.Job(name='testJob')
testNestedJob = job_pb2.NestedJob(name='testNestedJob')
jobs = [opencue.wrappers.job.Job(testJob), opencue.wrappers.job.NestedJob(testNestedJob)]
group = opencue.wrappers.group.Group(
job_pb2.Group(name=TEST_GROUP_NAME))
group.reparentJobs(jobs)
expected = job_pb2.JobSeq(jobs=[job_pb2.Job(name='testJob'),
job_pb2.Job(name='testNestedJob',
job_stats=job_pb2.JobStats())])
stubMock.ReparentJobs.assert_called_with(
job_pb2.GroupReparentJobsRequest(group=group.data, jobs=expected),
timeout=mock.ANY)
def testGetJobs(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetJobs.return_value = job_pb2.JobGetJobsResponse(
jobs=job_pb2.JobSeq(jobs=[job_pb2.Job(name=TEST_JOB_NAME)]))
getStubMock.return_value = stubMock
jobsByShow = opencue.api.getJobs(show=[TEST_SHOW_NAME], all=True)
stubMock.GetJobs.assert_called_with(
job_pb2.JobGetJobsRequest(
r=job_pb2.JobSearchCriteria(shows=[TEST_SHOW_NAME])), timeout=mock.ANY)
self.assertEqual(1, len(jobsByShow))
self.assertEqual(TEST_JOB_NAME, jobsByShow[0].name())
jobsByName = opencue.api.getJobs(name=[TEST_JOB_NAME], show=[TEST_SHOW_NAME])
stubMock.GetJobs.assert_called_with(
job_pb2.JobGetJobsRequest(
r=job_pb2.JobSearchCriteria(
jobs=[TEST_JOB_NAME], shows=[TEST_SHOW_NAME])), timeout=mock.ANY)
self.assertEqual(1, len(jobsByName))
self.assertEqual(TEST_JOB_NAME, jobsByName[0].name())
