class Temp_Lambda:
def __init__(self):
self.lambda_name = "temp_lambda_{0}".format(
Misc.random_string_and_numbers())
self.aws_lambda = Lambda(self.lambda_name)
self.tmp_folder = Temp_Folder_Code(self.lambda_name)
self.role_arn = Temp_Aws_Roles().for_lambda_invocation()
self.create_log = None
self.delete_on_exit = True
self.s3_bucket = Globals.lambda_s3_bucket
self.s3_key = 'unit_tests/lambdas/{0}.zip'.format(self.lambda_name)
self.s3 = self.aws_lambda.s3()
def __enter__(self):
(self.aws_lambda.set_role(self.role_arn).set_s3_bucket(
self.s3_bucket).set_s3_key(self.s3_key).set_folder_code(
self.tmp_folder.folder))
self.aws_lambda.upload()
self.create_log = self.aws_lambda.create()
assert self.aws_lambda.exists() == True
return self
def __exit__(self, type, value, traceback):
if self.delete_on_exit:
assert self.aws_lambda.delete() is True
self.delete_s3_file()
def arn(self):
return self.aws_lambda.function_Arn()
def info(self):
return self.aws_lambda.info()
def invoke(self, params=None):
return self.aws_lambda.invoke(params)
def invoke_raw(self, params=None):
return self.aws_lambda.invoke_raw(params)
def set_s3_bucket(self, value):
self.s3_bucket = value
def s3_file_exists(self):
return self.s3.file_exists(self.s3_bucket, self.s3_key)
def delete_s3_file(self):
self.s3.file_delete(self.s3_bucket, self.s3_key)
assert self.s3_file_exists() is False
return self
class test_Lambda(Test_Helper):
@aws_inject('region,account_id')
def setUp(self, region, account_id):
super().setUp()
self.lambda_name = 'tmp_lambda_dev_test'
self.setup = super().setUp()
self.s3_bucket = self.setup.s3_bucket_lambdas
self.s3_key = f'{Globals.lambda_s3_key_prefix}/{self.lambda_name}.zip' #'lambdas/{0}.zip'.format(self.lambda_name)
self.aws_lambda = Lambda(self.lambda_name)
def test__init__(self):
assert self.aws_lambda.runtime == 'python3.7'
assert self.aws_lambda.memory == 3008
assert self.aws_lambda.timeout == 60
assert self.aws_lambda.original_name == self.lambda_name
assert self.aws_lambda.s3().bucket_exists(self.s3_bucket)
def test_account_settings(self):
assert self.aws_lambda.account_settings().get('AccountLimit').get('CodeSizeUnzipped') == 262144000
def test_aliases(self):
lambda_api = self.aws_lambda # make it easier to read below
alias_name = 'temp_name'
alias_description = 'temp description'
function_version = '$LATEST'
with Temp_Lambda() as temp_lambda:
assert lambda_api.aliases (function_name = temp_lambda.arn() ) == []
assert lambda_api.alias_create(function_name = temp_lambda.lambda_name,
function_version = function_version ,
name = alias_name ,
description = alias_description ).get('Name') == alias_name
data = lambda_api.alias (function_name = temp_lambda.lambda_name,
name = alias_name )
del data['RevisionId']
assert data == { 'AliasArn' : f'{temp_lambda.arn()}:{alias_name}',
'Description' : alias_description ,
'FunctionVersion': function_version ,
'Name' : alias_name }
assert lambda_api.aliases(function_name= temp_lambda.arn(),index_by='Name').get(alias_name).get('Description') == alias_description
assert lambda_api.alias_delete(temp_lambda.lambda_name, alias_name) == {}
assert lambda_api.aliases(function_name=temp_lambda.arn()) == []
def test_create_function__no_params(self):
assert self.aws_lambda.create() == {'error': "missing fields in create_function: ['role', 's3_bucket', 's3_key']"}
def test_create_function(self):
role_arn = IAM_Role(self.lambda_name + '__tmp_role').create_for__lambda().get('role_arn')
tmp_folder = Temp_Folder_Code(self.lambda_name)
(
self.aws_lambda.set_role (role_arn)
.set_s3_bucket (self.s3_bucket )
.set_s3_key (self.s3_key )
.set_folder_code(tmp_folder.folder)
)
assert self.aws_lambda.create_params() == (self.lambda_name, 'python3.7' ,
role_arn , self.lambda_name + '.run',
3008 , 60 ,
{ 'Mode' : 'PassThrough' },
{ 'S3Bucket': self.s3_bucket ,
'S3Key' : self.s3_key }) # confirm values that will be passed to the boto3's create_function
assert self.aws_lambda.upload() is True
result = self.aws_lambda.create()
data = result.get('data')
name = result.get('name')
status = result.get('status')
assert status == 'ok'
assert name == self.lambda_name
expected_arn = 'arn:aws:lambda:{0}:{1}:function:{2}'.format(self.region,self.account_id,self.lambda_name) # expected arn value
(Assert(data).field_is_equal('CodeSize' , 209 ) # confirm lambda creation details
.field_is_equal('FunctionArn' , expected_arn )
.field_is_equal('FunctionName' , self.lambda_name )
.field_is_equal('Handler' , self.lambda_name + '.run')
.field_is_equal('MemorySize' , 3008 )
.field_is_equal('Role' , role_arn )
.field_is_equal('Runtime' , 'python3.7' )
.field_is_equal('Timeout' , 60 )
.field_is_equal('TracingConfig', {'Mode': 'PassThrough'} )
.field_is_equal('Version' , '$LATEST' )
)
assert self.aws_lambda.delete() is True # confirm Lambda was deleted
def test_create_function__bad_s3_key_(self):
self.aws_lambda.set_role('').set_s3_bucket('').set_s3_key('')
assert self.aws_lambda.create() == { 'data' : 'could not find provided s3 bucket and s3 key',
'name' : 'tmp_lambda_dev_test' ,
'status': 'error' }
def test_configuration(self):
with Temp_Lambda() as temp_lambda:
aws_lambda = temp_lambda.aws_lambda
value = 10
assert aws_lambda.configuration() == aws_lambda.info().get('Configuration')
assert aws_lambda.configuration_update(Timeout=value).get('Timeout') == value
assert aws_lambda.configuration().get('Timeout') == value
def test_event_sources(self):
#assert self.aws_lambda.event_sources() == []
self.result = self.aws_lambda.event_sources()
def test_event_source_create(self):
with Temp_Lambda() as temp_lambda: # create a temp lambda function that will deleted on exit
with Temp_Queue() as temp_queue: # create a temp sqs queue that will be deleted on exit
event_source_arn = temp_queue .queue.arn()
lambda_name = temp_lambda.lambda_name
aws_lambda = temp_lambda.aws_lambda
queue = temp_queue .queue
aws_lambda.configuration_update(Timeout=10) # needs to be less than 30 (which is the default value sent in the queue)
queue.policy_add_sqs_permissions_to_lambda_role(lambda_name) # add permissions needed to role
sleep(2) # todo: need a better solution to find out when the permission is available
aws_lambda.event_source_create(event_source_arn, lambda_name) # todo: add asserts
queue.policy_remove_sqs_permissions_to_lambda_role(lambda_name) # remove permissions todo: create custom role
def test_event_source_delete(self):
for uuid in self.aws_lambda.event_sources(index_by='UUID'):
self.result = self.aws_lambda.event_source_delete(uuid) # these take a bit of time to delete
def test_invoke_raw(self):
with Temp_Lambda() as temp_lambda:
result = temp_lambda.aws_lambda.invoke_raw({'name' : temp_lambda.lambda_name})
data = result.get('data')
name = result.get('name')
status = result.get('status')
response = result.get('response')
assert status == 'ok'
assert name == temp_lambda.lambda_name
assert data == 'hello {0}'.format(temp_lambda.lambda_name)
assert response.get('StatusCode') == 200
def test_invoke(self):
with Temp_Lambda() as temp_lambda:
result = temp_lambda.aws_lambda.invoke({'name': temp_lambda.lambda_name})
assert result == 'hello {0}'.format(temp_lambda.lambda_name)
def test_invoke_async(self):
with Temp_Lambda() as temp_lambda:
result = temp_lambda.aws_lambda.invoke_async({'name': temp_lambda.lambda_name})
assert result.get('StatusCode') == 202
def test_functions(self):
functions = list(self.aws_lambda.functions())
assert len(functions) > 0
assert set(functions.pop(0)) == {'CodeSha256', 'CodeSize' , 'Description' , 'FunctionArn', 'FunctionName',
'Handler' , 'LastModified' , 'MemorySize' , 'RevisionId' , 'Role' ,
'Runtime' , 'Timeout' , 'TracingConfig', 'Version' }
def test_layer_create(self):
#zip_bytes = Temp_Folder_Code('file_in_layer').zip_bytes()
layer_name = 'test_simple_layer'
#description = 'this is a test layer'
#compatible_runtimes = ['python3.8']
#self.result = self.aws_lambda.layer_create(layer_name, description, compatible_runtimes, zip_bytes)
layer = 'arn:aws:lambda:eu-west-1:311800962295:layer:test_simple_layer'
#with Temp_Lambda() as _:
# _.delete_on_exit= False
# self.result = _.lambda_name
lambda_name ='temp_lambda_YRWKWQ'
temp_lambda = Lambda(lambda_name)
temp_lambda.configuration_update(Layers=[f'{layer}:1'])
#self.result = temp_lambda.info()
self.result = temp_lambda.invoke({})
#self.folder.delete()
def test_layers(self):
self.result = self.aws_lambda.layers()
def test_policy__permissions(self):
with Temp_Lambda() as temp_lambda:
temp_lambda.aws_lambda.permission_add(temp_lambda.arn(), 'an_id', 'lambda:action', 'lambda.amazonaws.com')
assert temp_lambda.aws_lambda.permissions()[0].get('Action') == 'lambda:action'
def test_update(self):
with Temp_Lambda() as temp_lambda:
tmp_text = Misc.random_string_and_numbers(prefix='updated code ')
temp_lambda.tmp_folder.create_temp_file('def run(event, context): return "{0}"'.format(tmp_text))
result = temp_lambda.aws_lambda.update()
status = result.get('status')
data = result.get('data')
assert 'TracingConfig' in data #set(data) == { 'CodeSha256', 'CodeSize', 'Description', 'FunctionArn', 'FunctionName', 'Handler', 'LastModified', 'MemorySize', 'ResponseMetadata', 'RevisionId', 'Role', 'Runtime', 'Timeout', 'TracingConfig', 'Version'}
assert status == 'ok'
assert temp_lambda.aws_lambda.invoke() == tmp_text
def test_upload(self):
tmp_folder = Temp_Folder_Code(self.lambda_name)
(self.aws_lambda.set_s3_bucket (self.s3_bucket )
.set_s3_key (self.s3_key )
.set_folder_code(tmp_folder.folder ))
#self.aws_lambda.upload()
#assert tmp_folder.s3_file_exists() is True
downloaded_file = self.aws_lambda.s3().file_download(self.s3_bucket, self.s3_key) # download file uploaded
assert Files.exists(downloaded_file)
unzip_location = tmp_folder.folder + '_unzipped'
Files.unzip_file(downloaded_file,unzip_location) # unzip it
assert Files.contents(Files.find(unzip_location+'/*').pop()) == tmp_folder.lambda_code # confirm unzipped file's contents
self.aws_lambda.s3().file_delete(self.s3_bucket, self.s3_key)
class Temp_Lambda:
def __init__(self, lambda_name=None, delete_on_exit=True):
self.lambda_name = lambda_name or "temp_lambda_{0}".format(random_string_and_numbers())
self.aws_lambda = Lambda(self.lambda_name)
self.tmp_folder = Temp_Folder_With_Lambda_File(self.lambda_name).create_temp_file()
self.role_arn = Temp_Aws_Roles().for_lambda_invocation__role_arn() # todo: refactor to have option to create the role programatically (needs feature to wait for role to be available)
self.create_log = None
self.delete_on_exit = delete_on_exit
self.s3_bucket = AWS_Config().lambda_s3_bucket()
self.s3_key = 'unit_tests/lambdas/{0}.zip'.format(self.lambda_name)
self.s3 = self.aws_lambda.s3()
def __enter__(self):
self.create()
return self
def __exit__(self, type, value, traceback):
if self.delete_on_exit:
self.delete()
def arn(self):
return self.aws_lambda.function_Arn()
def create(self):
(self.aws_lambda.set_role (self.role_arn)
.set_s3_bucket (self.s3_bucket )
.set_s3_key (self.s3_key )
.set_folder_code(self.tmp_folder.folder )
.upload())
self.create_log = self.aws_lambda.create()
assert self.exists() is True
return self.create_log
def delete(self):
assert self.aws_lambda.delete()
self.delete_s3_file()
def exists(self):
return self.aws_lambda.exists()
def info(self):
return self.aws_lambda.info()
def invoke(self, params=None):
return self.aws_lambda.invoke(params)
def invoke_return_logs(self, params=None):
return self.aws_lambda.invoke_return_logs(params)
def invoke_raw(self, params=None):
return self.aws_lambda.invoke_raw(params)
def set_s3_bucket(self, value):
self.s3_bucket = value
def s3_file_exists(self):
return self.s3.file_exists(self.s3_bucket, self.s3_key)
def delete_s3_file(self):
self.s3.file_delete(self.s3_bucket, self.s3_key)
assert self.s3_file_exists() is False
class test_Lambda(Test_Helper):
@classmethod
def setUpClass(cls) -> None:
#STS().check_current_session_credentials()
cls.s3 = S3()
cls.aws_config = AWS_Config()
cls.account_id = cls.aws_config.aws_session_account_id()
cls.s3_bucket = cls.aws_config.lambda_s3_bucket()
cls.region = cls.aws_config.aws_session_region_name()
# if cls.s3.bucket_not_exists(cls.s3_bucket): # todo: find what this adds 3 seconds to the test
# cls.s3.bucket_create(bucket=cls.s3_bucket, region=cls.region)
# print('>>>> Created bucket', cls.s3_bucket)
def setUp(self):
super().setUp()
self.lambda_name = 'temp_lambda_dev_test' # todo: see if this is really neded here
self.s3_key = f'{AWS_Config().lambda_s3_folder_lambdas()}/{self.lambda_name}.zip'
self.aws_lambda = Lambda(self.lambda_name)
def test__init__(self):
assert self.aws_lambda.runtime == 'python3.8'
assert self.aws_lambda.memory == 10240
assert self.aws_lambda.timeout == 900
assert self.aws_lambda.original_name == self.lambda_name
assert self.aws_lambda.s3().bucket_exists(self.s3_bucket)
def test_account_settings(self):
assert self.aws_lambda.account_settings().get('AccountLimit').get(
'CodeSizeUnzipped') == 262144000
def test_aliases(self):
lambda_api = self.aws_lambda # make it easier to read below
alias_name = 'temp_name'
alias_description = 'temp description'
function_version = '$LATEST'
with Temp_Lambda() as temp_lambda:
assert lambda_api.aliases(function_name=temp_lambda.arn()) == []
assert lambda_api.alias_create(
function_name=temp_lambda.lambda_name,
function_version=function_version,
name=alias_name,
description=alias_description).get('Name') == alias_name
data = lambda_api.alias(function_name=temp_lambda.lambda_name,
name=alias_name)
del data['RevisionId']
assert data == {
'AliasArn': f'{temp_lambda.arn()}:{alias_name}',
'Description': alias_description,
'FunctionVersion': function_version,
'Name': alias_name
}
assert lambda_api.aliases(function_name=temp_lambda.arn(),
index_by='Name').get(alias_name).get(
'Description') == alias_description
assert lambda_api.alias_delete(temp_lambda.lambda_name,
alias_name) == {}
assert lambda_api.aliases(function_name=temp_lambda.arn()) == []
def test_create_function(self):
role_arn = IAM_Role(self.lambda_name +
'__tmp_role').create_for__lambda().get('role_arn')
tmp_folder = Temp_Folder_With_Lambda_File(
self.lambda_name).create_temp_file()
(self.aws_lambda.set_role(role_arn).set_s3_bucket(
self.s3_bucket).set_s3_key(self.s3_key).set_folder_code(
tmp_folder.folder))
create_kwargs = self.aws_lambda.create_kwargs()
assert create_kwargs == {
'Code': {
'S3Bucket': self.s3_bucket,
'S3Key': self.s3_key
},
'FunctionName': self.lambda_name,
'Handler': self.lambda_name + '.run',
'MemorySize': 10240,
'PackageType': 'Zip',
'Role': role_arn,
'Runtime': 'python3.8',
'Tags': {},
'Timeout': 900,
'TracingConfig': {
'Mode': 'PassThrough'
}
}
assert self.aws_lambda.upload() is True
result = self.aws_lambda.create()
data = result.get('data')
name = result.get('name')
status = result.get('status')
assert status == 'ok'
assert name == self.lambda_name
expected_arn = 'arn:aws:lambda:{0}:{1}:function:{2}'.format(
self.region, self.account_id,
self.lambda_name) # expected arn value
(Assert(data).field_is_equal('CodeSize',
242) # confirm lambda creation details
.field_is_equal('FunctionArn', expected_arn).field_is_equal(
'FunctionName', self.lambda_name).field_is_equal(
'Handler', self.lambda_name + '.run').field_is_equal(
'MemorySize',
10240).field_is_equal('Role', role_arn).field_is_equal(
'Runtime', 'python3.8').field_is_equal(
'Timeout',
900).field_is_equal('TracingConfig', {
'Mode': 'PassThrough'
}).field_is_equal('Version', '$LATEST'))
assert self.aws_lambda.invoke() == 'hello None'
assert self.aws_lambda.delete() is True # confirm Lambda was deleted
# def test_create_function_using_image_container(self):
# pass
# # self.image_uri = "785217600689.dkr.ecr.eu-west-1.amazonaws.com/python-for-lambda:latest"
# # todo write test that checks the ability to use images for deployment
def test_create_function__bad_s3_key_(self):
self.aws_lambda.set_role('').set_s3_bucket('').set_s3_key('')
assert self.aws_lambda.create() == {
'data': None,
'error': None,
'message':
'for function temp_lambda_dev_test, could not find provided s3 bucket and s3 key: ',
'status': 'error'
}
def test_create_function__no_params(self):
assert self.aws_lambda.create() == {
'data': None,
'error': None,
'message':
'for function temp_lambda_dev_test, could not find provided s3 bucket and s3 key: None None',
'status': 'error'
}
def test_configuration(self):
with Temp_Lambda() as temp_lambda:
aws_lambda = temp_lambda.aws_lambda
value = 10
assert aws_lambda.configuration() == aws_lambda.info().get(
'Configuration')
assert aws_lambda.configuration_update(
Timeout=value).get('Timeout') == value
assert aws_lambda.configuration().get('Timeout') == value
def test_event_sources(self):
assert type(self.aws_lambda.event_sources()
) == list # todo: add test with actual event_sources
#self.result = self.aws_lambda.event_sources()
def test_event_source_create(self):
iam_utils = IAM_Utils()
with Temp_Lambda(
) as temp_lambda: # create a temp lambda function that will deleted on exit
with Temp_SQS_Queue(
) as queue: # create a temp sqs queue that will be deleted on exit
event_source_arn = queue.arn()
lambda_name = temp_lambda.lambda_name
aws_lambda = temp_lambda.aws_lambda
aws_lambda.configuration_update(
Timeout=10
) # needs to be less than 30 (which is the default value sent in the queue)
iam_utils.policy_add_sqs_permissions_to_lambda_role(
lambda_name) # add permissions needed to role
sleep(
2
) # todo: need a better solution to find out when the permission is available
aws_lambda.event_source_create(
event_source_arn, lambda_name) # todo: add asserts
iam_utils.policy_remove_sqs_permissions_to_lambda_role(
lambda_name) # remove permissions todo: create custom role
def test_event_source_delete(self):
for uuid in self.aws_lambda.event_sources(index_by='UUID'):
self.result = self.aws_lambda.event_source_delete(
uuid) # these take a bit of time to delete
def test_invoke_raw(self):
with Temp_Lambda() as temp_lambda:
result = temp_lambda.aws_lambda.invoke_raw(
{'name': temp_lambda.lambda_name})
data = result.get('data')
name = result.get('name')
status = result.get('status')
response = result.get('response')
assert status == 'ok'
assert name == temp_lambda.lambda_name
assert data == 'hello {0}'.format(temp_lambda.lambda_name)
assert response.get('StatusCode') == 200
def test_invoke_raw_with_logs(self):
payload = {'name': 'test'}
with Temp_Lambda() as aws_lambda:
result = aws_lambda.invoke_return_logs(payload)
execution_logs = result.get('execution_logs')
request_id = result.get('request_id')
assert result['status'] == 'ok'
assert result['name'] == aws_lambda.lambda_name
assert result['return_value'] == f'hello {payload["name"]}'
assert f'START RequestId: {request_id}' in execution_logs
assert f'END RequestId: {request_id}' in execution_logs
assert 'OSBot inside an Lambda :)' in execution_logs
assert 'Version: $LATEST' in execution_logs
assert 'Duration:' in execution_logs
assert aws_lambda.aws_lambda.invoke_dry_run().get(
'StatusCode'
) == 204 # todo move to dedicated function (once we have one test function created for all tests)
def test_invoke(self):
with Temp_Lambda() as temp_lambda:
result = temp_lambda.aws_lambda.invoke(
{'name': temp_lambda.lambda_name})
assert result == 'hello {0}'.format(temp_lambda.lambda_name)
def test_invoke_async(self):
with Temp_Lambda() as temp_lambda:
result = temp_lambda.aws_lambda.invoke_async(
{'name': temp_lambda.lambda_name})
assert result.get('StatusCode') == 202
def test_invoke_return_logs(self):
with Temp_Lambda() as temp_lambda:
self.result = temp_lambda.aws_lambda.invoke_return_logs()
def test_functions(self):
functions = list(self.aws_lambda.functions())
if len(functions) == 0:
pytest.skip(
"There where no Lambda functions in the target AWS region ")
assert len(functions) > 0
assert set(functions.pop(0)) == {
'CodeSha256', 'CodeSize', 'Description', 'Environment',
'FunctionArn', 'FunctionName', 'Handler', 'LastModified',
'MemorySize', 'PackageType', 'RevisionId', 'Role', 'Runtime',
'Timeout', 'TracingConfig', 'Version'
}
@pytest.mark.skip(
"Takes 20 seconds to run due to the delays in AWS to deliver the logs from Lambda to CloudWatch"
)
def test_log_group(self):
with Temp_Lambda(delete_on_exit=True) as aws_lambda:
result = aws_lambda.invoke_return_logs()
request_id = result.get("request_id")
filter_pattern = f'"{request_id}"'
log_group = aws_lambda.aws_lambda.log_group()
messages = log_group.group_messages_wait_for_pattern(
filter_pattern=filter_pattern)
message = list_contains(messages, 'Duration:').pop()
assert len(messages) == 3
assert message.find(request_id) > -1
def test_policy__permissions(self):
with Temp_Lambda() as temp_lambda:
temp_lambda.aws_lambda.permission_add(temp_lambda.arn(), 'an_id',
'lambda:action',
'lambda.amazonaws.com')
if len(temp_lambda.aws_lambda.permissions()) == 0:
pytest.skip(
"in test_policy__permissions, len(temp_lambda.aws_lambda.permissions()) was 0 (doesn't happen when executing it directly"
) # todo: fix this race condition that happens when running all tests at the same time
assert temp_lambda.aws_lambda.permissions()[0].get(
'Action') == 'lambda:action'
def test_update(self):
with Temp_Lambda() as temp_lambda:
tmp_text = Misc.random_string_and_numbers(prefix='updated code ')
temp_lambda.tmp_folder.create_temp_file(
'def run(event, context): return "{0}"'.format(tmp_text))
temp_lambda.aws_lambda.set_folder_code(
temp_lambda.tmp_folder.folder)
result = temp_lambda.aws_lambda.update()
status = result.get('status')
data = result.get('data')
assert 'TracingConfig' in data[
'update_code'] #set(data) == { 'CodeSha256', 'CodeSize', 'Description', 'FunctionArn', 'FunctionName', 'Handler', 'LastModified', 'MemorySize', 'ResponseMetadata', 'RevisionId', 'Role', 'Runtime', 'Timeout', 'TracingConfig', 'Version'}
assert status == 'ok'
assert temp_lambda.aws_lambda.invoke() == tmp_text
def test_upload(self):
tmp_folder = Temp_Folder_With_Lambda_File(
self.lambda_name).create_temp_file()
(self.aws_lambda.set_s3_bucket(self.s3_bucket).set_s3_key(
self.s3_key).set_folder_code(tmp_folder.folder))
downloaded_file = self.aws_lambda.s3().file_download(
self.s3_bucket, self.s3_key) # download file uploaded
assert file_exists(downloaded_file) is True
unzip_location = tmp_folder.folder + '_unzipped'
Files.unzip_file(downloaded_file, unzip_location) # unzip it
assert file_contents(Files.find(unzip_location + '/*').pop(
)) == tmp_folder.lambda_code # confirm unzipped file's contents
self.aws_lambda.s3().file_delete(self.s3_bucket, self.s3_key)
