diff --git a/tests/unit_tests/_primaite/_simulator/_file_system/test_file.py b/tests/unit_tests/_primaite/_simulator/_file_system/test_file.py index 3c5e7989..9d9228d8 100644 --- a/tests/unit_tests/_primaite/_simulator/_file_system/test_file.py +++ b/tests/unit_tests/_primaite/_simulator/_file_system/test_file.py @@ -1,3 +1,5 @@ +import warnings + import pytest from primaite.simulator.file_system.file import File @@ -84,3 +86,14 @@ def test_file_corrupt_repair_restore(file_system): file.restore() assert file.health_status == FileSystemItemHealthStatus.GOOD + + +def test_file_warning_triggered(file_system): + file: File = file_system.create_file(file_name="test_file.txt", folder_name="test_folder") + + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") + file.check_hash() + # Check warning issued + assert len(w) == 1 + assert "not implemented" in str(w[-1].message) diff --git a/tests/unit_tests/_primaite/_simulator/_file_system/test_folder_actions.py b/tests/unit_tests/_primaite/_simulator/_file_system/test_folder_actions.py index 0067e56c..11043844 100644 --- a/tests/unit_tests/_primaite/_simulator/_file_system/test_folder_actions.py +++ b/tests/unit_tests/_primaite/_simulator/_file_system/test_folder_actions.py @@ -1,3 +1,4 @@ +import warnings from typing import Tuple import pytest @@ -63,6 +64,16 @@ def test_folder_checkhash_request(populated_file_system): assert folder.health_status == FileSystemItemHealthStatus.CORRUPT +def test_folder_warning_triggered(populated_file_system): + fs, folder, _ = populated_file_system + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter("always") + fs.apply_request(request=["folder", folder.name, "checkhash"]) + # Check warning issued + assert len(w) == 1 + assert "not implemented" in str(w[-1].message) + + def test_folder_repair_request(populated_file_system): """Test that an agent can request a folder repair.""" fs, folder, file = populated_file_system