1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
import pytest
from werkzeug.exceptions import NotFound
from flask_mongoengine import ListFieldPagination, Pagination
def test_queryset_paginator(app, todo):
Todo = todo
for i in range(42):
Todo(title="post: %s" % i).save()
with pytest.raises(NotFound):
Pagination(iterable=Todo.objects, page=0, per_page=10)
with pytest.raises(NotFound):
Pagination(iterable=Todo.objects, page=6, per_page=10)
paginator = Pagination(Todo.objects, 1, 10)
_test_paginator(paginator)
def test_paginate_plain_list():
with pytest.raises(NotFound):
Pagination(iterable=range(1, 42), page=0, per_page=10)
with pytest.raises(NotFound):
Pagination(iterable=range(1, 42), page=6, per_page=10)
paginator = Pagination(range(1, 42), 1, 10)
_test_paginator(paginator)
def test_list_field_pagination(app, todo):
Todo = todo
comments = ["comment: %s" % i for i in range(42)]
todo = Todo(
title="todo has comments",
comments=comments,
comment_count=len(comments),
).save()
# Check without providing a total
paginator = ListFieldPagination(Todo.objects, todo.id, "comments", 1, 10)
_test_paginator(paginator)
# Check with providing a total (saves a query)
paginator = ListFieldPagination(
Todo.objects, todo.id, "comments", 1, 10, todo.comment_count
)
_test_paginator(paginator)
paginator = todo.paginate_field("comments", 1, 10)
_test_paginator(paginator)
def _test_paginator(paginator):
assert 5 == paginator.pages
assert [1, 2, 3, 4, 5] == list(paginator.iter_pages())
for i in [1, 2, 3, 4, 5]:
if i == 1:
assert not paginator.has_prev
with pytest.raises(NotFound):
paginator.prev()
else:
assert paginator.has_prev
if i == 5:
assert not paginator.has_next
with pytest.raises(NotFound):
paginator.next()
else:
assert paginator.has_next
if i == 3:
assert [None, 2, 3, 4, None] == list(paginator.iter_pages(0, 1, 1, 0))
assert i == paginator.page
assert i - 1 == paginator.prev_num
assert i + 1 == paginator.next_num
# Paginate to the next page
if i < 5:
paginator = paginator.next()
|
