Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/core/tests/test_loader.py
| Show All 24 Lines | class DummyLoader: | ||||
| def store_data(self): | def store_data(self): | ||||
| pass | pass | ||||
| def prepare_origin_visit(self): | def prepare_origin_visit(self): | ||||
| origin = self.storage.origin_get( | origin = self.storage.origin_get( | ||||
| self._test_prepare_origin_visit_data['origin']) | self._test_prepare_origin_visit_data['origin']) | ||||
| self.origin = origin | self.origin = origin | ||||
| self.origin_id = origin['id'] | |||||
| self.origin_url = origin['url'] | self.origin_url = origin['url'] | ||||
| self.visit_date = datetime.datetime.utcnow() | self.visit_date = datetime.datetime.utcnow() | ||||
| self.storage.origin_visit_add(origin['id'], self.visit_date) | self.storage.origin_visit_add(origin['id'], self.visit_date) | ||||
| def parse_config_file(self, *args, **kwargs): | def parse_config_file(self, *args, **kwargs): | ||||
| return { | return { | ||||
| 'storage': { | 'storage': { | ||||
| 'cls': 'memory', | 'cls': 'memory', | ||||
| ▲ Show 20 Lines • Show All 238 Lines • ▼ Show 20 Lines | class CoreBufferedLoaderTest(DummyBaseLoaderTest): | ||||
| def test_origin_metadata(self): | def test_origin_metadata(self): | ||||
| self.loader.load() | self.loader.load() | ||||
| provider_id = self.loader.send_provider(self.in_provider) | provider_id = self.loader.send_provider(self.in_provider) | ||||
| tool_id = self.loader.send_tool(self.in_tool) | tool_id = self.loader.send_tool(self.in_tool) | ||||
| self.loader.send_origin_metadata( | self.loader.send_origin_metadata( | ||||
| self.loader.origin_id, self.loader.visit_date, provider_id, | self.loader.visit_date, provider_id, | ||||
| tool_id, {'test_metadata': 'foobar'}) | tool_id, {'test_metadata': 'foobar'}) | ||||
| self.assertOriginMetadataContains( | self.assertOriginMetadataContains( | ||||
| self.in_origin['type'], self.in_origin['url'], | self.in_origin['type'], self.in_origin['url'], | ||||
| {'test_metadata': 'foobar'}) | {'test_metadata': 'foobar'}) | ||||
| with self.assertRaises(AssertionError): | with self.assertRaises(AssertionError): | ||||
| self.assertOriginMetadataContains( | self.assertOriginMetadataContains( | ||||
| Show All 26 Lines | |||||