[1.2.X] Fixed #14444 -- Convert the pagination doctests to unittests. We have always been at war with doctests. Thanks to Gabriel Hurley for the patch. Backport of [14152].
git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@14153 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
parent
efc7524236
commit
01d1dd6935
|
@ -8,173 +8,10 @@ objects into easily readable pages.
|
|||
|
||||
from django.db import models
|
||||
|
||||
|
||||
class Article(models.Model):
|
||||
headline = models.CharField(max_length=100, default='Default headline')
|
||||
pub_date = models.DateTimeField()
|
||||
|
||||
def __unicode__(self):
|
||||
return self.headline
|
||||
|
||||
__test__ = {'API_TESTS':"""
|
||||
# Prepare a list of objects for pagination.
|
||||
>>> from datetime import datetime
|
||||
>>> for x in range(1, 10):
|
||||
... a = Article(headline='Article %s' % x, pub_date=datetime(2005, 7, 29))
|
||||
... a.save()
|
||||
|
||||
##################
|
||||
# Paginator/Page #
|
||||
##################
|
||||
|
||||
>>> from django.core.paginator import Paginator
|
||||
>>> paginator = Paginator(Article.objects.all(), 5)
|
||||
>>> paginator.count
|
||||
9
|
||||
>>> paginator.num_pages
|
||||
2
|
||||
>>> paginator.page_range
|
||||
[1, 2]
|
||||
|
||||
# Get the first page.
|
||||
>>> p = paginator.page(1)
|
||||
>>> p
|
||||
<Page 1 of 2>
|
||||
>>> p.object_list
|
||||
[<Article: Article 1>, <Article: Article 2>, <Article: Article 3>, <Article: Article 4>, <Article: Article 5>]
|
||||
>>> p.has_next()
|
||||
True
|
||||
>>> p.has_previous()
|
||||
False
|
||||
>>> p.has_other_pages()
|
||||
True
|
||||
>>> p.next_page_number()
|
||||
2
|
||||
>>> p.previous_page_number()
|
||||
0
|
||||
>>> p.start_index()
|
||||
1
|
||||
>>> p.end_index()
|
||||
5
|
||||
|
||||
# Get the second page.
|
||||
>>> p = paginator.page(2)
|
||||
>>> p
|
||||
<Page 2 of 2>
|
||||
>>> p.object_list
|
||||
[<Article: Article 6>, <Article: Article 7>, <Article: Article 8>, <Article: Article 9>]
|
||||
>>> p.has_next()
|
||||
False
|
||||
>>> p.has_previous()
|
||||
True
|
||||
>>> p.has_other_pages()
|
||||
True
|
||||
>>> p.next_page_number()
|
||||
3
|
||||
>>> p.previous_page_number()
|
||||
1
|
||||
>>> p.start_index()
|
||||
6
|
||||
>>> p.end_index()
|
||||
9
|
||||
|
||||
# Empty pages raise EmptyPage.
|
||||
>>> paginator.page(0)
|
||||
Traceback (most recent call last):
|
||||
...
|
||||
EmptyPage: ...
|
||||
>>> paginator.page(3)
|
||||
Traceback (most recent call last):
|
||||
...
|
||||
EmptyPage: ...
|
||||
|
||||
# Empty paginators with allow_empty_first_page=True.
|
||||
>>> paginator = Paginator(Article.objects.filter(id=0), 5, allow_empty_first_page=True)
|
||||
>>> paginator.count
|
||||
0
|
||||
>>> paginator.num_pages
|
||||
1
|
||||
>>> paginator.page_range
|
||||
[1]
|
||||
|
||||
# Empty paginators with allow_empty_first_page=False.
|
||||
>>> paginator = Paginator(Article.objects.filter(id=0), 5, allow_empty_first_page=False)
|
||||
>>> paginator.count
|
||||
0
|
||||
>>> paginator.num_pages
|
||||
0
|
||||
>>> paginator.page_range
|
||||
[]
|
||||
|
||||
# Paginators work with regular lists/tuples, too -- not just with QuerySets.
|
||||
>>> paginator = Paginator([1, 2, 3, 4, 5, 6, 7, 8, 9], 5)
|
||||
>>> paginator.count
|
||||
9
|
||||
>>> paginator.num_pages
|
||||
2
|
||||
>>> paginator.page_range
|
||||
[1, 2]
|
||||
|
||||
# Get the first page.
|
||||
>>> p = paginator.page(1)
|
||||
>>> p
|
||||
<Page 1 of 2>
|
||||
>>> p.object_list
|
||||
[1, 2, 3, 4, 5]
|
||||
>>> p.has_next()
|
||||
True
|
||||
>>> p.has_previous()
|
||||
False
|
||||
>>> p.has_other_pages()
|
||||
True
|
||||
>>> p.next_page_number()
|
||||
2
|
||||
>>> p.previous_page_number()
|
||||
0
|
||||
>>> p.start_index()
|
||||
1
|
||||
>>> p.end_index()
|
||||
5
|
||||
|
||||
# Paginator can be passed other objects with a count() method.
|
||||
>>> class CountContainer:
|
||||
... def count(self):
|
||||
... return 42
|
||||
>>> paginator = Paginator(CountContainer(), 10)
|
||||
>>> paginator.count
|
||||
42
|
||||
>>> paginator.num_pages
|
||||
5
|
||||
>>> paginator.page_range
|
||||
[1, 2, 3, 4, 5]
|
||||
|
||||
# Paginator can be passed other objects that implement __len__.
|
||||
>>> class LenContainer:
|
||||
... def __len__(self):
|
||||
... return 42
|
||||
>>> paginator = Paginator(LenContainer(), 10)
|
||||
>>> paginator.count
|
||||
42
|
||||
>>> paginator.num_pages
|
||||
5
|
||||
>>> paginator.page_range
|
||||
[1, 2, 3, 4, 5]
|
||||
|
||||
|
||||
##################
|
||||
# Orphan support #
|
||||
##################
|
||||
|
||||
# Add a few more records to test out the orphans feature.
|
||||
>>> for x in range(10, 13):
|
||||
... Article(headline="Article %s" % x, pub_date=datetime(2006, 10, 6)).save()
|
||||
|
||||
# With orphans set to 3 and 10 items per page, we should get all 12 items on a single page.
|
||||
>>> paginator = Paginator(Article.objects.all(), 10, orphans=3)
|
||||
>>> paginator.num_pages
|
||||
1
|
||||
|
||||
# With orphans only set to 1, we should get two pages.
|
||||
>>> paginator = Paginator(Article.objects.all(), 10, orphans=1)
|
||||
>>> paginator.num_pages
|
||||
2
|
||||
"""}
|
||||
|
|
|
@ -0,0 +1,132 @@
|
|||
from datetime import datetime
|
||||
from operator import attrgetter
|
||||
|
||||
from django.core.paginator import Paginator, InvalidPage, EmptyPage
|
||||
from django.test import TestCase
|
||||
|
||||
from models import Article
|
||||
|
||||
|
||||
class CountContainer(object):
|
||||
def count(self):
|
||||
return 42
|
||||
|
||||
class LenContainer(object):
|
||||
def __len__(self):
|
||||
return 42
|
||||
|
||||
class PaginationTests(TestCase):
|
||||
def setUp(self):
|
||||
# Prepare a list of objects for pagination.
|
||||
for x in range(1, 10):
|
||||
a = Article(headline='Article %s' % x, pub_date=datetime(2005, 7, 29))
|
||||
a.save()
|
||||
|
||||
def test_paginator(self):
|
||||
paginator = Paginator(Article.objects.all(), 5)
|
||||
self.assertEqual(9, paginator.count)
|
||||
self.assertEqual(2, paginator.num_pages)
|
||||
self.assertEqual([1, 2], paginator.page_range)
|
||||
|
||||
def test_first_page(self):
|
||||
paginator = Paginator(Article.objects.all(), 5)
|
||||
p = paginator.page(1)
|
||||
self.assertEqual(u"<Page 1 of 2>", unicode(p))
|
||||
self.assertQuerysetEqual(p.object_list, [
|
||||
"<Article: Article 1>",
|
||||
"<Article: Article 2>",
|
||||
"<Article: Article 3>",
|
||||
"<Article: Article 4>",
|
||||
"<Article: Article 5>"
|
||||
]
|
||||
)
|
||||
self.assertTrue(p.has_next())
|
||||
self.assertFalse(p.has_previous())
|
||||
self.assertTrue(p.has_other_pages())
|
||||
self.assertEqual(2, p.next_page_number())
|
||||
self.assertEqual(0, p.previous_page_number())
|
||||
self.assertEqual(1, p.start_index())
|
||||
self.assertEqual(5, p.end_index())
|
||||
|
||||
def test_last_page(self):
|
||||
paginator = Paginator(Article.objects.all(), 5)
|
||||
p = paginator.page(2)
|
||||
self.assertEqual(u"<Page 2 of 2>", unicode(p))
|
||||
self.assertQuerysetEqual(p.object_list, [
|
||||
"<Article: Article 6>",
|
||||
"<Article: Article 7>",
|
||||
"<Article: Article 8>",
|
||||
"<Article: Article 9>"
|
||||
]
|
||||
)
|
||||
self.assertFalse(p.has_next())
|
||||
self.assertTrue(p.has_previous())
|
||||
self.assertTrue(p.has_other_pages())
|
||||
self.assertEqual(3, p.next_page_number())
|
||||
self.assertEqual(1, p.previous_page_number())
|
||||
self.assertEqual(6, p.start_index())
|
||||
self.assertEqual(9, p.end_index())
|
||||
|
||||
def test_empty_page(self):
|
||||
paginator = Paginator(Article.objects.all(), 5)
|
||||
self.assertRaises(EmptyPage, paginator.page, 0)
|
||||
self.assertRaises(EmptyPage, paginator.page, 3)
|
||||
|
||||
# Empty paginators with allow_empty_first_page=True.
|
||||
paginator = Paginator(Article.objects.filter(id=0), 5, allow_empty_first_page=True)
|
||||
self.assertEqual(0, paginator.count)
|
||||
self.assertEqual(1, paginator.num_pages)
|
||||
self.assertEqual([1], paginator.page_range)
|
||||
|
||||
# Empty paginators with allow_empty_first_page=False.
|
||||
paginator = Paginator(Article.objects.filter(id=0), 5, allow_empty_first_page=False)
|
||||
self.assertEqual(0, paginator.count)
|
||||
self.assertEqual(0, paginator.num_pages)
|
||||
self.assertEqual([], paginator.page_range)
|
||||
|
||||
def test_invalid_page(self):
|
||||
paginator = Paginator(Article.objects.all(), 5)
|
||||
self.assertRaises(InvalidPage, paginator.page, 7)
|
||||
|
||||
def test_orphans(self):
|
||||
# Add a few more records to test out the orphans feature.
|
||||
for x in range(10, 13):
|
||||
Article(headline="Article %s" % x, pub_date=datetime(2006, 10, 6)).save()
|
||||
|
||||
# With orphans set to 3 and 10 items per page, we should get all 12 items on a single page.
|
||||
paginator = Paginator(Article.objects.all(), 10, orphans=3)
|
||||
self.assertEqual(1, paginator.num_pages)
|
||||
|
||||
# With orphans only set to 1, we should get two pages.
|
||||
paginator = Paginator(Article.objects.all(), 10, orphans=1)
|
||||
self.assertEqual(2, paginator.num_pages)
|
||||
|
||||
def test_paginate_list(self):
|
||||
# Paginators work with regular lists/tuples, too -- not just with QuerySets.
|
||||
paginator = Paginator([1, 2, 3, 4, 5, 6, 7, 8, 9], 5)
|
||||
self.assertEqual(9, paginator.count)
|
||||
self.assertEqual(2, paginator.num_pages)
|
||||
self.assertEqual([1, 2], paginator.page_range)
|
||||
p = paginator.page(1)
|
||||
self.assertEqual(u"<Page 1 of 2>", unicode(p))
|
||||
self.assertEqual([1, 2, 3, 4, 5], p.object_list)
|
||||
self.assertTrue(p.has_next())
|
||||
self.assertFalse(p.has_previous())
|
||||
self.assertTrue(p.has_other_pages())
|
||||
self.assertEqual(2, p.next_page_number())
|
||||
self.assertEqual(0, p.previous_page_number())
|
||||
self.assertEqual(1, p.start_index())
|
||||
self.assertEqual(5, p.end_index())
|
||||
|
||||
def test_paginate_misc_classes(self):
|
||||
# Paginator can be passed other objects with a count() method.
|
||||
paginator = Paginator(CountContainer(), 10)
|
||||
self.assertEqual(42, paginator.count)
|
||||
self.assertEqual(5, paginator.num_pages)
|
||||
self.assertEqual([1, 2, 3, 4, 5], paginator.page_range)
|
||||
|
||||
# Paginator can be passed other objects that implement __len__.
|
||||
paginator = Paginator(LenContainer(), 10)
|
||||
self.assertEqual(42, paginator.count)
|
||||
self.assertEqual(5, paginator.num_pages)
|
||||
self.assertEqual([1, 2, 3, 4, 5], paginator.page_range)
|
Loading…
Reference in New Issue