Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/pytest_plugin.py
| Show All 18 Lines | |||||
| from swh.model.model import ( | from swh.model.model import ( | ||||
| BaseModel, | BaseModel, | ||||
| Content, | Content, | ||||
| Directory, | Directory, | ||||
| MetadataAuthority, | MetadataAuthority, | ||||
| MetadataFetcher, | MetadataFetcher, | ||||
| Origin, | Origin, | ||||
| OriginVisit, | OriginVisit, | ||||
| Person, | |||||
| RawExtrinsicMetadata, | RawExtrinsicMetadata, | ||||
| Release, | Release, | ||||
| Revision, | Revision, | ||||
| SkippedContent, | SkippedContent, | ||||
| Snapshot, | Snapshot, | ||||
| ) | ) | ||||
| from swh.storage import get_storage | from swh.storage import get_storage | ||||
| from swh.storage.tests.storage_data import data | from swh.storage.tests.storage_data import data | ||||
| ▲ Show 20 Lines • Show All 172 Lines • ▼ Show 20 Lines | def sample_data() -> Dict: | ||||
| Returns: | Returns: | ||||
| Dict of data (keys: content, directory, revision, release, person, | Dict of data (keys: content, directory, revision, release, person, | ||||
| origin) | origin) | ||||
| """ | """ | ||||
| return { | return { | ||||
| "content": [data.content, data.content2, data.content3], | "content": [data.content, data.content2, data.content3], | ||||
| "skipped_content": [data.skipped_content, data.skipped_content2], | "skipped_content": [data.skipped_content, data.skipped_content2], | ||||
| "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": [data.metadata_fetcher, data.metadata_fetcher2], | "fetcher": [data.metadata_fetcher, data.metadata_fetcher2], | ||||
| "authority": [data.metadata_authority, data.metadata_authority2], | "authority": [data.metadata_authority, data.metadata_authority2], | ||||
| Show All 10 Lines | def sample_data() -> Dict: | ||||
| } | } | ||||
| # 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, | ||||
| "skipped_content": SkippedContent.from_dict, | "skipped_content": SkippedContent.from_dict, | ||||
| "person": Person.from_dict, | |||||
| "directory": Directory.from_dict, | "directory": Directory.from_dict, | ||||
| "revision": Revision.from_dict, | "revision": Revision.from_dict, | ||||
| "release": Release.from_dict, | "release": Release.from_dict, | ||||
| "snapshot": Snapshot.from_dict, | "snapshot": Snapshot.from_dict, | ||||
| "origin": Origin.from_dict, | "origin": Origin.from_dict, | ||||
| "origin_visit": OriginVisit.from_dict, | "origin_visit": OriginVisit.from_dict, | ||||
| "fetcher": MetadataFetcher.from_dict, | "fetcher": MetadataFetcher.from_dict, | ||||
| "authority": MetadataAuthority.from_dict, | "authority": MetadataAuthority.from_dict, | ||||
| Show All 21 Lines | |||||