def test_file_system_root_ids():
"""
Test the various versions of root and ensure the path is set correctly
"""
file_system = FileSystem()
root_drive_a = file_system.create('drive', 'a', '')
root_drive_b = file_system.create('drive', 'b', '\\')
root_drive_c = file_system.create('drive', 'c', None)
root_drive_d = file_system.create('drive', 'd', 'root')
assert root_drive_a.path == 'a'
assert root_drive_b.path == 'b'
assert root_drive_c.path == 'c'
assert root_drive_d.path == 'd'
def test_file_system_create_drives():
"""
Test a folder being added to a drive
"""
file_system = FileSystem()
drive_a = file_system.create('drive', 'a', '')
folder_a = file_system.create('folder', 'a', 'a')
assert folder_a.name in drive_a.get_names()
assert folder_a.entity_type == 'folder'
assert len(folder_a.get_names()) == 0
assert folder_a.path == 'a\\a'
def test_print_recursive():
"""
Test a helper function that recursively prints the contents of an entity
"""
file_system = FileSystem()
file_system.create('drive', 'A', '')
file_system.create('folder', 'stuff', 'A')
file_system.create('text', 'list', 'A\\stuff')
file_system.create('folder', 'more_stuff', 'A')
output_string = print_recursive(file_system._root)
expected_string = ' 0\nA 0\nA\\stuff 0\nA\\stuff\\list 0\nA\\more_stuff 0\n'
assert output_string == expected_string
def test_file_system_init():
"""
Test the initial state of the file system
"""
file_system = FileSystem()
assert len(file_system._root.get_names()) == 0
assert file_system._root.entity_type == 'root'
assert file_system._root.path == ''
assert file_system._root.name == 'root'
assert file_system._root.size == 0
def test_file_system_root():
"""
Test the root to ensure only drives can be added
"""
file_system = FileSystem()
with pytest.raises(IllegalFileSystemOperation):
file_system.create('folder', 'test', 'root')
with pytest.raises(IllegalFileSystemOperation):
file_system.create('zip', 'test', 'root')
with pytest.raises(IllegalFileSystemOperation):
file_system.create('text', 'test', 'root')
new_drive = file_system.create('drive', 'test', 'root')
assert new_drive.entity_type == 'drive'
assert new_drive.path == 'test'
assert new_drive.size == 0
def test_file_system_delete():
"""
Test content being deleted, and removal of size from ancestors
"""
file_system = FileSystem()
drive_a = file_system.create('drive', 'a', '')
folder_a = file_system.create('folder', 'a', 'a')
text_a = file_system.create('text', 'a', 'a\\a')
test_string = 'teststring'
file_system.write_to_file(text_a.path, test_string)
assert folder_a.size == len(test_string)
assert drive_a.size == len(test_string)
file_system.delete(text_a.path)
assert 'a' not in folder_a.get_names()
assert folder_a.size == 0
assert drive_a.size == 0
def test_file_system_write_to_file():
"""
Test content being added to a text file and the size being distributed
"""
file_system = FileSystem()
drive_a = file_system.create('drive', 'a', '')
folder_b = file_system.create('folder', 'b', 'a')
text_c = file_system.create('text', 'c', 'a\\b')
test_string = 'teststring'
file_system.write_to_file(text_c.path, test_string)
assert text_c.content == test_string
assert text_c.size == len(test_string)
assert folder_b.size == len(test_string)
assert drive_a.size == len(test_string)
def test_file_system_create_zip():
"""
Test content being added to a text file and the size being distributed up through a zip
"""
file_system = FileSystem()
drive_a = file_system.create('drive', 'a', '')
zip_a = file_system.create('zip', 'a', 'a')
text_a = file_system.create('text', 'a', 'a\\a')
assert text_a.name in zip_a.get_names()
file_system.write_to_file(text_a.path, 'teststring')
assert text_a.size == len('teststring')
assert zip_a.size == len('teststring') / 2
assert drive_a.size == len('teststring') / 2
def test_file_system_create_text():
"""
Test content being added to a text file and the size being distributed up
"""
file_system = FileSystem()
drive_a = file_system.create('drive', 'a', '')
folder_a = file_system.create('folder', 'a', 'a')
text_a = file_system.create('text', 'a', 'a\\a')
assert text_a.name in folder_a.get_names()
file_system.write_to_file(text_a.path, 'test_string')
assert text_a.size == len('test_string')
assert folder_a.size == len('test_string')
assert drive_a.size == len('test_string')
def test_file_system_string():
"""
Test the file system string method
"""
file_system = FileSystem()
file_system.create('drive', 'A', '')
file_system.create('drive', 'B', '')
file_system.create('folder', 'stuff1', 'A')
file_system.create('zip', 'zip1', 'A\\stuff1')
text_file = file_system.create('text', 'list1', 'A\\stuff1\\zip1')
file_system.create('folder', 'stuff2', 'A')
file_system.write_to_file(text_file.path, 'test')
file_system.create('zip', 'stuff3', 'B')
file_system.create('folder', 'stuff4', 'B\\stuff3')
expected_string = 'A 2\nA\\stuff1 2\nA\\stuff1\\zip1 2\nA\\stuff1\\zip1\\list1 4\nA\\stuff2 0\n' \
'B 0\nB\\stuff3 0\nB\\stuff3\\stuff4 0'
assert str(file_system) == expected_string
def test_file_system_move():
"""
Test content being moved and sizes being incremented and decremented
"""
file_system = FileSystem()
file_system.create('drive', 'a', '')
folder_b_source = file_system.create('folder', 'b', 'a')
folder_b_target = file_system.create('folder', 'bb', 'a')
text_c = file_system.create('text', 'c', 'a\\b')
test_string = 'teststring'
file_system.write_to_file(text_c.path, test_string)
file_system.move(text_c.path, folder_b_target.path)
assert 'c' not in folder_b_source.get_names()
assert 'c' in folder_b_target.get_names()
assert folder_b_source.size == 0
assert folder_b_target.size == len(test_string)
def main():
"""
Demonstrates the FileSystem implementation.
Executes methods and prints the file system structure after each operation.
"""
print('\n##############')
print('# File System')
print('##############\n')
print('Package in /file_system\n')
print('Pytest suite in /tests\n')
print('###################')
print('# File System Demo')
print('###################\n')
print('1. Creating File System')
file_system = FileSystem()
print('2. Adding DriveA to Root')
file_system.create('drive', 'DriveA', '')
print('Current File System Structure: \n' + str(file_system))
print()
print('3. Adding DriveB to Root')
file_system.create('drive', 'DriveB', '')
print('Current File System Structure: \n' + str(file_system))
print()
print('4. Adding FolderA to DriveA')
file_system.create('folder', 'FolderA', 'DriveA')
print('Current File System Structure: \n' + str(file_system))
print()
print('5. Adding FolderB to DriveB')
folder_b = file_system.create('folder', 'FolderB', 'DriveB')
print('Current File System Structure: \n' + str(file_system))
print()
print('6. Adding FolderC to FolderB')
file_system.create('folder', 'FolderC', 'DriveB\\FolderB')
print('Current File System Structure: \n' + str(file_system))
print()
print('7. Adding ZipA to DriveA')
zip_a = file_system.create('zip', 'ZipA', 'DriveA')
print('Current File System Structure: \n' + str(file_system))
print()
print('8. Adding TextA to FolderC')
text_a = file_system.create('text', 'TextA', 'DriveB\\FolderB\\FolderC')
print('Current File System Structure: \n' + str(file_system))
print()
print('9. Updating TextA Content with \'testabcd\'')
file_system.write_to_file(text_a.path, 'testabcd')
print('Current File System Structure: \n' + str(file_system))
print()
print('10. Adding TextB to ZipA')
text_b = file_system.create('text', 'TextB', 'DriveA\\ZipA')
print('Current File System Structure: \n' + str(file_system))
print()
print('11. Updating TextB Content with \'testabcdtestabcd\'')
file_system.write_to_file(text_b.path, 'testabcdtestabcd')
print('Current File System Structure: \n' + str(file_system))
print()
print('12. Move FolderB into ZipA')
file_system.move(folder_b.path, zip_a.path)
print('Current File System Structure: \n' + str(file_system))
print()
def _load_configs_and_init_data(context):
context.data_generator = DataGenerator(context)
context.file_system = FileSystem(context)
context.real_file_system = RealFileSystem(context)