Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
 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