Differential D3432 Diff 12177 swh/loader/tests/__init__.py

Changeset View

Standalone View

swh/loader/tests/init.py

	# Copyright (C) 2018-2020 The Software Heritage developers			# Copyright (C) 2018-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 os			import os
	import subprocess			import subprocess

	from pathlib import PosixPath			from pathlib import PosixPath
	from typing import Optional, Union			from typing import Optional, Union
				from swh.model.model import OriginVisitStatus

				from swh.storage.algos.origin import origin_get_latest_visit_status


				def assert_last_visit_matches(
				storage,
				url: str,
				status: str,
				type: Optional[str] = None,
				snapshot: Optional[bytes] = None,
				) -> OriginVisitStatus:
				"""This retrieves the last visit and visit_status which are expected to exist.

				This also checks that the {visit\|visit_status} have their respective properties
				correctly set.

				This returns the last visit_status for that given origin.

				Args:
				url: Origin url
				status: Check that the visit status has the given status
				type: Check that the returned visit has the given type
				snapshot: Check that the visit status points to the given snapshot

				Raises:
				AssertionError in case visit or visit status is not found, or any of the type,
				status and snapshot mismatch

				Returns:
				the visit status for further check during the remaining part of the test.

				"""
				visit_and_status = origin_get_latest_visit_status(storage, url)
				assert visit_and_status is not None, f"Origin {url} has no visits"
				visit, visit_status = visit_and_status
				if type:
				assert visit.type == type, f"Visit has type {visit.type} instead of {type}"
				assert (
				visit_status.status == status
				), f"Visit_status has status {visit_status.status} instead of {status}"
				if snapshot is not None:
				assert visit_status.snapshot is not None
				olasdUnsubmitted Done Inline Actions This string needs a `f` (just noticed that while breaking the git loader tests...). I also suggest using `.hex()` instead of `!r`. olasd: This string needs a `f` (just noticed that while breaking the git loader tests...). I also…
				ardumontAuthorUnsubmitted Done Inline Actions heh, nice catch ;) ardumont: heh, nice catch ;)
				assert visit_status.snapshot == snapshot, (
				f"Visit_status points to snapshot {visit_status.snapshot.hex()} "
				f"instead of {snapshot.hex()}"
				)
				return visit_status


	def prepare_repository_from_archive(			def prepare_repository_from_archive(
	archive_path: str,			archive_path: str,
	filename: Optional[str] = None,			filename: Optional[str] = None,
	tmp_path: Union[PosixPath, str] = "/tmp",			tmp_path: Union[PosixPath, str] = "/tmp",
	) -> str:			) -> str:
	"""Given an existing archive_path, uncompress it.			"""Given an existing archive_path, uncompress it.
	Show All 13 Lines