def _ApplyEnvVarsArgsToFunction(function, args):
updated_fields = []
old_env_vars = env_vars_api_util.GetFunctionEnvVarsAsDict(function)
env_var_flags = map_util.GetMapFlagsFromArgs('env-vars', args)
new_env_vars = map_util.ApplyMapFlags(old_env_vars, **env_var_flags)
if old_env_vars != new_env_vars:
function.environmentVariables = env_vars_api_util.DictToEnvVarsProperty(
new_env_vars)
updated_fields.append('environmentVariables')
return updated_fields
def testGetFunctionEnvVarsAsDict(self):
expected = {
'FOO': 'BAR',
'BAZ': 'BOO',
}
messages = api_util.GetApiMessagesModule()
env_vars_class = messages.CloudFunction.EnvironmentVariablesValue
function = messages.CloudFunction()
function.environmentVariables = (env_vars_class(additionalProperties=[
env_vars_class.AdditionalProperty(key='BAZ', value='BOO'),
env_vars_class.AdditionalProperty(key='FOO', value='BAR'),
]))
actual = env_vars.GetFunctionEnvVarsAsDict(function)
self.assertEqual(expected, actual)
def testGetFunctionEnvVarsAsDictNone(self):
messages = api_util.GetApiMessagesModule()
function = messages.CloudFunction()
actual = env_vars.GetFunctionEnvVarsAsDict(function)
self.assertEqual({}, actual)