Changeset View
Changeset View
Standalone View
Standalone View
swh/web/tests/common/test_service.py
Show First 20 Lines • Show All 857 Lines • ▼ Show 20 Lines | def test_lookup_directory_through_revision_ok_with_data(self, revision): | ||||
) | ) | ||||
@pytest.mark.origin_id | @pytest.mark.origin_id | ||||
@given(new_origins(20)) | @given(new_origins(20)) | ||||
def test_lookup_origins(self, new_origins): | def test_lookup_origins(self, new_origins): | ||||
nb_origins = len(new_origins) | nb_origins = len(new_origins) | ||||
expected_origins = self.storage.origin_add(new_origins) | expected_origins = self.storage.origin_add(new_origins) | ||||
expected_origins.sort(key=lambda orig: orig['id']) | |||||
origin_from_idx = random.randint(1, nb_origins-1) - 1 | origin_from_idx = random.randint(1, nb_origins-1) - 1 | ||||
origin_from = expected_origins[origin_from_idx]['id'] | origin_from = expected_origins[origin_from_idx]['id'] | ||||
max_origin_idx = expected_origins[-1]['id'] | max_origin_idx = expected_origins[-1]['id'] | ||||
origin_count = random.randint(1, max_origin_idx - origin_from) | origin_count = random.randint(1, max_origin_idx - origin_from) | ||||
actual_origins = list(service.lookup_origins(origin_from, | actual_origins = list(service.lookup_origins(origin_from, | ||||
origin_count)) | origin_count)) | ||||
expected_origins = list(self.storage.origin_get_range(origin_from, | expected_origins = list(self.storage.origin_get_range(origin_from, | ||||
origin_count)) | origin_count)) | ||||
self.assertEqual(actual_origins, expected_origins) | self.assertEqual(actual_origins, expected_origins) |