Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
| Show First 20 Lines • Show All 3,413 Lines • ▼ Show 20 Lines | def test_metadata_provider_get_by(self): | ||||
| actual_provider = self.storage.metadata_provider_get_by({ | actual_provider = self.storage.metadata_provider_get_by({ | ||||
| 'provider_name': self.provider['name'], | 'provider_name': self.provider['name'], | ||||
| 'provider_url': self.provider['url'] | 'provider_url': self.provider['url'] | ||||
| }) | }) | ||||
| # then | # then | ||||
| self.assertTrue(provider_id, actual_provider['id']) | self.assertTrue(provider_id, actual_provider['id']) | ||||
| def test_origin_metadata_add(self): | @given(strategies.booleans()) | ||||
| def test_origin_metadata_add(self, use_url): | |||||
| self.reset_storage() | |||||
| # given | # given | ||||
| origin_id = self.storage.origin_add([self.origin])[0]['id'] | origin = self.storage.origin_add([self.origin])[0] | ||||
| origin_id = origin['id'] | |||||
| if use_url: | |||||
| origin = origin['url'] | |||||
| else: | |||||
| origin = origin['id'] | |||||
| origin_metadata0 = list(self.storage.origin_metadata_get_by( | origin_metadata0 = list(self.storage.origin_metadata_get_by( | ||||
| origin_id)) | origin)) | ||||
| self.assertTrue(len(origin_metadata0) == 0) | self.assertEqual(len(origin_metadata0), 0, origin_metadata0) | ||||
| tools = self.storage.tool_add([self.metadata_tool]) | tools = self.storage.tool_add([self.metadata_tool]) | ||||
| tool = tools[0] | tool = tools[0] | ||||
| self.storage.metadata_provider_add( | self.storage.metadata_provider_add( | ||||
| self.provider['name'], | self.provider['name'], | ||||
| self.provider['type'], | self.provider['type'], | ||||
| self.provider['url'], | self.provider['url'], | ||||
| self.provider['metadata']) | self.provider['metadata']) | ||||
| provider = self.storage.metadata_provider_get_by({ | provider = self.storage.metadata_provider_get_by({ | ||||
| 'provider_name': self.provider['name'], | 'provider_name': self.provider['name'], | ||||
| 'provider_url': self.provider['url'] | 'provider_url': self.provider['url'] | ||||
| }) | }) | ||||
| # when adding for the same origin 2 metadatas | # when adding for the same origin 2 metadatas | ||||
| self.storage.origin_metadata_add( | self.storage.origin_metadata_add( | ||||
| origin_id, | origin, | ||||
| self.origin_metadata['discovery_date'], | self.origin_metadata['discovery_date'], | ||||
| provider['id'], | provider['id'], | ||||
| tool['id'], | tool['id'], | ||||
| self.origin_metadata['metadata']) | self.origin_metadata['metadata']) | ||||
| self.storage.origin_metadata_add( | self.storage.origin_metadata_add( | ||||
| origin_id, | origin, | ||||
| '2015-01-01 23:00:00+00', | '2015-01-01 23:00:00+00', | ||||
| provider['id'], | provider['id'], | ||||
| tool['id'], | tool['id'], | ||||
| self.origin_metadata2['metadata']) | self.origin_metadata2['metadata']) | ||||
| actual_om = list(self.storage.origin_metadata_get_by( | actual_om = list(self.storage.origin_metadata_get_by( | ||||
| origin_id)) | origin)) | ||||
| # then | # then | ||||
| self.assertCountEqual( | self.assertCountEqual( | ||||
| [item['origin_id'] for item in actual_om], | [item['origin_id'] for item in actual_om], | ||||
| [origin_id, origin_id]) | [origin_id, origin_id]) | ||||
| def test_origin_metadata_get(self): | def test_origin_metadata_get(self): | ||||
| # given | # given | ||||
| origin_id = self.storage.origin_add([self.origin])[0]['id'] | origin_id = self.storage.origin_add([self.origin])[0]['id'] | ||||
| ▲ Show 20 Lines • Show All 546 Lines • Show Last 20 Lines | |||||