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)
