51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from data.database import Role
|
||
|
|
from data.model.modelutil import paginate
|
||
|
|
from test.fixtures import *
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('page_size', [
|
||
|
|
10,
|
||
|
|
20,
|
||
|
|
50,
|
||
|
|
100,
|
||
|
|
200,
|
||
|
|
500,
|
||
|
|
1000,
|
||
|
|
])
|
||
|
|
@pytest.mark.parametrize('descending', [
|
||
|
|
False,
|
||
|
|
True,
|
||
|
|
])
|
||
|
|
def test_paginate(page_size, descending, initialized_db):
|
||
|
|
# Add a bunch of rows into a test table (`Role`).
|
||
|
|
for i in range(0, 522):
|
||
|
|
Role.create(name='testrole%s' % i)
|
||
|
|
|
||
|
|
query = Role.select().where(Role.name ** 'testrole%')
|
||
|
|
all_matching_roles = list(query)
|
||
|
|
assert len(all_matching_roles) == 522
|
||
|
|
|
||
|
|
# Paginate a query to lookup roles.
|
||
|
|
collected = []
|
||
|
|
page_token = None
|
||
|
|
while True:
|
||
|
|
results, page_token = paginate(query, Role, limit=page_size, descending=descending,
|
||
|
|
page_token=page_token)
|
||
|
|
assert len(results) <= page_size
|
||
|
|
collected.extend(results)
|
||
|
|
|
||
|
|
if page_token is None:
|
||
|
|
break
|
||
|
|
|
||
|
|
assert len(results) == page_size
|
||
|
|
|
||
|
|
for index, result in enumerate(results[1:]):
|
||
|
|
if descending:
|
||
|
|
assert result.id < results[index].id
|
||
|
|
else:
|
||
|
|
assert result.id > results[index].id
|
||
|
|
|
||
|
|
assert len(collected) == len(all_matching_roles)
|
||
|
|
assert {c.id for c in collected} == {a.id for a in all_matching_roles}
|