def getService(cls, name):
try:
response = cls.stub.GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout)
except grpc.RpcError, e:
if e.code() == grpc.StatusCode.NOT_FOUND:
return None
raise e
def getService(name):
"""
Return the default service list. Services
define the default application features.
@rtype Service
"""
return Cuebot.getStub('service').GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout).service
def testGet(self, stubMock):
stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse(
service=self.service)
service = opencue.api.getService(self.testName)
stubMock.GetService.assert_called_with(
service_pb2.ServiceGetServiceRequest(name=self.testName), timeout=mock.ANY)
self.assertEqual(self.testName, service.name())
def getService(name):
try:
response = Cuebot.getStub('service').GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout)
except grpc.RpcError as e:
if e.code() == grpc.StatusCode.NOT_FOUND:
return None
raise e
return Service(response.service)
def testGetService(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse(
service=service_pb2.Service(name=TEST_SERVICE_NAME))
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.service.Service()
service = wrapper.getService(name=TEST_SERVICE_NAME)
stubMock.GetService.assert_called_with(
service_pb2.ServiceGetServiceRequest(name=TEST_SERVICE_NAME), timeout=mock.ANY)
self.assertEqual(service.name(), TEST_SERVICE_NAME)
def getService(name):
"""Returns a service by name.
:type name: str
:param name: service name to find
"""
try:
response = Cuebot.getStub('service').GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout)
except grpc.RpcError as e:
# pylint: disable=no-member
if e.code() == grpc.StatusCode.NOT_FOUND:
return None
# pylint: enable=no-member
raise e
return Service(response.service)
def testGpus(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse(
service=service_pb2.Service(name=TEST_SERVICE_NAME))
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.service.Service()
service = wrapper.getService(name=TEST_SERVICE_NAME)
self.assertEqual(service.minGpus(), 0)
self.assertEqual(service.maxGpus(), 0)
self.assertEqual(service.minGpuMemory(), 0)
service.setMinGpus(TEST_MIN_GPUS)
service.setMaxGpus(TEST_MAX_GPUS)
service.setMinGpuMemory(TEST_MIN_GPU_MEMORY)
self.assertEqual(service.minGpus(), TEST_MIN_GPUS)
self.assertEqual(service.maxGpus(), TEST_MAX_GPUS)
self.assertEqual(service.minGpuMemory(), TEST_MIN_GPU_MEMORY)
stubMock.GetService.assert_called_with(
service_pb2.ServiceGetServiceRequest(name=TEST_SERVICE_NAME),
timeout=mock.ANY)
self.assertEqual(service.name(), TEST_SERVICE_NAME)