def test_file_eq(historian: mincepy.Historian):
file1 = historian.create_file('file1')
file1_again = historian.create_file('file1')
file3 = historian.create_file('file3')
file1.write_text('hello 1!')
file1_again.write_text('hello 1!') # Same again
file3.write_text('hello 3!') # Different
assert file1 == file1_again
assert file1 != file3
assert file1_again != file3
def test_file_from_disk(tmp_path, historian: mincepy.Historian):
disk_file_path = tmp_path / 'test.txt'
disk_file_path.write_text('TEST')
file = historian.create_file('test.txt')
file.from_disk(disk_file_path)
contents = file.read_text()
assert contents == 'TEST'
def test_merge_file(historian: mincepy.Historian):
"""Test that merging files works correctly"""
file = historian.create_file('test.dat')
file.write_text('bla bla')
file.save()
with testing.temporary_historian(
testing.create_archive_uri(db_name='test_historian')) as remote:
local = historian
# Merge the file into the remote
result = remote.merge(local.find(obj_id=file.obj_id))
assert len(result.all) == len(result.merged) == local.find().count()
assert local.find().count() == remote.find().count()
remote_file = remote.get(file.obj_id)
assert file.read_text() == remote_file.read_text()
# Now check that files contained within objects are correctly merged
file_list = mincepy.List((file, ))
file_list.save() # pylint: disable=no-member
result = remote.merge(local.find(obj_id=file_list.obj_id)) # pylint: disable=no-member
assert len(result.merged) == 1
assert historian.get_snapshot_id(file_list) in result.merged
remote_file_list = remote.get(file_list.obj_id) # pylint: disable=no-member
assert file.read_text() == remote_file_list[0].read_text()
def test_nested_files_in_list_mutating(tmp_path, historian: mincepy.Historian): # pylint: disable=unused-argument
encoding = 'utf-8'
INITIAL_DATA = 'First string'.encode(encoding)
my_file = historian.create_file()
with my_file.open('wb') as file:
file.write(INITIAL_DATA)
my_list = mincepy.builtins.List()
my_list.append(my_file)
list_id = historian.save(my_list)
# Now let's append to the file
NEW_DATA = 'Second string'.encode(encoding)
with my_file.open('ab') as file:
file.write(NEW_DATA)
# Save the list again
historian.save(my_list)
del my_list
loaded = historian.load(list_id)
with loaded[0].open('rb') as contents:
buffer = io.BytesIO()
shutil.copyfileobj(contents, buffer)
assert buffer.getvalue() == INITIAL_DATA + NEW_DATA
def test_file_changing(tmp_path, historian: mincepy.Historian): # pylint: disable=unused-argument
encoding = 'utf-8'
INITIAL_DATA = 'Initial string'
mince_file = historian.create_file(encoding=encoding)
with mince_file.open('w') as file:
file.write(INITIAL_DATA)
historian.save(mince_file)
# Now let's append to the file
NEW_DATA = 'Second string'
with mince_file.open('a') as file:
file.write(NEW_DATA)
historian.save(mince_file)
history = historian.history(mince_file)
assert len(history) == 2
with history[0].obj.open() as file:
buffer = io.StringIO()
shutil.copyfileobj(file, buffer)
assert INITIAL_DATA == buffer.getvalue()
with history[1].obj.open() as file:
buffer = io.StringIO()
shutil.copyfileobj(file, buffer)
assert INITIAL_DATA + NEW_DATA == buffer.getvalue()
def test_nested_files_in_dict(historian: mincepy.Historian):
file = historian.create_file()
my_dict = mincepy.builtins.Dict()
my_dict['file'] = file
list_id = historian.save(my_dict)
del my_dict
loaded = historian.load(list_id)
assert len(loaded) == 1
assert loaded['file'].filename is None
def test_nested_files_in_list(historian: mincepy.Historian):
file = historian.create_file()
my_list = mincepy.builtins.List()
my_list.append(file)
list_id = historian.save(my_list)
del my_list
loaded = historian.load(list_id)
assert len(loaded) == 1
assert loaded[0].filename is None
def test_file_to_disk(tmp_path, historian: mincepy.Historian):
disk_file_path = tmp_path / 'test.txt'
file = historian.create_file('test.txt')
file.write_text('TEST')
file.to_disk(tmp_path)
assert disk_file_path.read_text() == 'TEST'
# Now test writing to a custom file name
custom_filename = tmp_path / 'test_file.txt'
file.to_disk(custom_filename)
assert custom_filename.read_text() == 'TEST'
def test_file_basics(historian: mincepy.Historian):
ENCODING = 'utf-8'
INITIAL_DATA = 'hello there'
file = historian.create_file(ENCODING)
with file.open('w') as stream:
stream.write(INITIAL_DATA)
file_id = historian.save(file)
del file
loaded = historian.load(file_id)
with loaded.open('r') as file:
buffer = io.StringIO()
shutil.copyfileobj(file, buffer)
assert buffer.getvalue() == INITIAL_DATA