def test_update_config_value_does_not_exist(self):
request = config_messages.UpdateConfigRequest(config=[
config_messages.UpdateConfig(
name='Does not exist!',
config_type=config_messages.ConfigType.BOOLEAN,
boolean_value=False)
])
self.assertRaisesRegexp(config_api.endpoints.BadRequestException,
'No such name', self.service.update_config,
request)
def _create_config_request(config_value_type, mock_config, mock_config_value):
"""Creates a config_messages UpdateConfigRequest for testing.
Args:
config_value_type: str, the type of value.
mock_config: str, the name of the config.
mock_config_value: str, bool, int, list, the config value.
Returns:
A config_messages.UpdateConfigRequest ProtoRPC message.
"""
if config_value_type == 'list_value':
config = [
config_messages.UpdateConfig(
name=mock_config,
config_type=config_messages.ConfigType.LIST,
list_value=mock_config_value)
]
elif config_value_type == 'integer_value':
config = [
config_messages.UpdateConfig(
name=mock_config,
config_type=config_messages.ConfigType.INTEGER,
integer_value=mock_config_value)
]
elif config_value_type == 'string_value':
config = [
config_messages.UpdateConfig(
name=mock_config,
config_type=config_messages.ConfigType.STRING,
string_value=mock_config_value)
]
elif config_value_type == 'boolean_value':
config = [
config_messages.UpdateConfig(
name=mock_config,
config_type=config_messages.ConfigType.BOOLEAN,
boolean_value=mock_config_value)
]
return config_messages.UpdateConfigRequest(config=config)
def testUpdateConfig(self):
config = config_messages.UpdateConfig(
name='test',
config_type=config_messages.ConfigType(1),
integer_value=1,
boolean_value=False,
list_value=['test'])
self.assertEqual(config.name, 'test')
self.assertEqual(config.config_type.name, 'STRING')
self.assertEqual(config.integer_value, 1)
self.assertEqual(config.boolean_value, False)
self.assertEqual(config.list_value, ['test'])
def test_update_config_multi(self, mock_load_config_from_yaml):
new_string_value = 'new_string_value'
new_int_value = 9
mock_load_config_from_yaml.return_value = {
'test_config_integer': 2,
'test_config_string': 'old_string_value',
}
request = config_messages.UpdateConfigRequest(
config=[
config_messages.UpdateConfig(
name='test_config_integer',
config_type=config_messages.ConfigType.INTEGER,
integer_value=new_int_value),
config_messages.UpdateConfig(
name='test_config_string',
config_type=config_messages.ConfigType.STRING,
string_value=new_string_value)
])
self.service.update_config(request)
updated_string_value = config_model.Config.get(name='test_config_string')
self.assertEqual(updated_string_value, new_string_value)
updated_int_value = config_model.Config.get(name='test_config_integer')
self.assertEqual(updated_int_value, new_int_value)
def testUpdateConfigRequest(self):
config = config_messages.UpdateConfigRequest(
config=[config_messages.UpdateConfig(name='test')])
self.assertEqual(config.config[0].name, 'test')