oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#1714: FileSystemItem is no longer an abstract base class + Added enums and enum sizes + stream lined FileSystemFile init

Browse Source
This commit is contained in:
Czar Echavez
2023-08-07 11:52:54 +01:00
parent d57c2a936e
commit b58a3a3e24
7 changed files with 143 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit_tests/_primaite/_simulator/_file_system/test_file_system.py
View File

@@ -10,7 +10,7 @@ def test_create_folder_and_file():
file_system.create_file(file_name="test_file", file_size=10, folder_uuid=folder.uuid)
assert len(file_system.get_folders()[0].get_files()) is 1
assert file_system.get_folders()[0].get_files()[0].get_file_name() is "test_file"
assert file_system.get_folders()[0].get_files()[0].get_file_size() is 10
assert file_system.get_folders()[0].get_files()[0].get_file_size() == 10
def test_create_file():
@@ -92,6 +92,6 @@ def test_serialisation():
assert len(file_system.get_folders()[0].get_files()) is 1
serialised_file_sys = file_system.model_dump_json()
deserialised_file_sys = FileSystem().model_validate_json(serialised_file_sys)
deserialised_file_sys = FileSystem.model_validate_json(serialised_file_sys)
assert file_system == deserialised_file_sys

4
tests/unit_tests/_primaite/_simulator/_file_system/test_file_system_file.py
View File

@@ -11,13 +11,13 @@ def test_file_type():
def test_get_file_size():
"""Tests that the file size is being returned properly."""
file = FileSystemFile(item_name="test", item_size=1.5)
assert file.get_file_size() is 1.5
assert file.get_file_size() == 1.5
def test_serialisation():
"""Test to check that the object serialisation works correctly."""
file = FileSystemFile(item_name="test", item_size=1.5, file_type=FileSystemFileType.DOC)
serialised_file = file.model_dump_json()
deserialised_file = FileSystemFile(item_name="test").model_validate_json(serialised_file)
deserialised_file = FileSystemFile.model_validate_json(serialised_file)
assert file == deserialised_file

8
tests/unit_tests/_primaite/_simulator/_file_system/test_file_system_folder.py
View File

@@ -10,11 +10,11 @@ def test_adding_removing_file():
file = FileSystemFile(item_name="test_file", item_size=10, file_type=FileSystemFileType.DOC)
folder.add_file(file)
assert folder.get_folder_size() is 10
assert folder.get_folder_size() == 10
assert len(folder.get_files()) is 1
folder.remove_file(file)
assert folder.get_folder_size() is 0
assert folder.get_folder_size() == 0
assert len(folder.get_files()) is 0
@@ -27,7 +27,7 @@ def test_get_file_by_id():
folder.add_file(file)
folder.add_file(file2)
assert folder.get_folder_size() is 20
assert folder.get_folder_size() == 20
assert len(folder.get_files()) is 2
assert folder.get_file(file_id=file.uuid) is file
@@ -54,6 +54,6 @@ def test_serialisation():
serialised_folder = folder.model_dump_json()
deserialised_folder = FileSystemFolder(item_name="test").model_validate_json(serialised_folder)
deserialised_folder = FileSystemFolder.model_validate_json(serialised_folder)
assert folder == deserialised_folder