Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/test_provenance_db.py
# Copyright (C) 2021 The Software Heritage developers | # Copyright (C) 2021 The Software Heritage developers | ||||
# See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
# License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
# See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
from swh.provenance.interface import ProvenanceInterface | from swh.provenance.interface import ProvenanceInterface | ||||
from swh.provenance.postgresql.provenancedb import ProvenanceDB | from swh.provenance.postgresql.provenance import ProvenanceStoragePostgreSql | ||||
def test_provenance_flavor(provenance: ProvenanceInterface) -> None: | def test_provenance_flavor(provenance: ProvenanceInterface) -> None: | ||||
if isinstance(provenance.storage, ProvenanceDB): | if isinstance(provenance.storage, ProvenanceStoragePostgreSql): | ||||
assert provenance.storage.flavor in ( | assert provenance.storage.flavor in ( | ||||
"with-path", | "with-path", | ||||
"without-path", | "without-path", | ||||
"with-path-denormalized", | "with-path-denormalized", | ||||
"without-path-denormalized", | "without-path-denormalized", | ||||
) | ) |