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