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)
def test_folder_zip(self):
folder = temp_folder_with_temp_file(file_contents=random_string())
print()
zip_file = folder_zip(folder)
assert file_exists(zip_file)
assert file_extension(zip_file) == '.zip'
unziped_folder = Files.unzip_file(zip_file)
source_files = folder_files(folder)
target_files = folder_files(unziped_folder)
assert len(source_files) == 1
assert len(target_files) == 1
assert source_files[0] != target_files[0]
assert file_contents(source_files[0]) == file_contents(target_files[0])
assert zip_file_list(zip_file) == ['temp_file.txt']
def __enter__(self):
if Files.exists(self.zip_file):
if self.target_folder is None:
self.target_folder = Files.temp_folder("unzipped_")
return Files.unzip_file(self.zip_file, self.target_folder)
return None
