catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Allow lambda get function by arn and addded environment configuration

Browse source
This commit is contained in:
Ian Yon 2019-11-04 10:44:01 -03:00
commit 90a9235f4c
3 changed files with 39 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

16
moto/awslambda/models.py
View file

@ -291,6 +291,8 @@ class LambdaFunction(BaseModel):
self.timeout = value
elif key == "VpcConfig":
self.vpc_config = value
elif key == "Environment":
self.environment_vars = value["Variables"]
return self.get_configuration()
@ -589,7 +591,7 @@ class LambdaStorage(object):
def _get_alias(self, name, alias):
return self._functions[name]['alias'].get(alias, None)
def get_function(self, name, qualifier=None):
def get_function_by_name(self, name, qualifier=None):
if name not in self._functions:
return None
@ -612,8 +614,8 @@ class LambdaStorage(object):
def get_arn(self, arn):
return self._arns.get(arn, None)
def get_function_by_name_or_arn(self, input):
return self.get_function(input) or self.get_arn(input)
def get_function_by_name_or_arn(self, input, qualifier=None):
return self.get_function_by_name(input, qualifier) or self.get_arn(input)
def put_function(self, fn):
"""
@ -671,7 +673,7 @@ class LambdaStorage(object):
return True
else:
fn = self.get_function(name, qualifier)
fn = self.get_function_by_name(name, qualifier)
if fn:
self._functions[name]['versions'].remove(fn)
@ -766,8 +768,8 @@ class LambdaBackend(BaseBackend):
def publish_function(self, function_name):
return self._lambdas.publish_function(function_name)
def get_function(self, function_name, qualifier=None):
return self._lambdas.get_function(function_name, qualifier)
def get_function(self, function_name_or_arn, qualifier=None):
return self._lambdas.get_function_by_name_or_arn(function_name_or_arn, qualifier)
def list_versions_by_function(self, function_name):
return self._lambdas.list_versions_by_function(function_name)
@ -877,7 +879,7 @@ class LambdaBackend(BaseBackend):
]
}
func = self._lambdas.get_function(function_name, qualifier)
func = self._lambdas.get_function_by_name_or_arn(function_name, qualifier)
func.invoke(json.dumps(event), {}, {})
def send_dynamodb_items(self, function_arn, items, source):