1 files changed, 166 insertions, 0 deletions
diff --git a/tests/mlia/test_utils_filesystem.py b/tests/mlia/test_utils_filesystem.py
new file mode 100644
index 0000000..4d8d955
--- /dev/null
+++ b/tests/mlia/test_utils_filesystem.py
@@ -0,0 +1,166 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Tests for the filesystem module."""
+import contextlib
+import json
+from pathlib import Path
+from unittest.mock import MagicMock
+
+import pytest
+
+from mlia.utils.filesystem import all_files_exist
+from mlia.utils.filesystem import all_paths_valid
+from mlia.utils.filesystem import copy_all
+from mlia.utils.filesystem import get_mlia_resources
+from mlia.utils.filesystem import get_profile
+from mlia.utils.filesystem import get_profiles_data
+from mlia.utils.filesystem import get_profiles_file
+from mlia.utils.filesystem import get_supported_profile_names
+from mlia.utils.filesystem import get_vela_config
+from mlia.utils.filesystem import sha256
+from mlia.utils.filesystem import temp_directory
+from mlia.utils.filesystem import temp_file
+
+
+def test_get_mlia_resources() -> None:
+    """Test resources getter."""
+    assert get_mlia_resources().is_dir()
+
+
+def test_get_vela_config() -> None:
+    """Test Vela config files getter."""
+    assert get_vela_config().is_file()
+    assert get_vela_config().name == "vela.ini"
+
+
+def test_profiles_file() -> None:
+    """Test profiles file getter."""
+    assert get_profiles_file().is_file()
+    assert get_profiles_file().name == "profiles.json"
+
+
+def test_profiles_data() -> None:
+    """Test profiles data getter."""
+    assert list(get_profiles_data().keys()) == [
+        "ethos-u55-256",
+        "ethos-u55-128",
+        "ethos-u65-512",
+    ]
+
+
+def test_profiles_data_wrong_format(
+    monkeypatch: pytest.MonkeyPatch, tmp_path: Path
+) -> None:
+    """Test if profile data has wrong format."""
+    wrong_profile_data = tmp_path / "bad.json"
+    with open(wrong_profile_data, "w", encoding="utf-8") as file:
+        json.dump([], file)
+
+    monkeypatch.setattr(
+        "mlia.utils.filesystem.get_profiles_file",
+        MagicMock(return_value=wrong_profile_data),
+    )
+
+    with pytest.raises(Exception, match="Profiles data format is not valid"):
+        get_profiles_data()
+
+
+def test_get_supported_profile_names() -> None:
+    """Test profile names getter."""
+    assert list(get_supported_profile_names()) == [
+        "ethos-u55-256",
+        "ethos-u55-128",
+        "ethos-u65-512",
+    ]
+
+
+def test_get_profile() -> None:
+    """Test getting profile data."""
+    assert get_profile("ethos-u55-256") == {
+        "target": "ethos-u55",
+        "mac": 256,
+        "system_config": "Ethos_U55_High_End_Embedded",
+        "memory_mode": "Shared_Sram",
+    }
+
+    with pytest.raises(Exception, match="Unable to find target profile unknown"):
+        get_profile("unknown")
+
+
+@pytest.mark.parametrize("raise_exception", [True, False])
+def test_temp_file(raise_exception: bool) -> None:
+    """Test temp_file context manager."""
+    with contextlib.suppress(Exception):
+        with temp_file() as tmp_path:
+            assert tmp_path.is_file()
+
+            if raise_exception:
+                raise Exception("Error!")
+
+    assert not tmp_path.exists()
+
+
+def test_sha256(tmp_path: Path) -> None:
+    """Test getting sha256 hash."""
+    sample = tmp_path / "sample.txt"
+
+    with open(sample, "w", encoding="utf-8") as file:
+        file.write("123")
+
+    expected_hash = "a665a45920422f9d417e4867efdc4fb8a04a1f3fff1fa07e998e86f7f7a27ae3"
+    assert sha256(sample) == expected_hash
+
+
+def test_temp_dir_context_manager() -> None:
+    """Test context manager for temporary directories."""
+    with temp_directory() as tmpdir:
+        assert isinstance(tmpdir, Path)
+        assert tmpdir.is_dir()
+
+    assert not tmpdir.exists()
+
+
+def test_all_files_exist(tmp_path: Path) -> None:
+    """Test function all_files_exist."""
+    sample1 = tmp_path / "sample1.txt"
+    sample1.touch()
+
+    sample2 = tmp_path / "sample2.txt"
+    sample2.touch()
+
+    sample3 = tmp_path / "sample3.txt"
+
+    assert all_files_exist([sample1, sample2]) is True
+    assert all_files_exist([sample1, sample2, sample3]) is False
+
+
+def test_all_paths_valid(tmp_path: Path) -> None:
+    """Test function all_paths_valid."""
+    sample = tmp_path / "sample.txt"
+    sample.touch()
+
+    sample_dir = tmp_path / "sample_dir"
+    sample_dir.mkdir()
+
+    unknown = tmp_path / "unknown.txt"
+
+    assert all_paths_valid([sample, sample_dir]) is True
+    assert all_paths_valid([sample, sample_dir, unknown]) is False
+
+
+def test_copy_all(tmp_path: Path) -> None:
+    """Test function copy_all."""
+    sample = tmp_path / "sample1.txt"
+    sample.touch()
+
+    sample_dir = tmp_path / "sample_dir"
+    sample_dir.mkdir()
+
+    sample_nested_file = sample_dir / "sample_nested.txt"
+    sample_nested_file.touch()
+
+    dest_dir = tmp_path / "dest"
+    copy_all(sample, sample_dir, dest=dest_dir)
+
+    assert (dest_dir / sample.name).is_file()
+    assert (dest_dir / sample_nested_file.name).is_file()