test_revision_iterator.py
No OneTemporary
Actions

Size

971 B

Subscribers

None

test_revision_iterator.py
View Options

	# Copyright (C) 2021 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.provenance.revision import CSVRevisionIterator
	from swh.provenance.tests.conftest import fill_storage, load_repo_data, ts2dt
	from swh.storage.interface import StorageInterface


	@pytest.mark.parametrize(
	"repo",
	(
	"cmdbts2",
	"out-of-order",
	),
	)
	def test_revision_iterator(swh_storage: StorageInterface, repo: str) -> None:
	"""Test CSVRevisionIterator"""
	data = load_repo_data(repo)
	fill_storage(swh_storage, data)

	revisions_csv = [
	(rev["id"], ts2dt(rev["date"]), rev["directory"]) for rev in data["revision"]
	]
	revisions = list(CSVRevisionIterator(revisions_csv))

	assert revisions
	assert len(revisions) == len(data["revision"])