def set_num_instances_async(instances, module=None, version=None): """Returns a UserRPC to set the number of instances on the module version. Args: instances: The number of instances to set. module: The module to set the number of instances for, if None the current module will be used. version: The version set the number of instances for, if None the current version will be used. Returns: A UserRPC to set the number of instances on the module version. """ def _ResultHook(rpc): mapped_errors = [ modules_service_pb2.ModulesServiceError.INVALID_VERSION, modules_service_pb2.ModulesServiceError.TRANSIENT_ERROR ] _CheckAsyncResult(rpc, mapped_errors, {}) if not isinstance(instances, six.integer_types): raise TypeError("'instances' arg must be of type long or int.") request = modules_service_pb2.SetNumInstancesRequest() request.instances = instances if module: request.module = module if version: request.version = version response = modules_service_pb2.SetNumInstancesResponse() return _MakeAsyncCall('SetNumInstances', request, response, _ResultHook)
def testSetNumInstances_TransientError(self): """Test we raise an error when we receive on from the underlying API.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.instances = 23 self.SetExceptionExpectations( 'SetNumInstances', expected_request, modules_service_pb2.ModulesServiceError.TRANSIENT_ERROR) self.assertRaises(modules.TransientError, modules.set_num_instances, 23)
def testSetNumInstances_InvalidVersionError(self): """Test we raise an error when we receive on from the underlying API.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.instances = 23 self.SetExceptionExpectations( 'SetNumInstances', expected_request, modules_service_pb2.ModulesServiceError.INVALID_VERSION) self.assertRaises(modules.InvalidVersionError, modules.set_num_instances, 23)
def testSetNumInstances_AllDefaults(self): """Test we return the expected results when no args are passed.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.instances = 15 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(15)
def testSetNumInstances_NoModule(self): """Test we return the expected results when no module is passed.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.version = 'v1' expected_request.instances = 14 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(14, version='v1')
def testSetNumInstances_NoVersion(self): """Test we return the expected results when no version is passed.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.module = 'module1' expected_request.instances = 13 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(13, 'module1')
def testSetNumInstances(self): """Test we return the expected results.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.module = 'module1' expected_request.version = 'v1' expected_request.instances = 12 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(12, 'module1', 'v1')