summaryrefslogtreecommitdiff
path: root/trunk/dimbola/grid_tests.py
blob: 8af2f292dd88ee4ab01e154cdb330a1c71fc60d8 (plain)
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
# Copyright (C) 2009  Lars Wirzenius <liw@liw.fi>
# 
# This program 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.
# 
# This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.


import unittest

import dimbola


class GridDimensionsTests(unittest.TestCase):

    def setUp(self):
        self.model = dimbola.GridModel()
        self.model.scale_value = 10
        self.model.widget_width = 100
        self.model.widget_height = 100
        self.cell_dim = self.model.scale_value + self.model.padding
        self.per_row = self.model.widget_width / self.cell_dim

    def test_sets_padding_by_default(self):
        self.assert_(self.model.padding > 0)
        
    def test_computes_vertical_pixels_correctly_for_no_photos(self):
        self.assertEqual(self.model.vertical_pixels, 0)
        
    def test_computes_vertical_pixels_correctly_for_one_photo(self):
        self.model.photoids = [1]
        self.assertEqual(self.model.vertical_pixels, self.cell_dim)
        
    def test_computes_vertical_pixels_correctly_for_one_row(self):
        self.model.photoids = range(self.per_row)
        self.assertEqual(self.model.vertical_pixels, self.cell_dim)
        
    def test_computes_vertical_pixels_correctly_for_two_rows(self):
        self.model.photoids = range(1 + self.per_row)
        self.assertEqual(self.model.vertical_pixels, self.cell_dim * 2)

    def test_computes_vertical_pixels_correctly_for_lots_of_rows(self):
        self.model.photoids = range(self.per_row * 100**2)
        self.assertEqual(self.model.vertical_pixels, self.cell_dim * 100**2)


class GridModelTests(unittest.TestCase):

    def setUp(self):
        self.model = dimbola.GridModel()
    
    def fake_emit(self, *args):
        self.emit_args = args
        
    def test_has_empty_list_of_photoids_initially(self):
        self.assertEqual(self.model.photoids, [])

    def test_sets_photoids_correctly(self):
        self.model.photoids = [1, 2, 3]
        self.assertEqual(self.model.photoids, [1, 2, 3])
        
    def test_photoids_changed_emits_signals(self):
        self.model.emit = self.fake_emit
        self.model.photoids_changed()
        self.assertEqual(self.emit_args, ('photoids-changed',))

    def test_setting_photoids_emits_signal(self):
        self.model.emit = self.fake_emit
        self.model.photoids = [1]
        self.assertEqual(self.emit_args, ('photoids-changed',))
        
    def test_has_empty_list_of_selected_initially(self):
        self.assertEqual(self.model.selected, [])

    def test_selecting_unknown_photoids_raises_exception(self):
        self.assertRaises(AssertionError, self.model.set_selected, [1, 2, 3])

    def test_sets_selected_correctly(self):
        self.model.photoids = [1, 2, 3]
        self.model.selected = [1, 2, 3]
        self.assertEqual(self.model.selected, [1, 2, 3])
        
    def test_selection_changed_emits_signals(self):
        self.model.emit = self.fake_emit
        self.model.selection_changed()
        self.assertEqual(self.emit_args, ('selection-changed',))

    def test_setting_selection_emits_signal(self):
        self.model.emit = self.fake_emit
        self.model.photoids = [1, 2, 3]
        self.model.selected = [1]
        self.assertEqual(self.emit_args, ('selection-changed',))

    def test_setting_photoids_removes_selection(self):
        self.model.photoids = [1]
        self.model.selected = [1]
        self.model.photoids = [2]
        self.assertEqual(self.model.selected, [])

    def test_setting_photoids_clears_angles(self):
        self.model.photoids = [1]
        self.model.angles[1] = 180
        self.model.photoids = [1]
        self.assert_(1 not in self.model.angles)

    def test_setting_photoids_clears_thumbnails(self):
        self.model.photoids = [1]
        self.model.thumbnails[1] = 'mock thumbnail'
        self.model.photoids = [1]
        self.assert_(1 not in self.model.thumbnails)
        
    def test_sets_thumbnail(self):
        self.model.photoids = [1]
        self.model.set_thumbnail(1, 'mock thumbnail')
        self.assertEqual(self.model.thumbnails[1], 'mock thumbnail')
        
    def test_setting_thumbnail_for_nonexistent_photo_raises_exception(self):
        self.assertRaises(AssertionError, self.model.set_thumbnail, 1, '')
        
    def test_setting_thumbnail_emits_signal(self):
        self.model.emit = self.fake_emit
        self.model.photoids = [1]
        self.model.set_thumbnail(1, 'mock thumbnail')
        self.assertEqual(self.emit_args, ('photoids-changed',))
        
    def test_sets_angle(self):
        self.model.photoids = [1]
        self.model.set_angle(1, 90)
        self.assertEqual(self.model.angles[1], 90)
        
    def test_setting_angle_for_nonexistent_photo_raises_exception(self):
        self.assertRaises(AssertionError, self.model.set_angle, 1, 90)
        
    def test_setting_angle_emits_signal(self):
        self.model.emit = self.fake_emit
        self.model.photoids = [1]
        self.model.set_angle(1, 90)
        self.assertEqual(self.emit_args, ('photoids-changed',))


class ThumbnailPosTests(unittest.TestCase):

    def setUp(self):
        self.model = dimbola.GridModel()
        self.model.photoids = range(10)
        self.model.widget_width = 100
        self.model.padding = 10
        self.model.scale_value = 20
        self.cell_dim = self.model.padding + self.model.scale_value
        self.per_row = self.model.widget_width / self.cell_dim

    def test_computes_xy_of_zeroth_photo_correctly(self):
        self.assertEqual(self.model.thumbnail_pos(0), (0, 0))

    def test_computes_xy_of_last_photo_on_first_row_correctly(self):
        self.assertEqual(self.model.thumbnail_pos(self.per_row - 1), 
                         ((self.per_row - 1) * self.cell_dim, 0))

    def test_computes_xy_of_first_photo_on_second_row_correctly(self):
        self.assertEqual(self.model.thumbnail_pos(self.per_row), 
                         (0, self.cell_dim))

    def test_computes_xy_of_last_photo_row_correctly(self):
        i = len(self.model.photoids) - 1
        row = i / self.per_row
        col = i % self.per_row
        self.assertEqual(self.model.thumbnail_pos(i), 
                         (col * self.cell_dim, row * self.cell_dim))


class GridModelSelectionTests(unittest.TestCase):

    def setUp(self):
        self.model = dimbola.GridModel()
        self.model.photoids = [1, 2, 3]

    def test_select_next_selects_first_photo_if_nothing_selected(self):
        self.model.select_next()
        self.assertEqual(self.model.selected, [1])

    def test_select_next_selects_next_photo_if_something_is_selected(self):
        self.model.selected = [1, 2]
        self.model.select_next()
        self.assertEqual(self.model.selected, [2])

    def test_select_previous_selects_first_photo_if_nothing_selected(self):
        self.model.select_previous()
        self.assertEqual(self.model.selected, [1])

    def test_select_previous_selects_previous_if_something_is_selected(self):
        self.model.selected = [2, 3]
        self.model.select_previous()
        self.assertEqual(self.model.selected, [1])