fix: lambda tests and backwards compatibility

foremast · Oct 24, 2021 · 4d4a49e · 4d4a49e
1 parent bb4ab47
commit 4d4a49e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/foremast/awslambda/awslambda.py b/src/foremast/awslambda/awslambda.py
@@ -40,6 +40,7 @@ def __init__(self, app, env, region, prop_path, artifact_path):
             env (str): Environment/Account
             region (str): AWS Region
             prop_path (str): Path of environment property file
+            artifact_path (str): Path or URI for code artifact
         """
         self.app_name = app
         self.env = env
@@ -58,7 +59,7 @@ def __init__(self, app, env, region, prop_path, artifact_path):
         self.description = self.pipeline['app_description']
         self.handler = self.pipeline['handler']
         self.vpc_enabled = self.pipeline['vpc_enabled']
-        self.package_type = self.pipeline["package_type"]
+        self.package_type = "zip" if 'package_type' not in self.pipeline else self.pipeline['package_type']
 
         self.settings = get_properties(prop_path, env=self.env, region=self.region)
         app = self.settings['app']

diff --git a/tests/lambda/test_aws.py b/tests/lambda/test_aws.py
@@ -10,7 +10,7 @@
             'app_description': None,
             'handler': None,
             'runtime': None,
-            'vpc_enabled': None,
+            'vpc_enabled': None
         },
     },
     'app': {