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)