1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# -*- coding: utf-8 -*-
# Copyright (C) 1998-2012 by the Free Software Foundation, Inc.
#
# This file is part of HyperKitty.
#
# HyperKitty is free software: you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# HyperKitty is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# HyperKitty. If not, see <http://www.gnu.org/licenses/>.
#
# Author: Aamir Khan <syst3m.w0rm@gmail.com>
#
import datetime
from hyperkitty.tests.utils import TestCase
from hyperkitty.lib.view_helpers import get_display_dates, paginate
class GetDisplayDatesTestCase(TestCase):
def test_month(self):
begin_date, end_date = get_display_dates('2012', '6', None)
self.assertEqual(begin_date, datetime.datetime(2012, 6, 1))
self.assertEqual(end_date, datetime.datetime(2012, 7, 1))
def test_month_december(self):
try:
begin_date, end_date = get_display_dates('2012', '12', None)
except ValueError, e:
self.fail(e)
self.assertEqual(begin_date, datetime.datetime(2012, 12, 1))
self.assertEqual(end_date, datetime.datetime(2013, 1, 1))
def test_day(self):
begin_date, end_date = get_display_dates('2012', '4', '2')
self.assertEqual(begin_date, datetime.datetime(2012, 4, 2))
self.assertEqual(end_date, datetime.datetime(2012, 4, 3))
class PaginateTestCase(TestCase):
def test_page_range(self):
objects = range(1000)
self.assertEqual(paginate(objects, 1).page_range,
[1, 2, 3, 4, '...', 100])
self.assertEqual(paginate(objects, 2).page_range,
[1, 2, 3, 4, 5, '...', 100])
self.assertEqual(paginate(objects, 3).page_range,
[1, 2, 3, 4, 5, 6, '...', 100])
self.assertEqual(paginate(objects, 4).page_range,
[1, 2, 3, 4, 5, 6, 7, '...', 100])
self.assertEqual(paginate(objects, 5).page_range,
[1, 2, 3, 4, 5, 6, 7, 8, '...', 100])
self.assertEqual(paginate(objects, 6).page_range,
[1, 2, 3, 4, 5, 6, 7, 8, 9, '...', 100])
self.assertEqual(paginate(objects, 7).page_range,
[1, '...', 4, 5, 6, 7, 8, 9, 10, '...', 100])
self.assertEqual(paginate(objects, 8).page_range,
[1, '...', 5, 6, 7, 8, 9, 10, 11, '...', 100])
self.assertEqual(paginate(objects, 9).page_range,
[1, '...', 6, 7, 8, 9, 10, 11, 12, '...', 100])
self.assertEqual(paginate(objects, 10).page_range,
[1, '...', 7, 8, 9, 10, 11, 12, 13, '...', 100])
self.assertEqual(paginate(objects, 40).page_range,
[1, '...', 37, 38, 39, 40, 41, 42, 43, '...', 100])
self.assertEqual(paginate(objects, 90).page_range,
[1, '...', 87, 88, 89, 90, 91, 92, 93, '...', 100])
self.assertEqual(paginate(objects, 91).page_range,
[1, '...', 88, 89, 90, 91, 92, 93, 94, '...', 100])
self.assertEqual(paginate(objects, 92).page_range,
[1, '...', 89, 90, 91, 92, 93, 94, 95, '...', 100])
self.assertEqual(paginate(objects, 93).page_range,
[1, '...', 90, 91, 92, 93, 94, 95, 96, '...', 100])
self.assertEqual(paginate(objects, 94).page_range,
[1, '...', 91, 92, 93, 94, 95, 96, 97, '...', 100])
self.assertEqual(paginate(objects, 95).page_range,
[1, '...', 92, 93, 94, 95, 96, 97, 98, 99, 100])
self.assertEqual(paginate(objects, 96).page_range,
[1, '...', 93, 94, 95, 96, 97, 98, 99, 100])
self.assertEqual(paginate(objects, 97).page_range,
[1, '...', 94, 95, 96, 97, 98, 99, 100])
self.assertEqual(paginate(objects, 98).page_range,
[1, '...', 95, 96, 97, 98, 99, 100])
self.assertEqual(paginate(objects, 99).page_range,
[1, '...', 96, 97, 98, 99, 100])
self.assertEqual(paginate(objects, 100).page_range,
[1, '...', 97, 98, 99, 100])
|
