Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_pytest_plugin.py
Show All 23 Lines | assert set(sample_data.keys()) == set( | ||||
"origin_visit", | "origin_visit", | ||||
"fetcher", | "fetcher", | ||||
"authority", | "authority", | ||||
"origin_metadata", | "origin_metadata", | ||||
"content_metadata", | "content_metadata", | ||||
] | ] | ||||
) | ) | ||||
for object_type, objs in sample_data.items(): | for object_type, objs in sample_data.items(): | ||||
if object_type in [ | |||||
"content", | |||||
"skipped_content", | |||||
"directory", | |||||
"revision", | |||||
"origin", | |||||
"origin_visit", | |||||
"fetcher", | |||||
"authority", | |||||
"origin_metadata", | |||||
"content_metadata", | |||||
"release", | |||||
]: | |||||
type_ = BaseModel | |||||
else: | |||||
type_ = dict | |||||
for obj in objs: | for obj in objs: | ||||
assert isinstance(obj, type_) | assert isinstance(obj, BaseModel) | ||||
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", | ||||
"skipped_content", | "skipped_content", | ||||
"directory", | "directory", | ||||
Show All 27 Lines |