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())