data.py
View Options

	# Copyright (C) 2022 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

	from swh.model.model import (
	Directory,
	DirectoryEntry,
	ObjectType,
	Release,
	Revision,
	RevisionType,
	)
	from swh.model.tests import swh_model_data


	def populate_search_data(search):
	search.origin_update({"url": origin.url} for origin in get_origins())


	def get_origins():
	return swh_model_data.ORIGINS


	def get_visits():
	return swh_model_data.ORIGIN_VISITS


	def get_visit_status():
	return swh_model_data.ORIGIN_VISIT_STATUSES


	def get_snapshots():
	return swh_model_data.SNAPSHOTS


	def get_releases():
	return swh_model_data.RELEASES


	def get_revisions():
	return swh_model_data.REVISIONS


	def get_contents():
	return swh_model_data.CONTENTS


	def get_directories():
	return swh_model_data.DIRECTORIES


	def get_releases_with_target():
	"""
	GraphQL will not return a target object unless the target id
	is present in the DB.
	Return release objects with real targets instead of dummy
	targets in swh.model.tests.swh_model_data
	"""
	with_revision = Release(
	name=b"v0.0.1",
	target_type=ObjectType.REVISION,
	target=get_revisions()[0].id,
	message=b"foo",
	synthetic=False,
	)
	with_release = Release(
	name=b"v0.0.1",
	target_type=ObjectType.RELEASE,
	target=get_releases()[0].id,
	message=b"foo",
	synthetic=False,
	)
	with_directory = Release(
	name=b"v0.0.1",
	target_type=ObjectType.DIRECTORY,
	target=get_directories()[0].id,
	message=b"foo",
	synthetic=False,
	)
	with_content = Release(
	name=b"v0.0.1",
	target_type=ObjectType.CONTENT,
	target=get_contents()[0].sha1_git,
	message=b"foo",
	synthetic=False,
	)
	return [with_revision, with_release, with_directory, with_content]


	def get_revisions_with_parents():
	"""
	Revisions with real revisions as parents
	"""
	return [
	Revision(
	message=b"hello",
	date=swh_model_data.DATES[0],
	committer=swh_model_data.COMMITTERS[0],
	author=swh_model_data.COMMITTERS[0],
	committer_date=swh_model_data.DATES[0],
	type=RevisionType.GIT,
	directory=b"\x01" * 20,
	synthetic=False,
	parents=(get_revisions()[0].id, get_revisions()[1].id),
	)
	]


	def get_directories_with_nested_path():
	return [
	Directory(
	entries=(
	DirectoryEntry(
	name=b"sub-dir",
	perms=0o644,
	type="dir",
	target=get_directories()[1].id,
	),
	)
	)
	]


	GRAPHQL_EXTRA_TEST_OBJECTS = {
	"release": get_releases_with_target(),
	"revision": get_revisions_with_parents(),
	"directory": get_directories_with_nested_path(),
	}


	def populate_dummy_data(storage):
	for object_type, objects in swh_model_data.TEST_OBJECTS.items():
	method = getattr(storage, object_type + "_add")
	method(objects)
	for object_type, objects in GRAPHQL_EXTRA_TEST_OBJECTS.items():
	method = getattr(storage, object_type + "_add")
	method(objects)

File Metadata

Mime Type: text/x-python
Expires: Thu, Jul 3, 10:17 AM (2 w, 5 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3355510

data.py
No OneTemporary
Actions

data.py
View Options

File Metadata

Event Timeline

data.pyNo OneTemporaryActions

data.pyView Options

File Metadata

Event Timeline

data.py
No OneTemporary
Actions

data.py
View Options