Changeset View
Changeset View
Standalone View
Standalone View
swh/objstorage/tests/test_objstorage_random_generator.py
| # Copyright (C) 2019-2020 The Software Heritage developers | # Copyright (C) 2019-2020 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| from collections.abc import Iterator | from collections.abc import Iterator | ||||
| from swh.objstorage.factory import get_objstorage | from swh.objstorage.factory import get_objstorage | ||||
| def test_random_generator_objstorage(): | def test_random_generator_objstorage(): | ||||
| sto = get_objstorage("random", {}) | sto = get_objstorage("random") | ||||
| assert sto | assert sto | ||||
| blobs = [sto.get(None) for i in range(100)] | blobs = [sto.get(None) for i in range(100)] | ||||
| lengths = [len(x) for x in blobs] | lengths = [len(x) for x in blobs] | ||||
| assert max(lengths) <= 55056238 | assert max(lengths) <= 55056238 | ||||
| def test_random_generator_objstorage_list_content(): | def test_random_generator_objstorage_list_content(): | ||||
| sto = get_objstorage("random", {"total": 100}) | sto = get_objstorage("random", total=100) | ||||
| assert isinstance(sto.list_content(), Iterator) | assert isinstance(sto.list_content(), Iterator) | ||||
| assert list(sto.list_content()) == [b"%d" % i for i in range(1, 101)] | assert list(sto.list_content()) == [b"%d" % i for i in range(1, 101)] | ||||
| assert list(sto.list_content(limit=10)) == [b"%d" % i for i in range(1, 11)] | assert list(sto.list_content(limit=10)) == [b"%d" % i for i in range(1, 11)] | ||||
| assert list(sto.list_content(last_obj_id=b"10", limit=10)) == [ | assert list(sto.list_content(last_obj_id=b"10", limit=10)) == [ | ||||
| b"%d" % i for i in range(11, 21) | b"%d" % i for i in range(11, 21) | ||||
| ] | ] | ||||
| def test_random_generator_objstorage_total(): | def test_random_generator_objstorage_total(): | ||||
| sto = get_objstorage("random", {"total": 5}) | sto = get_objstorage("random", total=5) | ||||
| assert len([x for x in sto]) == 5 | assert len([x for x in sto]) == 5 | ||||
| def test_random_generator_objstorage_size(): | def test_random_generator_objstorage_size(): | ||||
| sto = get_objstorage("random", {"filesize": 10}) | sto = get_objstorage("random", filesize=10) | ||||
| for i in range(10): | for i in range(10): | ||||
| assert len(sto.get(None)) == 10 | assert len(sto.get(None)) == 10 | ||||