Changeset View
Changeset View
Standalone View
Standalone View
swh/indexer/tests/tasks.py
| from celery import current_app as app | from celery import current_app as app | ||||
| from swh.indexer.metadata import ( | from swh.indexer.metadata import ( | ||||
| OriginMetadataIndexer, RevisionMetadataIndexer | OriginMetadataIndexer, RevisionMetadataIndexer | ||||
| ) | ) | ||||
| from .test_origin_head import OriginHeadTestIndexer | |||||
| from .test_metadata import ContentMetadataTestIndexer | from .test_metadata import ContentMetadataTestIndexer | ||||
| from .utils import BASE_TEST_CONFIG | from .utils import BASE_TEST_CONFIG | ||||
| class RevisionMetadataTestIndexer(RevisionMetadataIndexer): | class RevisionMetadataTestIndexer(RevisionMetadataIndexer): | ||||
| """Specific indexer whose configuration is enough to satisfy the | """Specific indexer whose configuration is enough to satisfy the | ||||
| indexing tests. | indexing tests. | ||||
| """ | """ | ||||
| Show All 15 Lines | |||||
| class OriginMetadataTestIndexer(OriginMetadataIndexer): | class OriginMetadataTestIndexer(OriginMetadataIndexer): | ||||
| def parse_config_file(self, *args, **kwargs): | def parse_config_file(self, *args, **kwargs): | ||||
| return { | return { | ||||
| **BASE_TEST_CONFIG, | **BASE_TEST_CONFIG, | ||||
| 'tools': [] | 'tools': [] | ||||
| } | } | ||||
| def _prepare_sub_indexers(self): | |||||
| self.origin_head_indexer = OriginHeadTestIndexer() | |||||
| self.revision_metadata_indexer = RevisionMetadataTestIndexer() | |||||
| @app.task | @app.task | ||||
| def revision_metadata(*args, **kwargs): | def revision_metadata(*args, **kwargs): | ||||
| indexer = RevisionMetadataTestIndexer() | indexer = RevisionMetadataTestIndexer() | ||||
| indexer.run(*args, **kwargs) | indexer.run(*args, **kwargs) | ||||
| print('REV RESULT=', indexer.results) | print('REV RESULT=', indexer.results) | ||||
| @app.task | @app.task | ||||
| def origin_intrinsic_metadata(*args, **kwargs): | def origin_intrinsic_metadata(*args, **kwargs): | ||||
| indexer = OriginMetadataTestIndexer() | indexer = OriginMetadataTestIndexer() | ||||
| indexer.run(*args, **kwargs) | indexer.run(*args, **kwargs) | ||||