Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
| Show First 20 Lines • Show All 3,999 Lines • ▼ Show 20 Lines | class TestStorageGeneratedData: | ||||
| @pytest.mark.parametrize("limit", [1, 7, 10, 100, 1000]) | @pytest.mark.parametrize("limit", [1, 7, 10, 100, 1000]) | ||||
| def test_origin_list(self, swh_storage, swh_origins, limit): | def test_origin_list(self, swh_storage, swh_origins, limit): | ||||
| returned_origins = [] | returned_origins = [] | ||||
| page_token = None | page_token = None | ||||
| i = 0 | i = 0 | ||||
| while True: | while True: | ||||
| result = swh_storage.origin_list(page_token=page_token, limit=limit) | actual_page = swh_storage.origin_list(page_token=page_token, limit=limit) | ||||
| assert len(result["origins"]) <= limit | assert len(actual_page.results) <= limit | ||||
| returned_origins.extend(origin["url"] for origin in result["origins"]) | returned_origins.extend(actual_page.results) | ||||
| i += 1 | i += 1 | ||||
| page_token = result.get("next_page_token") | page_token = actual_page.next_page_token | ||||
| if page_token is None: | if page_token is None: | ||||
| assert i * limit >= len(swh_origins) | assert i * limit >= len(swh_origins) | ||||
| break | break | ||||
| else: | else: | ||||
| assert len(result["origins"]) == limit | assert len(actual_page.results) == limit | ||||
| expected_origins = [origin.url for origin in swh_origins] | assert sorted(returned_origins) == sorted(swh_origins) | ||||
| assert sorted(returned_origins) == sorted(expected_origins) | |||||
| def test_origin_count(self, swh_storage, sample_data): | def test_origin_count(self, swh_storage, sample_data): | ||||
| swh_storage.origin_add(sample_data.origins) | swh_storage.origin_add(sample_data.origins) | ||||
| assert swh_storage.origin_count("github") == 3 | assert swh_storage.origin_count("github") == 3 | ||||
| assert swh_storage.origin_count("gitlab") == 2 | assert swh_storage.origin_count("gitlab") == 2 | ||||
| assert swh_storage.origin_count(".*user.*", regexp=True) == 5 | assert swh_storage.origin_count(".*user.*", regexp=True) == 5 | ||||
| assert swh_storage.origin_count(".*user.*", regexp=False) == 0 | assert swh_storage.origin_count(".*user.*", regexp=False) == 0 | ||||
| ▲ Show 20 Lines • Show All 322 Lines • Show Last 20 Lines | |||||