def testGetProcs(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetProcs.return_value = host_pb2.ProcGetProcsResponse(
procs=host_pb2.ProcSeq(procs=[host_pb2.Proc(name=TEST_PROC_NAME)]))
getStubMock.return_value = stubMock
procs = opencue.api.getProcs(show=[TEST_SHOW_NAME], alloc=[TEST_ALLOC_NAME])
stubMock.GetProcs.assert_called_with(
host_pb2.ProcGetProcsRequest(
r=host_pb2.ProcSearchCriteria(shows=[TEST_SHOW_NAME], allocs=[TEST_ALLOC_NAME])),
timeout=mock.ANY)
self.assertEqual([TEST_PROC_NAME], [proc.name() for proc in procs])
def criteriaFromOptions(**options):
return _setOptions(host_pb2.ProcSearchCriteria(), options)
def criteriaFromOptions(**options):
"""Constructs a search criteria object for the given options."""
return _setOptions(host_pb2.ProcSearchCriteria(), options)
