Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/pytest_plugin.py
| Show First 20 Lines • Show All 215 Lines • ▼ Show 20 Lines | return { | ||||
| "skipped_content": [data.skipped_cont, data.skipped_cont2], | "skipped_content": [data.skipped_cont, data.skipped_cont2], | ||||
| "person": [data.person], | "person": [data.person], | ||||
| "directory": [data.dir2, data.dir, data.dir3, data.dir4], | "directory": [data.dir2, data.dir, data.dir3, data.dir4], | ||||
| "revision": [data.revision, data.revision2, data.revision3, data.revision4], | "revision": [data.revision, data.revision2, data.revision3, data.revision4], | ||||
| "release": [data.release, data.release2, data.release3], | "release": [data.release, data.release2, data.release3], | ||||
| "snapshot": [data.snapshot, data.empty_snapshot, data.complete_snapshot], | "snapshot": [data.snapshot, data.empty_snapshot, data.complete_snapshot], | ||||
| "origin": [data.origin, data.origin2], | "origin": [data.origin, data.origin2], | ||||
| "origin_visit": [data.origin_visit, data.origin_visit2, data.origin_visit3], | "origin_visit": [data.origin_visit, data.origin_visit2, data.origin_visit3], | ||||
| "fetcher": [ | "fetcher": [data.metadata_fetcher, data.metadata_fetcher2], | ||||
| m.to_dict() for m in [data.metadata_fetcher, data.metadata_fetcher2] | "authority": [data.metadata_authority, data.metadata_authority2], | ||||
| ], | |||||
| "authority": [ | |||||
| m.to_dict() for m in [data.metadata_authority, data.metadata_authority2] | |||||
| ], | |||||
| "origin_metadata": [ | "origin_metadata": [ | ||||
| m.to_dict() | |||||
| for m in [ | |||||
| data.origin_metadata, | data.origin_metadata, | ||||
| data.origin_metadata2, | data.origin_metadata2, | ||||
| data.origin_metadata3, | data.origin_metadata3, | ||||
| ] | |||||
| ], | ], | ||||
| "content_metadata": [ | "content_metadata": [ | ||||
| m.to_dict() | |||||
| for m in [ | |||||
| data.content_metadata, | data.content_metadata, | ||||
| data.content_metadata2, | data.content_metadata2, | ||||
| data.content_metadata3, | data.content_metadata3, | ||||
| ] | |||||
| ], | ], | ||||
| } | } | ||||
| # FIXME: Add the metadata keys when we can (right now, we cannot as the data model | # FIXME: Add the metadata keys when we can (right now, we cannot as the data model | ||||
| # changed but not the endpoints yet) | # changed but not the endpoints yet) | ||||
| OBJECT_FACTORY = { | OBJECT_FACTORY = { | ||||
| "content": Content.from_dict, | "content": Content.from_dict, | ||||
| Show All 18 Lines | def sample_data_model(sample_data) -> Dict[str, Iterable[BaseModel]]: | ||||
| """Pre-defined sample storage object model to manipulate | """Pre-defined sample storage object model to manipulate | ||||
| Returns: | Returns: | ||||
| Dict of data (keys: content, directory, revision, release, person, origin, ...) | Dict of data (keys: content, directory, revision, release, person, origin, ...) | ||||
| values list of object data model with the corresponding types | values list of object data model with the corresponding types | ||||
| """ | """ | ||||
| return { | return { | ||||
| object_type: [convert_fn(obj) for obj in sample_data[object_type]] | object_type: [ | ||||
| convert_fn(obj) if isinstance(obj, dict) else obj | |||||
| for obj in sample_data[object_type] | |||||
| ] | |||||
| for object_type, convert_fn in OBJECT_FACTORY.items() | for object_type, convert_fn in OBJECT_FACTORY.items() | ||||
| } | } | ||||