#1714: updated file system classes

tests/unit_tests/_primaite/_simulator/_file_system/test_file_system.py

@@ -0,0 +1,80 @@
+from primaite.simulator.file_system.file_system import FileSystem
+
+
+def test_create_folder_and_file():
+    """Test creating a folder and a file."""
+    file_system = FileSystem()
+    folder = file_system.create_folder()
+    assert len(file_system.get_folders()) is 1
+
+    file_system.create_file(file_size=10, folder_uuid=folder.uuid)
+    assert len(file_system.get_folders()[0].get_files()) is 1
+
+
+def test_create_file():
+    """Tests that creating a file without a folder creates a folder and sets that as the file's parent."""
+    file_system = FileSystem()
+
+    file = file_system.create_file(file_size=10)
+    assert len(file_system.get_folders()) is 1
+    assert file_system.get_folders()[0].get_file(file.uuid) is file
+
+
+def test_delete_file():
+    """Tests that a file can be deleted."""
+    file_system = FileSystem()
+
+    file = file_system.create_file(file_size=10)
+    assert len(file_system.get_folders()) is 1
+    assert file_system.get_folders()[0].get_file(file.uuid) is file
+
+    file_system.delete_file(file.uuid)
+    assert len(file_system.get_folders()) is 1
+    assert len(file_system.get_folders()[0].get_files()) is 0
+
+
+def test_delete_folder():
+    file_system = FileSystem()
+    folder = file_system.create_folder()
+    assert len(file_system.get_folders()) is 1
+
+    file_system.delete_folder(folder.uuid)
+    assert len(file_system.get_folders()) is 0
+
+
+def test_move_file():
+    """Tests the file move function."""
+    file_system = FileSystem()
+    src_folder = file_system.create_folder()
+    assert len(file_system.get_folders()) is 1
+
+    target_folder = file_system.create_folder()
+    assert len(file_system.get_folders()) is 2
+
+    file = file_system.create_file(file_size=10, folder_uuid=src_folder.uuid)
+    assert len(file_system.get_folder_by_id(src_folder.uuid).get_files()) is 1
+    assert len(file_system.get_folder_by_id(target_folder.uuid).get_files()) is 0
+
+    file_system.move_file(src_folder.uuid, target_folder.uuid, file.uuid)
+
+    assert len(file_system.get_folder_by_id(src_folder.uuid).get_files()) is 0
+    assert len(file_system.get_folder_by_id(target_folder.uuid).get_files()) is 1
+
+
+def test_copy_file():
+    """Tests the file copy function."""
+    file_system = FileSystem()
+    src_folder = file_system.create_folder()
+    assert len(file_system.get_folders()) is 1
+
+    target_folder = file_system.create_folder()
+    assert len(file_system.get_folders()) is 2
+
+    file = file_system.create_file(file_size=10, folder_uuid=src_folder.uuid)
+    assert len(file_system.get_folder_by_id(src_folder.uuid).get_files()) is 1
+    assert len(file_system.get_folder_by_id(target_folder.uuid).get_files()) is 0
+
+    file_system.copy_file(src_folder.uuid, target_folder.uuid, file.uuid)
+
+    assert len(file_system.get_folder_by_id(src_folder.uuid).get_files()) is 1
+    assert len(file_system.get_folder_by_id(target_folder.uuid).get_files()) is 1

tests/unit_tests/_primaite/_simulator/_file_system/test_file_system_file.py

@@ -0,0 +1,14 @@
+from primaite.simulator.file_system.file_system_file import FileSystemFile
+from primaite.simulator.file_system.file_system_file_type import FileSystemFileType
+
+
+def test_file_type():
+    """Tests tha the FileSystemFile type is set correctly."""
+    file = FileSystemFile(file_size=1.5, file_type=FileSystemFileType.TBD)
+    assert file.get_file_type() is FileSystemFileType.TBD
+
+
+def test_get_file_size():
+    """Tests that the file size is being returned properly."""
+    file = FileSystemFile(file_size=1.5, file_type=FileSystemFileType.TBD)
+    assert file.get_file_size() is 1.5

tests/unit_tests/_primaite/_simulator/_file_system/test_file_system_folder.py

@@ -0,0 +1,41 @@
+from primaite.simulator.file_system.file_system_file import FileSystemFile
+from primaite.simulator.file_system.file_system_file_type import FileSystemFileType
+from primaite.simulator.file_system.file_system_folder import FileSystemFolder
+
+
+def test_adding_removing_file():
+    folder = FileSystemFolder()
+
+    file = FileSystemFile(file_size=10, file_type=FileSystemFileType.TBD)
+
+    folder.add_file(file)
+    assert folder.get_folder_size() is 10
+    assert len(folder.get_files()) is 1
+
+    folder.remove_file(file_id=file.uuid)
+    assert folder.get_folder_size() is 0
+    assert len(folder.get_files()) is 0
+
+
+def test_get_file_by_id():
+    folder = FileSystemFolder()
+
+    file = FileSystemFile(file_size=10, file_type=FileSystemFileType.TBD)
+
+    folder.add_file(file)
+    assert folder.get_folder_size() is 10
+    assert len(folder.get_files()) is 1
+
+    assert folder.get_file(file_id=file.uuid) is file
+
+
+def test_folder_quarantine_state():
+    folder = FileSystemFolder()
+
+    assert folder.quarantine_status() is False
+
+    folder.quarantine()
+    assert folder.quarantine_status() is True
+
+    folder.end_quarantine()
+    assert folder.quarantine_status() is False