Allow for a variable as the paginate_by parameter. Also added tests for this case...

[django-pagination.git] / pagination / tests.py

diff --git a/pagination/tests.py b/pagination/tests.py
index a1d23f0..1579166 100644 (file)
--- a/pagination/tests.py
+++ b/pagination/tests.py
@@ -19,6 +19,15 @@
 >>> paginate({'paginator': p, 'page_obj': p.page(1)})['pages']
 [1, 2, 3, 4, None, 8, 9, 10, 11]
 
+# Testing orphans
+>>> p = Paginator(range(5), 2, 1)
+>>> paginate({'paginator': p, 'page_obj': p.page(1)})['pages']
+[1, 2]
+
+>>> p = Paginator(range(21), 2, 1)
+>>> paginate({'paginator': p, 'page_obj': p.page(1)})['pages']
+[1, 2, 3, 4, None, 7, 8, 9, 10]
+
 >>> t = Template("{% load pagination_tags %}{% autopaginate var 2 %}{% paginate %}")
 
 # WARNING: Please, please nobody read this portion of the code!
@@ -26,6 +35,7 @@
 ...     def __iter__(self): yield self.__dict__.__iter__
 ...     def copy(self): return self
 ...     def urlencode(self): return u''
+...     def keys(self): return []
 >>> class RequestProxy(object):
 ...     page = 1
 ...     GET = GetProxy()
@@ -33,10 +43,16 @@
 # ENDWARNING
 
 >>> t.render(Context({'var': range(21), 'request': RequestProxy()}))
-u'\\n<div class="pagination">...
+u'\\n\\n<div class="pagination">...
 >>>
 >>> t = Template("{% load pagination_tags %}{% autopaginate var %}{% paginate %}")
 >>> t.render(Context({'var': range(21), 'request': RequestProxy()}))
-u'\\n<div class="pagination">...
+u'\\n\\n<div class="pagination">...
+>>> t = Template("{% load pagination_tags %}{% autopaginate var 20 %}{% paginate %}")
+>>> t.render(Context({'var': range(21), 'request': RequestProxy()}))
+u'\\n\\n<div class="pagination">...
+>>> t = Template("{% load pagination_tags %}{% autopaginate var by %}{% paginate %}")
+>>> t.render(Context({'var': range(21), 'by': 20, 'request': RequestProxy()}))
+u'\\n\\n<div class="pagination">...
 >>>
 """
\ No newline at end of file