Changeset View
Changeset View
Standalone View
Standalone View
swh/indexer/tests/tasks.py
- This file was added.
from swh.scheduler.celery_backend.config import app | |||||
from swh.indexer.metadata import ( | |||||
OriginMetadataIndexer, RevisionMetadataIndexer | |||||
) | |||||
from .test_metadata import ContentMetadataTestIndexer | |||||
from .test_utils import BASE_TEST_CONFIG | |||||
class RevisionMetadataTestIndexer(RevisionMetadataIndexer): | |||||
"""Specific indexer whose configuration is enough to satisfy the | |||||
indexing tests. | |||||
""" | |||||
ContentMetadataIndexer = ContentMetadataTestIndexer | |||||
def parse_config_file(self, *args, **kwargs): | |||||
return { | |||||
**BASE_TEST_CONFIG, | |||||
'tools': { | |||||
'name': 'swh-metadata-detector', | |||||
'version': '0.0.2', | |||||
'configuration': { | |||||
'type': 'local', | |||||
'context': 'NpmMapping' | |||||
} | |||||
} | |||||
} | |||||
class OriginMetadataTestIndexer(OriginMetadataIndexer): | |||||
def parse_config_file(self, *args, **kwargs): | |||||
return { | |||||
**BASE_TEST_CONFIG, | |||||
'tools': [] | |||||
} | |||||
@app.task | |||||
def revision_metadata(*args, **kwargs): | |||||
indexer = RevisionMetadataTestIndexer() | |||||
indexer.run(*args, **kwargs) | |||||
print('REV RESULT=', indexer.results) | |||||
@app.task | |||||
def origin_intrinsic_metadata(*args, **kwargs): | |||||
indexer = OriginMetadataTestIndexer() | |||||
indexer.run(*args, **kwargs) |