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
|
from django.core.paginator import EmptyPage, PageNotAnInteger, Paginator
from django.test import TestCase
from django_tables2 import LazyPaginator
class FakeQuerySet:
objects = range(1, 10**6)
def count(self):
raise AssertionError("LazyPaginator should not call QuerySet.count()")
def __getitem__(self, key):
return self.objects[key]
def __iter__(self):
yield next(self.objects)
class LazyPaginatorTest(TestCase):
def test_compare_to_default_paginator(self):
objects = list(range(1, 1000))
paginator = Paginator(objects, 10)
lazy_paginator = LazyPaginator(objects, 10)
self.assertEqual(paginator.page(1).object_list, lazy_paginator.page(1).object_list)
self.assertEqual(paginator.page(10).object_list, lazy_paginator.page(10).object_list)
self.assertEqual(paginator.page(100).object_list, lazy_paginator.page(100).object_list)
def test_no_count_call(self):
paginator = LazyPaginator(FakeQuerySet(), 10)
# num_pages initially is None, but is page_number + 1 after requesting a page.
self.assertEqual(paginator.num_pages, None)
paginator.page(1)
self.assertEqual(paginator.num_pages, 2)
paginator.page(3)
self.assertEqual(paginator.num_pages, 4)
paginator.page(1.0)
# and again decreases when a lower page nu
self.assertEqual(paginator.num_pages, 2)
with self.assertRaisesMessage(PageNotAnInteger, "That page number is not an integer"):
paginator.page(1.5)
with self.assertRaisesMessage(EmptyPage, "That page number is less than 1"):
paginator.page(-1)
with self.assertRaises(NotImplementedError):
paginator.count()
with self.assertRaises(NotImplementedError):
paginator.page_range()
# last page
last_page_number = 10**5
paginator.page(last_page_number)
with self.assertRaisesMessage(EmptyPage, "That page contains no results"):
paginator.page(last_page_number + 1)
def test_lookahead(self):
objects = list(range(1, 1000))
paginator = LazyPaginator(objects, 10, look_ahead=3)
self.assertEqual(paginator.look_ahead, 3)
self.assertEqual(paginator.page(1).object_list, list(range(1, 11)))
self.assertEqual(paginator.num_pages, 4)
self.assertEqual(paginator.page(98).object_list, list(range(971, 981)))
self.assertEqual(paginator.num_pages, 100)
def test_number_is_none(self):
"""
When number=None is supplied, the paginator should serve its first page.
"""
objects = list(range(1, 1000))
paginator = LazyPaginator(objects, 10, look_ahead=3)
self.assertEqual(paginator.page(None).object_list, list(range(1, 11)))
self.assertEqual(paginator.num_pages, 4)
objects = list(range(1, 20))
paginator = LazyPaginator(objects, 10, look_ahead=3)
self.assertEqual(paginator.page(None).object_list, list(range(1, 11)))
|
