Changeset View
Changeset View
Standalone View
Standalone View
swh/indexer/tests/test_origin_metadata.py
| Show All 10 Lines | |||||
| from swh.indexer.metadata import OriginMetadataIndexer | from swh.indexer.metadata import OriginMetadataIndexer | ||||
| from swh.indexer.tests.test_utils import MockObjStorage, MockStorage | from swh.indexer.tests.test_utils import MockObjStorage, MockStorage | ||||
| from swh.indexer.tests.test_utils import MockIndexerStorage | from swh.indexer.tests.test_utils import MockIndexerStorage | ||||
| from swh.indexer.tests.test_origin_head import OriginHeadTestIndexer | from swh.indexer.tests.test_origin_head import OriginHeadTestIndexer | ||||
| from swh.indexer.tests.test_metadata import RevisionMetadataTestIndexer | from swh.indexer.tests.test_metadata import RevisionMetadataTestIndexer | ||||
| from swh.scheduler.tests.scheduler_testing import SchedulerTestFixture | from swh.scheduler.tests.scheduler_testing import SchedulerTestFixture | ||||
| from swh.model.hashutil import hash_to_bytes | |||||
| class OriginMetadataTestIndexer(OriginMetadataIndexer): | class OriginMetadataTestIndexer(OriginMetadataIndexer): | ||||
| def prepare(self): | def prepare(self): | ||||
| self.config = { | self.config = { | ||||
| 'storage': { | 'storage': { | ||||
| 'cls': 'remote', | 'cls': 'remote', | ||||
| 'args': { | 'args': { | ||||
| 'url': 'http://localhost:9999', | 'url': 'http://localhost:9999', | ||||
| ▲ Show 20 Lines • Show All 72 Lines • ▼ Show 20 Lines | def test_pipeline(self): | ||||
| 'description': | 'description': | ||||
| 'Tiny web service for parsing yarn.lock files', | 'Tiny web service for parsing yarn.lock files', | ||||
| 'codemeta:issueTracker': | 'codemeta:issueTracker': | ||||
| 'https://github.com/librariesio/yarn-parser/issues', | 'https://github.com/librariesio/yarn-parser/issues', | ||||
| 'name': 'yarn-parser', | 'name': 'yarn-parser', | ||||
| 'keywords': ['yarn', 'parse', 'lock', 'dependencies'], | 'keywords': ['yarn', 'parse', 'lock', 'dependencies'], | ||||
| } | } | ||||
| rev_metadata = { | rev_metadata = { | ||||
| 'id': '8dbb6aeb036e7fd80664eb8bfd1507881af1ba9f', | 'id': hash_to_bytes('8dbb6aeb036e7fd80664eb8bfd1507881af1ba9f'), | ||||
| 'translated_metadata': metadata, | 'translated_metadata': metadata, | ||||
| 'indexer_configuration_id': 7, | 'indexer_configuration_id': 7, | ||||
| } | } | ||||
| origin_metadata = { | origin_metadata = { | ||||
| 'origin_id': 54974445, | 'origin_id': 54974445, | ||||
| 'from_revision': '8dbb6aeb036e7fd80664eb8bfd1507881af1ba9f', | 'from_revision': hash_to_bytes( | ||||
| '8dbb6aeb036e7fd80664eb8bfd1507881af1ba9f'), | |||||
| 'metadata': metadata, | 'metadata': metadata, | ||||
| 'indexer_configuration_id': 7, | 'indexer_configuration_id': 7, | ||||
| } | } | ||||
| expected_results = [ | expected_results = [ | ||||
| ('origin_intrinsic_metadata', True, [origin_metadata]), | ('origin_intrinsic_metadata', True, [origin_metadata]), | ||||
| ('revision_metadata', True, [rev_metadata])] | ('revision_metadata', True, [rev_metadata])] | ||||
| results = list(indexer.idx_storage.added_data) | results = list(indexer.idx_storage.added_data) | ||||
| self.assertCountEqual(expected_results, results) | self.assertCountEqual(expected_results, results) | ||||