Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/algos/test_origin.py
| Show All 20 Lines | |||||
| @pytest.fixture | @pytest.fixture | ||||
| def swh_storage_backend_config(): | def swh_storage_backend_config(): | ||||
| yield {"cls": "validate", "storage": {"cls": "memory",}} | yield {"cls": "validate", "storage": {"cls": "memory",}} | ||||
| def test_iter_origins(swh_storage): | def test_iter_origins(swh_storage): | ||||
| origins = swh_storage.origin_add([{"url": "bar"}, {"url": "qux"}, {"url": "quuz"},]) | origins = [ | ||||
| {"url": "bar"}, | |||||
| {"url": "qux"}, | |||||
| {"url": "quuz"}, | |||||
| ] | |||||
| assert swh_storage.origin_add(origins) == {"origin:add": 3} | |||||
| assert_list_eq(iter_origins(swh_storage), origins) | assert_list_eq(iter_origins(swh_storage), origins) | ||||
| assert_list_eq(iter_origins(swh_storage, batch_size=1), origins) | assert_list_eq(iter_origins(swh_storage, batch_size=1), origins) | ||||
| assert_list_eq(iter_origins(swh_storage, batch_size=2), origins) | assert_list_eq(iter_origins(swh_storage, batch_size=2), origins) | ||||
| for i in range(1, 5): | for i in range(1, 5): | ||||
| assert_list_eq(iter_origins(swh_storage, origin_from=i + 1), origins[i:], i) | assert_list_eq(iter_origins(swh_storage, origin_from=i + 1), origins[i:], i) | ||||
| assert_list_eq( | assert_list_eq( | ||||
| ▲ Show 20 Lines • Show All 269 Lines • Show Last 20 Lines | |||||