Merge pull request #34 from atugushev/master
[django-pagination.git] / linaro_django_pagination / tests / test_main.py
diff --git a/linaro_django_pagination/tests/test_main.py b/linaro_django_pagination/tests/test_main.py
new file mode 100644 (file)
index 0000000..2409b0f
--- /dev/null
@@ -0,0 +1,399 @@
+# Copyright (c) 2008, Eric Florenzano
+# Copyright (c) 2010, 2011 Linaro Limited
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+#     * Redistributions of source code must retain the above copyright
+#       notice, this list of conditions and the following disclaimer.
+#     * Redistributions in binary form must reproduce the above
+#       copyright notice, this list of conditions and the following
+#       disclaimer in the documentation and/or other materials provided
+#       with the distribution.
+#     * Neither the name of the author nor the names of other
+#       contributors may be used to endorse or promote products derived
+#       from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (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 NoEllipsisTestCase(SimpleTestCase):
+    """
+    Tests a case where should be no any ellipsis pages.
+    """
+    def setUp(self):
+        self.p = Paginator(range(100), 25)
+
+    def test_on_start(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(1)}, 2, 0)['pages'],
+            [1, 2, 3, 4],
+        )
+
+    def test_in_middle_page_2(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(2)}, 2, 0)['pages'],
+            [1, 2, 3, 4],
+        )
+
+    def test_in_middle_page_3(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(3)}, 2, 0)['pages'],
+            [1, 2, 3, 4],
+        )
+
+    def test_on_end(self):
+        self.assertListEqual(
+            paginate({'paginator': self.p, 'page_obj': self.p.page(4)}, 2, 0)['pages'],
+            [1, 2, 3, 4],
+        )
+
+
+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)
+
+    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)