def testGetDefault(self, stubMock):
service1 = 'service1'
service2 = 'service2'
stubMock.GetDefaultServices.return_value = service_pb2.ServiceGetDefaultServicesResponse(
services=service_pb2.ServiceSeq(
services=[service_pb2.Service(name=service1), service_pb2.Service(name=service2)]))
services = opencue.api.getDefaultServices()
stubMock.GetDefaultServices.assert_called_with(
service_pb2.ServiceGetDefaultServicesRequest(), timeout=mock.ANY)
self.assertEqual([service1, service2], [service.name() for service in services])
def testCreateService(self, getStubMock):
stubMock = mock.Mock()
stubMock.CreateService.return_value = service_pb2.ServiceCreateServiceResponse(
service=service_pb2.Service(name=TEST_SERVICE_NAME))
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.service.Service(
service_pb2.Service(name=TEST_SERVICE_NAME))
service = wrapper.create()
stubMock.CreateService.assert_called_with(
service_pb2.ServiceCreateServiceRequest(data=wrapper.data), timeout=mock.ANY)
self.assertEqual(wrapper.name(), service.name())
def testUpdate(self, getStubMock):
stubMock = mock.Mock()
stubMock.Update.return_value = service_pb2.ServiceUpdateResponse()
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.service.Service(service=service_pb2.Service(
name=TEST_SERVICE_NAME))
wrapper.update()
stubMock.Update.assert_called_with(
service_pb2.ServiceUpdateRequest(service=wrapper.data), timeout=mock.ANY)
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 testGetDefaultServices(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetDefaultServices.return_value = service_pb2.ServiceGetDefaultServicesResponse(
services=service_pb2.ServiceSeq(
services=[service_pb2.Service(name=TEST_SERVICE_NAME)]))
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.service.Service()
services = wrapper.getDefaultServices()
stubMock.GetDefaultServices.assert_called_with(
service_pb2.ServiceGetDefaultServicesRequest(), timeout=mock.ANY)
self.assertEqual(len(services), 1)
self.assertEqual(services[0].name(), TEST_SERVICE_NAME)
def testGetServiceOverrides(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetServiceOverrides.return_value = show_pb2.ShowGetServiceOverridesResponse(
service_overrides=service_pb2.ServiceOverrideSeq(
service_overrides=[
service_pb2.ServiceOverride(data=service_pb2.Service(name=TEST_SERVICE_NAME))
]))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
overrides = show.getServiceOverrides()
stubMock.GetServiceOverrides.assert_called_with(
show_pb2.ShowGetServiceOverridesRequest(show=show.data),
timeout=mock.ANY)
self.assertEqual(overrides[0].data.name, TEST_SERVICE_NAME)
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)
def setUp(self):
self.service = service_pb2.Service(
name=self.testName, threadable=self.testThreadable, min_cores=self.testMinCores,
max_cores=self.testMaxCores, tags=self.testTags)
def __init__(self, service=None):
self.data = service or service_pb2.Service()
self.stub = Cuebot.getStub('service')
def __init__(self, service=None):
self.data = service or service_pb2.Service()
