Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/algos/test_directory.py
- This file was added.
| # Copyright (C) 2022 The Software Heritage developers | |||||
| # See the AUTHORS file at the top-level directory of this distribution | |||||
| # License: GNU General Public License version 3, or any later version | |||||
| # See top-level LICENSE file for more information | |||||
| import pytest | |||||
| from swh.model.model import Directory, DirectoryEntry | |||||
| from swh.storage.algos.directory import directory_get | |||||
| from ..storage_data import StorageData | |||||
| @pytest.mark.parametrize("directory_id", [d.id for d in StorageData.directories]) | |||||
| def test_directory_small(swh_storage, directory_id): | |||||
| swh_storage.directory_add(StorageData.directories) | |||||
| (expected_directory,) = [d for d in StorageData.directories if d.id == directory_id] | |||||
| returned_directory = directory_get(swh_storage, directory_id) | |||||
| assert returned_directory.id == expected_directory.id | |||||
| assert set(returned_directory.entries) == set(expected_directory.entries) | |||||
| assert returned_directory.raw_manifest == expected_directory.raw_manifest | |||||
| def test_directory_large(swh_storage): | |||||
| expected_directory = Directory( | |||||
| entries=tuple( | |||||
| DirectoryEntry( | |||||
| name=f"entry{i:04}".encode(), | |||||
| type="file", | |||||
| target=b"\x00" * 20, | |||||
| perms=0o000664, | |||||
| ) | |||||
| for i in range(10) | |||||
| ) | |||||
| ) | |||||
| swh_storage.directory_add([expected_directory]) | |||||
| returned_directory = directory_get(swh_storage, expected_directory.id) | |||||
| assert returned_directory.id == expected_directory.id | |||||
| assert set(returned_directory.entries) == set(expected_directory.entries) | |||||
| assert returned_directory.raw_manifest == expected_directory.raw_manifest | |||||