Fix bug with multiple paginations
[django-pagination.git] / linaro_django_pagination / tests / test_main.py
index ba1baa1..1dc6d2c 100644 (file)
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
+from django.core.paginator import Paginator
+from django.http import HttpRequest as DjangoHttpRequest
+from django.template import Template, Context
+
+try:
+    from django.test import SimpleTestCase
+except ImportError:  # Django 1.2 compatible
+    from django.test import TestCase as SimpleTestCase
+
+from linaro_django_pagination.paginator import InfinitePaginator, FinitePaginator
+from linaro_django_pagination.templatetags.pagination_tags import paginate
+from linaro_django_pagination.middleware import PaginationMiddleware
+
+
+class HttpRequest(DjangoHttpRequest):
+     page = lambda self, suffix: 1
+
+
+class CommonTestCase(SimpleTestCase):
+    def test_records_for_the_first_page(self):
+        p = Paginator(range(15), 2)
+        pg = paginate({'paginator': p, 'page_obj': p.page(1)})
+        self.assertListEqual(pg['pages'], [1, 2, 3, 4, 5, 6, 7, 8])
+        self.assertEqual(pg['records']['first'], 1)
+        self.assertEqual(pg['records']['last'], 2)
+
+    def test_records_for_the_last_page(self):
+        p = Paginator(range(15), 2)
+        pg = paginate({'paginator': p, 'page_obj': p.page(8)})
+        self.assertListEqual(pg['pages'], [1, 2, 3, 4, 5, 6, 7, 8])
+        self.assertEqual(pg['records']['first'], 15)
+        self.assertEqual(pg['records']['last'], 15)
+
+    def test_pages_list(self):
+        p = Paginator(range(17), 2)
+        self.assertEqual(paginate({'paginator': p, 'page_obj': p.page(1)})['pages'], [1, 2, 3, 4, 5, 6, 7, 8, 9])
+
+    def test_page_with_empty_objects_list(self):
+        p = Paginator(range(0), 2)
+        self.assertListEqual(paginate({'paginator': p, 'page_obj': p.page(1)})['pages'], [1])
+
+
+class DefaultWindowTestCase(SimpleTestCase):
+    """
+    Test paginate using default window setting
+    moving the window from 1 ... to end
+    window size = 2, margin = 2
+    window = 2 -> show 5 pages
+    """
+    def setUp(self):
+        self.p = Paginator(range(31), 2)
+
+    def test_on_start_page_1(self):
+        # [1] 2 3 4 5 ... 15, 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(1)}, 2, 2)['pages'],
+            [1, 2, 3, 4, 5, None, 15, 16]
+        )
+
+    def test_on_start_page_2(self):
+        # 1 [2] 3 4 5 ... 15, 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(2)}, 2, 2)['pages'],
+            [1, 2, 3, 4, 5, None, 15, 16]
+        )
+
+    def test_on_start_page_3(self):
+        # 1 2 [3] 4 5 ... 15, 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(3)}, 2, 2)['pages'],
+            [1, 2, 3, 4, 5, None, 15, 16]
+        )
+
+    def test_on_start_page_4(self):
+        # 1 2 3 [4] 5 6 ... 15, 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(4)}, 2, 2)['pages'],
+            [1, 2, 3, 4, 5, 6, None, 15, 16])
+
+    def test_on_start_page_5(self):
+        # 1 2 3 4 [5] 6 7 ... 15, 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(5)}, 2, 2)['pages'],
+            [1, 2, 3, 4, 5, 6, 7, None, 15, 16]
+        )
+
+    def test_in_middle(self):
+        # 1 2 ... 5 6 [7] 8 9 ... 15, 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(7)}, 2, 2)['pages'],
+            [1, 2, None, 5, 6, 7, 8, 9, None, 15, 16]
+        )
+
+    def test_on_end_page_13(self):
+        # 1 2 ... 12 [13] 14 15 16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(13)}, 2, 2)['pages'],
+            [1, 2, None, 11, 12, 13, 14, 15, 16],
+        )
+
+    def test_on_end(self):
+        # 1 2 ... 12 13 14 15 [16
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(16)}, 2, 2)['pages'],
+            [1, 2, None, 12, 13, 14, 15, 16]
+        )
+
+
+class NoMarginTestCase(SimpleTestCase):
+    def setUp(self):
+        self.p = Paginator(range(31), 2)
+
+    def test_on_start(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(3)}, 2, 0)['pages'],
+            [1, 2, 3, 4, 5, None],
+        )
+
+    def test_in_middle(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(5)}, 2, 0)['pages'],
+            [None, 3, 4, 5, 6, 7, None],
+        )
+
+    def test_on_end(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(16)}, 2, 0)['pages'],
+            [None, 12, 13, 14, 15, 16],
+        )
+
+
+class ZeroWindowZeroMarginTestCase(SimpleTestCase):
+    """
+    Test paginate using window=0 and margin=0
+    """
+    def setUp(self):
+        self.p = Paginator(range(31), 2)
+
+    def test_on_start_page_1(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(1)}, 0, 0)['pages'],
+            [1, None],
+        )
+
+    def test_in_middle_page_2(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(2)}, 0, 0)['pages'],
+            [None, 2, None],
+        )
+
+    def test_in_middle_page_3(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(3)}, 0, 0)['pages'],
+            [None, 3, None],
+        )
+
+    def test_in_middle_page_10(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(10)}, 0, 0)['pages'],
+            [None, 10, None],
+        )
+
+    def test_in_middle_page_14(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(14)}, 0, 0)['pages'],
+            [None, 14, None],
+        )
+
+    def test_in_middle_page_15(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(15)}, 0, 0)['pages'],
+            [None, 15, None],
+        )
+
+    def test_on_end_page_16(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(16)}, 0, 0)['pages'],
+            [None, 16],
+        )
+
+
+class SpecialTestCase(SimpleTestCase):
+    def test_middle_with_no_window_and_margin_1(self):
+        p = Paginator(range(31), 2)
+        self.assertListEqual(
+            paginate({'paginator': p, 'page_obj': p.page(5)}, 0, 1)['pages'],
+            [1, None, 5, None, 16],
+        )
+
+    def test_middle_with_no_window_and_margin_4(self):
+        p = Paginator(range(21), 2, 1)
+        self.assertListEqual(
+            paginate({'paginator': p, 'page_obj': p.page(1)}, 0, 4)['pages'],
+            [1, 2, 3, 4, None, 7, 8, 9, 10],
+        )
+
+
+class TemplateRenderingTestCase(SimpleTestCase):
+    def test_default_tag_options(self):
+        t = Template("{% load pagination_tags %}{% autopaginate var %}{% paginate %}")
+        self.assertIn(
+            '<div class="pagination">',
+            t.render(Context({'var': range(21), 'request': HttpRequest()})),
+        )
+
+    def test_paginate_by_option(self):
+        t = Template("{% load pagination_tags %}{% autopaginate var 20 %}{% paginate %}")
+        self.assertIn(
+            '<div class="pagination">',
+            t.render(Context({'var': range(21), 'request': HttpRequest()})),
+        )
+
+    def test_orphans_option(self):
+        t = Template("{% load pagination_tags %}{% autopaginate var by %}{% paginate %}")
+        content = t.render(Context({'var': range(21), 'by': 20, 'request': HttpRequest()}))
+        self.assertIn('<div class="pagination">', content)
+        self.assertIn('<a href="?page=2"', content)
+
+    def test_as_option(self):
+        t = Template("{% load pagination_tags %}{% autopaginate var by as foo %}{{ foo }}")
+        self.assertEqual(
+            t.render(Context({'var': range(21), 'by': 20, 'request': HttpRequest()})),
+            str(range(20)),
+        )
+
+    def test_multiple_pagination(self):
+        t = Template("{% load pagination_tags %}{% autopaginate var2 by as foo2 %}{% paginate %}"
+                     "{% autopaginate var by as foo %}{% paginate %}")
+        content = t.render(Context({'var': range(21), 'var2': range(50, 121), 'by': 20, 'request': HttpRequest()}))
+        self.assertIn('<div class="pagination">', content)
+        self.assertIn('<a href="?page_var2=2"', content)
+        self.assertIn('<a href="?page_var=2"', content)
+
+
+class InfinitePaginatorTestCase(SimpleTestCase):
+    def setUp(self):
+        self.p = InfinitePaginator(range(20), 2, link_template='/bacon/page/%d')
+
+    def test_paginator_repr(self):
+        self.assertEqual(
+            repr(InfinitePaginator),
+            "<class 'linaro_django_pagination.paginator.InfinitePaginator'>",
+        )
+
+    def test_validate_number(self):
+        self.assertEqual(self.p.validate_number(2), 2)
+
+    def test_orphans(self):
+        self.assertEqual(self.p.orphans, 0)
+
+    def test_page_repr(self):
+        self.assertEqual(repr(self.p.page(3)), '<Page 3>')
+
+    def test_page_end_index(self):
+        self.assertEqual(self.p.page(3).end_index(), 6)
 
-"""
->>> from django.core.paginator import Paginator
->>> from linaro_django_pagination.templatetags.pagination_tags import paginate
->>> from django.template import Template, Context
-
->>> p = Paginator(range(15), 2)
->>> pg = paginate({'paginator': p, 'page_obj': p.page(1)})
->>> pg['pages']
-[1, 2, 3, 4, 5, 6, 7, 8]
->>> pg['records']['first']
-1
->>> pg['records']['last']
-2
-
->>> p = Paginator(range(15), 2)
->>> pg = paginate({'paginator': p, 'page_obj': p.page(8)})
->>> pg['pages']
-[1, 2, 3, 4, 5, 6, 7, 8]
->>> pg['records']['first']
-15
->>> pg['records']['last']
-15
-
->>> p = Paginator(range(17), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(1)})['pages']
-[1, 2, 3, 4, 5, 6, 7, 8, 9]
-
-
-# on start
-# moving the window from 1 ... to end
-# window size = 2, margin = 2
-# [1] 2 3 4 5 ... 15, 16
-# 1 [2] 3 4 5 ... 15, 16
-# 1 2 [3] 4 5 ... 15, 16
-# 1 2 3 [4] 5 6 ... 15, 16
-# 1 2 3 4 [5] 6 7 ... 15, 16
-# 1 2 3 4 5 [6] 7 8 ... 15, 16
-# 1 2 ... 5 6 [7] 8 9 ... 15, 16
-
-# window = 2 -> show 5 pages
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(1)}, 2, 2)['pages']
-[1, 2, 3, 4, 5, None, 15, 16]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(2)}, 2, 2)['pages']
-[1, 2, 3, 4, 5, None, 15, 16]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(3)}, 2, 2)['pages']
-[1, 2, 3, 4, 5, None, 15, 16]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(4)}, 2, 2)['pages']
-[1, 2, 3, 4, 5, 6, None, 15, 16]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(5)}, 2, 2)['pages']
-[1, 2, 3, 4, 5, 6, 7, None, 15, 16]
-
-# in the middle
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(7)}, 2, 2)['pages']
-[1, 2, None, 5, 6, 7, 8, 9, None, 15, 16]
-
-# on end
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(16)}, 2, 2)['pages']
-[1, 2, None, 12, 13, 14, 15, 16]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(13)}, 2, 2)['pages']
-[1, 2, None, 11, 12, 13, 14, 15, 16]
-
-
->>> p = Paginator(range(0), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(1)})['pages']
-[1]
-
-
-
-# no margin
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(3)}, 2, 0)['pages']
-[1, 2, 3, 4, 5, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(5)}, 2, 0)['pages']
-[None, 3, 4, 5, 6, 7, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(16)}, 2, 0)['pages']
-[None, 12, 13, 14, 15, 16]
-
-
-# special
-# zero window, zero margin
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(1)}, 0, 0)['pages']
-[1, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(2)}, 0, 0)['pages']
-[None, 2, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(3)}, 0, 0)['pages']
-[None, 3, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(10)}, 0, 0)['pages']
-[None, 10, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(14)}, 0, 0)['pages']
-[None, 14, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(15)}, 0, 0)['pages']
-[None, 15, None]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(16)}, 0, 0)['pages']
-[None, 16]
-
->>> p = Paginator(range(31), 2)
->>> paginate({'paginator': p, 'page_obj': p.page(5)}, 0, 1)['pages']
-[1, None, 5, None, 16]
-
->>> p = Paginator(range(21), 2, 1)
->>> paginate({'paginator': p, 'page_obj': p.page(1)}, 0, 4)['pages']
-[1, 2, 3, 4, None, 7, 8, 9, 10]
-
-# Testing template rendering
-
->>> t = Template("{% load pagination_tags %}{% autopaginate var 2 %}{% paginate %}")
-
->>> from django.http import HttpRequest as DjangoHttpRequest
->>> class HttpRequest(DjangoHttpRequest):
-...     page = lambda self, suffix: 1
-
->>> t.render(Context({'var': range(21), 'request': HttpRequest()})) #doctest: +ELLIPSIS
-u'\\n...\\n...<div class="pagination">...
->>>
->>> t = Template("{% load pagination_tags %}{% autopaginate var %}{% paginate %}")
->>> t.render(Context({'var': range(21), 'request': HttpRequest()})) #doctest: +ELLIPSIS
-u'\\n...\\n...<div class="pagination">...
->>> t = Template("{% load pagination_tags %}{% autopaginate var 20 %}{% paginate %}")
->>> t.render(Context({'var': range(21), 'request': HttpRequest()})) #doctest: +ELLIPSIS
-u'\\n...\\n...<div class="pagination">...
->>> t = Template("{% load pagination_tags %}{% autopaginate var by %}{% paginate %}")
->>> t.render(Context({'var': range(21), 'by': 20, 'request': HttpRequest()})) #doctest: +ELLIPSIS
-u'\\n...\\n...<div class="pagination">...<a href="?page=2"...
->>> t = Template("{% load pagination_tags %}{% autopaginate var by as foo %}{{ foo }}")
->>> t.render(Context({'var': range(21), 'by': 20, 'request': HttpRequest()}))
-u'[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]'
->>>
->>> t = Template("{% load pagination_tags %}{% autopaginate var2 by as foo2 %}{% paginate %}{% autopaginate var by as foo %}{% paginate %}")
->>> t.render(Context({'var': range(21), 'var2': range(50, 121), 'by': 20, 'request': HttpRequest()})) #doctest: +ELLIPSIS
-u'\\n...\\n...<div class="pagination">...<a href="?page_var2=2"...<a href="?page_var=2"...
->>>
-
-# Testing InfinitePaginator
-
->>> from linaro_django_pagination.paginator import InfinitePaginator
-
->>> InfinitePaginator
-<class 'linaro_django_pagination.paginator.InfinitePaginator'>
->>> p = InfinitePaginator(range(20), 2, link_template='/bacon/page/%d')
->>> p.validate_number(2)
-2
->>> p.orphans
-0
->>> p3 = p.page(3)
->>> p3
-<Page 3>
->>> p3.end_index()
-6
->>> p3.has_next()
-True
->>> p3.has_previous()
-True
->>> p.page(10).has_next()
-False
->>> p.page(1).has_previous()
-False
->>> p3.next_link()
-'/bacon/page/4'
->>> p3.previous_link()
-'/bacon/page/2'
-
-# Testing FinitePaginator
-
->>> from linaro_django_pagination.paginator import FinitePaginator
-
->>> FinitePaginator
-<class 'linaro_django_pagination.paginator.FinitePaginator'>
->>> p = FinitePaginator(range(20), 2, offset=10, link_template='/bacon/page/%d')
->>> p.validate_number(2)
-2
->>> p.orphans
-0
->>> p3 = p.page(3)
->>> p3
-<Page 3>
->>> p3.start_index()
-10
->>> p3.end_index()
-6
->>> p3.has_next()
-True
->>> p3.has_previous()
-True
->>> p3.next_link()
-'/bacon/page/4'
->>> p3.previous_link()
-'/bacon/page/2'
-
->>> p = FinitePaginator(range(20), 20, offset=10, link_template='/bacon/page/%d')
->>> p2 = p.page(2)
->>> p2
-<Page 2>
->>> p2.has_next()
-False
->>> p3.has_previous()
-True
->>> p2.next_link()
-
->>> p2.previous_link()
-'/bacon/page/1'
-
->>> from linaro_django_pagination.middleware import PaginationMiddleware
->>> from django.core.handlers.wsgi import WSGIRequest
->>> from StringIO import StringIO
->>> middleware = PaginationMiddleware()
->>> request = WSGIRequest({'REQUEST_METHOD': 'POST', 'CONTENT_TYPE': 'multipart', 'wsgi.input': StringIO()})
->>> middleware.process_request(request)
->>> request.upload_handlers.append('asdf')
-"""
+    def test_page_has_next(self):
+        self.assertTrue(self.p.page(3).has_next(), True)
+
+    def test_page_has_previous(self):
+        self.assertTrue(self.p.page(3).has_previous(), True)
+
+    def test_page_next_link(self):
+        self.assertEqual(self.p.page(3).next_link(), '/bacon/page/4')
+
+    def test_page_previous_link(self):
+        self.assertEqual(self.p.page(3).previous_link(), '/bacon/page/2')
+
+    def test_last_page_which_has_no_next_page(self):
+        self.assertFalse(self.p.page(10).has_next())
+
+    def test_first_page_which_has_no_previous_page(self):
+        self.assertFalse(self.p.page(1).has_previous())
+
+
+class FinitePaginatorTestCase(SimpleTestCase):
+    def setUp(self):
+        self.p = FinitePaginator(range(20), 2, offset=10, link_template='/bacon/page/%d')
+
+    def test_repr(self):
+        self.assertEqual(
+            repr(FinitePaginator),
+            "<class 'linaro_django_pagination.paginator.FinitePaginator'>"
+        )
+
+    def test_validate_number(self):
+        self.assertEqual(self.p.validate_number(2), 2)
+
+    def test_orphans(self):
+        self.assertEqual(self.p.orphans, 0)
+
+    def test_page_repr(self):
+        self.assertEqual(repr(self.p.page(3)), '<Page 3>')
+
+    def test_page_start_index(self):
+        self.assertEqual(self.p.page(3).start_index(), 10)
+
+    def test_page_end(self):
+        self.assertEqual(self.p.page(3).end_index(), 6)
+
+    def test_page_has_next(self):
+        self.assertTrue(self.p.page(3).has_next(), True)
+
+    def test_page_has_previous(self):
+        self.assertTrue(self.p.page(3).has_previous(), True)
+
+    def test_page_next_link(self):
+        self.assertEqual(self.p.page(3).next_link(), '/bacon/page/4')
+
+    def test_page_previous_link(self):
+        self.assertEqual(self.p.page(3).previous_link(), '/bacon/page/2')
+
+    def test_on_start_page_repr(self):
+        self.assertEqual(repr(self.p.page(2)), '<Page 2>')
+
+    def test_on_start_has_no_next(self):
+        self.assertTrue(self.p.page(2).has_next(), False)
+
+    def test_on_start_has_previous(self):
+        self.assertTrue(self.p.page(2).has_previous(), True)
+
+    def test_on_start_has_next_link(self):
+        self.assertEqual(self.p.page(2).next_link(), '/bacon/page/3')
+
+    def test_on_start_has_previous_link(self):
+        self.assertEqual(self.p.page(2).previous_link(), '/bacon/page/1')
+
+
+class MiddlewareTestCase(SimpleTestCase):
+    """
+    Test middleware
+    """
+    def test_get_page_in_request(self):
+        middleware = PaginationMiddleware()
+        request = DjangoHttpRequest()
+        middleware.process_request(request)
+        self.assertEqual(request.page(''), 1)