def test_best_compression_method_is_chosen(blockchain_path, compressible_data):
fss = FileSystemStorage(blockchain_path, compressors=('gz', 'bz2', 'xz'))
compressed_path = blockchain_path / 'file.txt.gz'
fss.save('file.txt', binary_data=compressible_data, is_final=True)
assert compressed_path.exists()
assert len(compressed_path.read_bytes()) == 46
def test_finalized_data_is_compressed(blockchain_path, compression, compressible_data):
fss = FileSystemStorage(blockchain_path, compressors=(compression,))
file_path = blockchain_path / f'file.txt.{compression}'
fss.save('file.txt', compressible_data, is_final=True)
decompressed_data = decompress(file_path, compression)
assert decompressed_data == compressible_data
def test_save_to_raw_finalized_file_raises_error(blockchain_path, compressible_data, is_final):
fss = FileSystemStorage(blockchain_path, compressors=())
filename = 'file.txt'
fss.save(filename, binary_data=compressible_data, is_final=True)
with pytest.raises(exceptions.FinalizedFileWriteError):
fss.save(filename, compressible_data, is_final)
def test_no_compression_file_storage_saves_raw_files(blockchain_path, compressible_data):
fss = FileSystemStorage(blockchain_path, compressors=())
file_path = blockchain_path / 'file.txt'
fss.save('file.txt', binary_data=compressible_data, is_final=True)
assert file_path.exists()
assert file_path.read_bytes() == compressible_data
def test_incompressible_data_is_saved_raw(blockchain_path, compression, incompressible_data):
fss = FileSystemStorage(blockchain_path)
file_path = blockchain_path / 'incompressible_file.txt'
fss.save('incompressible_file.txt', binary_data=incompressible_data, is_final=True)
assert file_path.exists()
assert file_path.read_bytes() == incompressible_data
def test_cannot_append_to_raw_finalized_file(blockchain_path, incompressible_data, is_final):
fss = FileSystemStorage(blockchain_path, compressors=())
filename = 'file.txt'
fss.save(filename, binary_data=incompressible_data, is_final=True)
with pytest.raises(exceptions.FinalizedFileWriteError):
fss.append(filename, b'appended data', is_final)
def test_move(blockchain_path):
fss = FileSystemStorage(blockchain_path)
source = 'file1.txt'
destination = 'file2.txt'
fss.save(source, b'AAA')
fss.move(source, destination)
assert os.path.isfile(blockchain_path / destination)
assert not os.path.isfile(blockchain_path / source)
assert fss.load(destination) == b'AAA'
def write_to_file(blockchain_state, path):
directory, filename = os.path.split(path)
storage = FileSystemStorage(directory)
storage.save(filename, blockchain_state.to_messagepack(), is_final=True)
def test_cannot_use_absolute_path(blockchain_path, compressible_data):
fss = FileSystemStorage(blockchain_path)
file_path = blockchain_path / 'subdir/file.txt'
with pytest.raises(ValueError):
fss.save(file_path, compressible_data)
def test_cannot_save_file_out_of_base_directory(blockchain_path, compressible_data):
fss = FileSystemStorage(blockchain_path / 'subdir')
file_path = '../file.txt'
with pytest.raises(ValueError):
fss.save(file_path, compressible_data)