Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[lambda] Do not use ZipFile as a context manager (it is not supported in Python 2.6)

Browse Source
This commit is contained in:
Hugo Lopes Tavares 2016-02-16 16:56:20 -05:00
parent ac66a8eb87
commit 45d91bf963
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_awslambda/__init__.py
View File

@ -13,11 +13,12 @@ from moto import mock_lambda, mock_s3
def get_test_zip_file(): def get_test_zip_file():
zip_output = io.BytesIO() zip_output = io.BytesIO()
with zipfile.ZipFile(zip_output, 'w') as f: zip_file = zipfile.ZipFile(zip_output, 'w')
f.writestr('lambda_function.py', b'''\ zip_file.writestr('lambda_function.py', b'''\
def handler(event, context): def handler(event, context):
return "hello world" return "hello world"
''') ''')
zip_file.close()
zip_output.seek(0) zip_output.seek(0)
return zip_output.read() return zip_output.read()