def testByOptions(self, getStubMock): jobId = 'A0000000-0000-0000-0000-000000000000' stubMock = mock.Mock() getStubMock.return_value = stubMock opencue.search.JobSearch.byOptions(show=['pipe'], match=['v6']) stubMock.GetJobs.assert_called_with(job_pb2.JobGetJobsRequest( r=job_pb2.JobSearchCriteria(shows=['pipe'], substr=['v6'])), timeout=mock.ANY) opencue.search.JobSearch.byOptions(id=[jobId]) stubMock.GetJobs.assert_called_with(job_pb2.JobGetJobsRequest( r=job_pb2.JobSearchCriteria(ids=[jobId])), timeout=mock.ANY)
def testBaseSearchJob(self, getStubMock): stubMock = mock.Mock() getStubMock.return_value = stubMock jobSearch = opencue.search.JobSearch(show=['pipe'], match=['v6']) jobSearch.search() stubMock.GetJobs.assert_called_with(job_pb2.JobGetJobsRequest( r=job_pb2.JobSearchCriteria(shows=['pipe'], substr=['v6'])), timeout=mock.ANY)
def testGetJobNames(self, getStubMock): stubMock = mock.Mock() stubMock.GetJobNames.return_value = job_pb2.JobGetJobNamesResponse( names=[TEST_JOB_NAME]) getStubMock.return_value = stubMock jobNames = opencue.api.getJobNames(show=[TEST_SHOW_NAME]) stubMock.GetJobNames.assert_called_with( job_pb2.JobGetJobNamesRequest( r=job_pb2.JobSearchCriteria(shows=[TEST_SHOW_NAME])), timeout=mock.ANY) self.assertEqual(1, len(jobNames)) self.assertEqual(TEST_JOB_NAME, jobNames[0])
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())
def criteriaFromOptions(**options): return _setOptions(job_pb2.JobSearchCriteria(), options)
def criteriaFromOptions(**options): """Constructs a search criteria object for the given options.""" return _setOptions(job_pb2.JobSearchCriteria(), options)