Differential D3575 Diff 12591 swh/storage/tests/test_pytest_plugin.py

Changeset View

Standalone View

swh/storage/tests/test_pytest_plugin.py

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


	from swh.storage.pytest_plugin import OBJECT_FACTORY			from swh.storage.pytest_plugin import OBJECT_FACTORY


	from swh.model.model import BaseModel			from swh.model.model import BaseModel


	def test_sample_data(sample_data, sample_data_model):			def test_sample_data(sample_data, sample_data_model):
	assert set(sample_data.keys()) == set(			assert set(sample_data.keys()) == set(
	[			[
	"content",			"content",
	"content_metadata",			"content_no_data",
	"skipped_content",			"skipped_content",
	"person",			"person",
	"directory",			"directory",
	"revision",			"revision",
	"release",			"release",
	"snapshot",			"snapshot",
	"origin",			"origin",
	"origin_visit",			"origin_visit",
	"fetcher",			"fetcher",
	"authority",			"authority",
	"origin_metadata",			"origin_metadata",
				"content_metadata",
	]			]
	)			)
	for object_type, objs in sample_data.items():			for object_type, objs in sample_data.items():
	for obj in objs:			for obj in objs:
	assert isinstance(obj, dict)			assert isinstance(obj, dict)

	if sample_data_model.get(object_type):			if sample_data_model.get(object_type):
	# metadata keys are missing because conversion is not possible yet			# metadata keys are missing because conversion is not possible yet
	assert len(objs) == len(sample_data_model[object_type])			assert len(objs) == len(sample_data_model[object_type])


	def test_sample_data_model(sample_data, sample_data_model):			def test_sample_data_model(sample_data, sample_data_model):
	assert set(sample_data_model.keys()) == set(			assert set(sample_data_model.keys()) == set(
	[			[
	"content",			"content",
	"content_metadata",			"content_no_data",
	"skipped_content",			"skipped_content",
	"person",			"person",
	"directory",			"directory",
	"revision",			"revision",
	"release",			"release",
	"snapshot",			"snapshot",
	"origin",			"origin",
	"origin_visit",			"origin_visit",
	"fetcher",			"fetcher",
	"authority",			"authority",
	"origin_metadata",			"origin_metadata",
				"content_metadata",
	]			]
	)			)

	for object_type, objs in sample_data_model.items():			for object_type, objs in sample_data_model.items():
	assert object_type in OBJECT_FACTORY			assert object_type in OBJECT_FACTORY

	for obj in objs:			for obj in objs:
	assert isinstance(obj, BaseModel)			assert isinstance(obj, BaseModel)
	Show All 12 Lines