test_directory.py
No OneTemporary
Actions

Size

1 KB

Subscribers

None

test_directory.py
View Options

	# 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