fix tests for python 3.x

2017-09-15 15:07:02 +12:00 · 2017-09-15 15:07:02 +12:00 · 101dfaa412
commit 101dfaa412
parent 2c8326e9b9
2 changed files with 2 additions and 2 deletions
--- a/moto/awslambda/models.py
+++ b/moto/awslambda/models.py
@ -111,7 +111,7 @@ class LambdaFunction(BaseModel):
        }

    def get_code(self):
-        if 'S3Key' in self.code:
+        if isinstance(self.code, dict):
            return {
                "Code": {
                    "Location": "s3://lambda-functions.aws.amazon.com/{0}".format(self.code['S3Key']),
--- a/tests/test_awslambda/test_lambda.py
+++ b/tests/test_awslambda/test_lambda.py
@ -624,4 +624,4 @@ def test_get_function_created_with_zipfile():
                "SubnetIds": [],
            }
        },
-    )
+    )