fix tests for python 3.x

This commit is contained in:
Chris Keogh 2017-09-15 15:07:02 +12:00
parent 2c8326e9b9
commit 101dfaa412
2 changed files with 2 additions and 2 deletions

View File

@ -111,7 +111,7 @@ class LambdaFunction(BaseModel):
} }
def get_code(self): def get_code(self):
if 'S3Key' in self.code: if isinstance(self.code, dict):
return { return {
"Code": { "Code": {
"Location": "s3://lambda-functions.aws.amazon.com/{0}".format(self.code['S3Key']), "Location": "s3://lambda-functions.aws.amazon.com/{0}".format(self.code['S3Key']),

View File

@ -624,4 +624,4 @@ def test_get_function_created_with_zipfile():
"SubnetIds": [], "SubnetIds": [],
} }
}, },
) )