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 testDelete(self, stubMock): stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse( service=self.service) stubMock.Delete.return_value = service_pb2.ServiceDeleteResponse() opencue.api.getService(self.testName).delete() stubMock.Delete.assert_called_with( service_pb2.ServiceDeleteRequest(service=self.service), timeout=mock.ANY)
def testUpdate(self, stubMock): stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse( service=self.service) stubMock.Update.return_value = service_pb2.ServiceUpdateResponse() updatedService = opencue.api.getService(self.testName) updatedService.setTags(['util']) updatedService.update() stubMock.Update.assert_called_with( service_pb2.ServiceUpdateRequest(service=updatedService.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 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)