Misago/misago/threads/tests/test_paginator.py

from django.test import TestCase

from misago.threads.paginator import PostsPaginator


class PostsPaginatorTests(TestCase):
    def test_paginator(self):
        """pages share first and last items with each other"""
        items = [i + 1 for i in range(30)]

        paginator = PostsPaginator(items, 5)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4, 5],
            [5, 6, 7, 8, 9],
            [9, 10, 11, 12, 13],
            [13, 14, 15, 16, 17],
            [17, 18, 19, 20, 21],
            [21, 22, 23, 24, 25],
            [25, 26, 27, 28, 29],
            [29, 30],
        ])

    def test_paginator_orphans(self):
        """paginator handles orphans"""
        items = [i + 1 for i in range(16)]

        paginator = PostsPaginator(items, 8, 6)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4, 5, 6, 7, 8],
            [8, 9, 10, 11, 12, 13, 14, 15, 16],
        ])

        paginator = PostsPaginator(items, 4, 4)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4],
            [4, 5, 6, 7],
            [7, 8, 9, 10],
            [10, 11, 12, 13, 14, 15, 16],
        ])

        paginator = PostsPaginator(items, 5, 3)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4, 5],
            [5, 6, 7, 8, 9],
            [9, 10, 11, 12, 13, 14, 15, 16],
        ])

        paginator = PostsPaginator(items, 6, 2)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4, 5, 6],
            [6, 7, 8, 9, 10, 11],
            [11, 12, 13, 14, 15, 16],
        ])

        paginator = PostsPaginator(items, 7, 1)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4, 5, 6, 7],
            [7, 8, 9, 10, 11, 12, 13],
            [13, 14, 15, 16],
        ])

        paginator = PostsPaginator(items, 7, 3)
        self.assertEqual(self.get_paginator_items_list(paginator), [
            [1, 2, 3, 4, 5, 6, 7],
            [7, 8, 9, 10, 11, 12, 13, 14, 15, 16],
        ])

        paginator = PostsPaginator(items, 10, 6)
        self.assertEqual(self.get_paginator_items_list(paginator), [items])

        # regression test for #732
        items = [i + 1 for i in range(24)]

        paginator = PostsPaginator(items, 18, 6)
        self.assertEqual(paginator.num_pages, 1)
        self.assertEqual(self.get_paginator_items_list(paginator), [items])

        # extra tests for catching issues in excessively long datasets
        paginator = PostsPaginator([i + 1 for i in range(144)], 14, 6)
        last_page = self.get_paginator_items_list(paginator)[-1]
        self.assertEqual(last_page[-4:], [141, 142, 143, 144])

        paginator = PostsPaginator([i + 1 for i in range(321)], 14, 6)
        last_page = self.get_paginator_items_list(paginator)[-1]
        self.assertEqual(last_page[-4:], [318, 319, 320, 321])

    def get_paginator_items_list(self, paginator):
        items_list = []
        for page in paginator.page_range:
            items_list.append(paginator.page(page).object_list)
        return items_list