Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/package/tests/test_loader_metadata.py
| # Copyright (C) 2019-2020 The Software Heritage developers | # Copyright (C) 2019-2020 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 | ||||
| import datetime | import datetime | ||||
| from typing import Iterator, Optional, Sequence, Tuple | from typing import Iterator, Optional, Sequence, Tuple | ||||
| import attr | import attr | ||||
| from swh.loader.package.loader import BasePackageInfo, PackageLoader | from swh.loader.package.loader import ( | ||||
| BasePackageInfo, | |||||
| PackageLoader, | |||||
| RawExtrinsicMetadataCore, | |||||
| ) | |||||
| from swh.model.hashutil import hash_to_bytes | from swh.model.hashutil import hash_to_bytes | ||||
| from swh.model.identifiers import SWHID | from swh.model.identifiers import SWHID | ||||
| from swh.model.model import ( | from swh.model.model import ( | ||||
| MetadataAuthority, | MetadataAuthority, | ||||
| MetadataAuthorityType, | MetadataAuthorityType, | ||||
| MetadataFetcher, | MetadataFetcher, | ||||
| MetadataTargetType, | MetadataTargetType, | ||||
| RawExtrinsicMetadata, | RawExtrinsicMetadata, | ||||
| ▲ Show 20 Lines • Show All 56 Lines • ▼ Show 20 Lines | class MetadataTestLoader(PackageLoader[BasePackageInfo]): | ||||
| def get_package_info(self, version: str) -> Iterator[Tuple[str, BasePackageInfo]]: | def get_package_info(self, version: str) -> Iterator[Tuple[str, BasePackageInfo]]: | ||||
| m0 = METADATA[0] | m0 = METADATA[0] | ||||
| m1 = METADATA[1] | m1 = METADATA[1] | ||||
| p_info = BasePackageInfo( | p_info = BasePackageInfo( | ||||
| url=ORIGIN_URL, | url=ORIGIN_URL, | ||||
| filename="archive.tgz", | filename="archive.tgz", | ||||
| revision_extrinsic_metadata=[ | revision_extrinsic_metadata=[ | ||||
| (m0.discovery_date, m0.format, m0.metadata), | RawExtrinsicMetadataCore(m0.format, m0.metadata, m0.discovery_date), | ||||
| (m1.discovery_date, m1.format, m1.metadata), | RawExtrinsicMetadataCore(m1.format, m1.metadata, m1.discovery_date), | ||||
| ], | ], | ||||
| ) | ) | ||||
| yield (version, p_info) | yield (version, p_info) | ||||
| def test_load_revision_metadata(swh_config, caplog): | def test_load_revision_metadata(swh_config, caplog): | ||||
| storage = get_storage("memory") | storage = get_storage("memory") | ||||
| ▲ Show 20 Lines • Show All 65 Lines • Show Last 20 Lines | |||||