diff --git a/swh/provenance/tests/test_revision_iterator.py b/swh/provenance/tests/test_revision_iterator.py index 7a4afd9..25d7c35 100644 --- a/swh/provenance/tests/test_revision_iterator.py +++ b/swh/provenance/tests/test_revision_iterator.py @@ -1,19 +1,30 @@ # 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 from swh.provenance.tests.test_provenance_db import ts2dt -def test_archive_direct_revision_iterator(storage_and_CMDBTS, archive_direct): +@pytest.mark.parametrize( + "repo", + ( + "cmdbts2", + "out-of-order", + ), +) +def test_archive_direct_revision_iterator(swh_storage, archive_direct, repo): """Test CSVRevisionIterator""" - storage, data = storage_and_CMDBTS + data = load_repo_data(repo) + fill_storage(swh_storage, data) revisions_csv = [ (rev["id"], ts2dt(rev["date"]).isoformat(), rev["directory"]) for rev in data["revision"] ] revisions = list(CSVRevisionIterator(revisions_csv, archive_direct)) assert revisions assert len(revisions) == len(data["revision"])