|
1 # -*- coding: utf-8 -*- |
|
2 |
|
3 import os |
|
4 from django import forms |
|
5 from django.db.models import Q |
|
6 from django.test import TestCase |
|
7 from tagging.forms import TagField |
|
8 from tagging import settings |
|
9 from tagging.models import Tag, TaggedItem |
|
10 from tagging.tests.models import Article, Link, Perch, Parrot, FormTest |
|
11 from tagging.utils import calculate_cloud, edit_string_for_tags, get_tag_list, get_tag, parse_tag_input |
|
12 from tagging.utils import LINEAR |
|
13 |
|
14 ############# |
|
15 # Utilities # |
|
16 ############# |
|
17 |
|
18 class TestParseTagInput(TestCase): |
|
19 def test_with_simple_space_delimited_tags(self): |
|
20 """ Test with simple space-delimited tags. """ |
|
21 |
|
22 self.assertEquals(parse_tag_input('one'), [u'one']) |
|
23 self.assertEquals(parse_tag_input('one two'), [u'one', u'two']) |
|
24 self.assertEquals(parse_tag_input('one two three'), [u'one', u'three', u'two']) |
|
25 self.assertEquals(parse_tag_input('one one two two'), [u'one', u'two']) |
|
26 |
|
27 def test_with_comma_delimited_multiple_words(self): |
|
28 """ Test with comma-delimited multiple words. |
|
29 An unquoted comma in the input will trigger this. """ |
|
30 |
|
31 self.assertEquals(parse_tag_input(',one'), [u'one']) |
|
32 self.assertEquals(parse_tag_input(',one two'), [u'one two']) |
|
33 self.assertEquals(parse_tag_input(',one two three'), [u'one two three']) |
|
34 self.assertEquals(parse_tag_input('a-one, a-two and a-three'), |
|
35 [u'a-one', u'a-two and a-three']) |
|
36 |
|
37 def test_with_double_quoted_multiple_words(self): |
|
38 """ Test with double-quoted multiple words. |
|
39 A completed quote will trigger this. Unclosed quotes are ignored. """ |
|
40 |
|
41 self.assertEquals(parse_tag_input('"one'), [u'one']) |
|
42 self.assertEquals(parse_tag_input('"one two'), [u'one', u'two']) |
|
43 self.assertEquals(parse_tag_input('"one two three'), [u'one', u'three', u'two']) |
|
44 self.assertEquals(parse_tag_input('"one two"'), [u'one two']) |
|
45 self.assertEquals(parse_tag_input('a-one "a-two and a-three"'), |
|
46 [u'a-one', u'a-two and a-three']) |
|
47 |
|
48 def test_with_no_loose_commas(self): |
|
49 """ Test with no loose commas -- split on spaces. """ |
|
50 self.assertEquals(parse_tag_input('one two "thr,ee"'), [u'one', u'thr,ee', u'two']) |
|
51 |
|
52 def test_with_loose_commas(self): |
|
53 """ Loose commas - split on commas """ |
|
54 self.assertEquals(parse_tag_input('"one", two three'), [u'one', u'two three']) |
|
55 |
|
56 def test_tags_with_double_quotes_can_contain_commas(self): |
|
57 """ Double quotes can contain commas """ |
|
58 self.assertEquals(parse_tag_input('a-one "a-two, and a-three"'), |
|
59 [u'a-one', u'a-two, and a-three']) |
|
60 self.assertEquals(parse_tag_input('"two", one, one, two, "one"'), |
|
61 [u'one', u'two']) |
|
62 |
|
63 def test_with_naughty_input(self): |
|
64 """ Test with naughty input. """ |
|
65 |
|
66 # Bad users! Naughty users! |
|
67 self.assertEquals(parse_tag_input(None), []) |
|
68 self.assertEquals(parse_tag_input(''), []) |
|
69 self.assertEquals(parse_tag_input('"'), []) |
|
70 self.assertEquals(parse_tag_input('""'), []) |
|
71 self.assertEquals(parse_tag_input('"' * 7), []) |
|
72 self.assertEquals(parse_tag_input(',,,,,,'), []) |
|
73 self.assertEquals(parse_tag_input('",",",",",",","'), [u',']) |
|
74 self.assertEquals(parse_tag_input('a-one "a-two" and "a-three'), |
|
75 [u'a-one', u'a-three', u'a-two', u'and']) |
|
76 |
|
77 class TestNormalisedTagListInput(TestCase): |
|
78 def setUp(self): |
|
79 self.cheese = Tag.objects.create(name='cheese') |
|
80 self.toast = Tag.objects.create(name='toast') |
|
81 |
|
82 def test_single_tag_object_as_input(self): |
|
83 self.assertEquals(get_tag_list(self.cheese), [self.cheese]) |
|
84 |
|
85 def test_space_delimeted_string_as_input(self): |
|
86 ret = get_tag_list('cheese toast') |
|
87 self.assertEquals(len(ret), 2) |
|
88 self.failUnless(self.cheese in ret) |
|
89 self.failUnless(self.toast in ret) |
|
90 |
|
91 def test_comma_delimeted_string_as_input(self): |
|
92 ret = get_tag_list('cheese,toast') |
|
93 self.assertEquals(len(ret), 2) |
|
94 self.failUnless(self.cheese in ret) |
|
95 self.failUnless(self.toast in ret) |
|
96 |
|
97 def test_with_empty_list(self): |
|
98 self.assertEquals(get_tag_list([]), []) |
|
99 |
|
100 def test_list_of_two_strings(self): |
|
101 ret = get_tag_list(['cheese', 'toast']) |
|
102 self.assertEquals(len(ret), 2) |
|
103 self.failUnless(self.cheese in ret) |
|
104 self.failUnless(self.toast in ret) |
|
105 |
|
106 def test_list_of_tag_primary_keys(self): |
|
107 ret = get_tag_list([self.cheese.id, self.toast.id]) |
|
108 self.assertEquals(len(ret), 2) |
|
109 self.failUnless(self.cheese in ret) |
|
110 self.failUnless(self.toast in ret) |
|
111 |
|
112 def test_list_of_strings_with_strange_nontag_string(self): |
|
113 ret = get_tag_list(['cheese', 'toast', 'ŠĐĆŽćžšđ']) |
|
114 self.assertEquals(len(ret), 2) |
|
115 self.failUnless(self.cheese in ret) |
|
116 self.failUnless(self.toast in ret) |
|
117 |
|
118 def test_list_of_tag_instances(self): |
|
119 ret = get_tag_list([self.cheese, self.toast]) |
|
120 self.assertEquals(len(ret), 2) |
|
121 self.failUnless(self.cheese in ret) |
|
122 self.failUnless(self.toast in ret) |
|
123 |
|
124 def test_tuple_of_instances(self): |
|
125 ret = get_tag_list((self.cheese, self.toast)) |
|
126 self.assertEquals(len(ret), 2) |
|
127 self.failUnless(self.cheese in ret) |
|
128 self.failUnless(self.toast in ret) |
|
129 |
|
130 def test_with_tag_filter(self): |
|
131 ret = get_tag_list(Tag.objects.filter(name__in=['cheese', 'toast'])) |
|
132 self.assertEquals(len(ret), 2) |
|
133 self.failUnless(self.cheese in ret) |
|
134 self.failUnless(self.toast in ret) |
|
135 |
|
136 def test_with_invalid_input_mix_of_string_and_instance(self): |
|
137 try: |
|
138 get_tag_list(['cheese', self.toast]) |
|
139 except ValueError, ve: |
|
140 self.assertEquals(str(ve), |
|
141 'If a list or tuple of tags is provided, they must all be tag names, Tag objects or Tag ids.') |
|
142 except Exception, e: |
|
143 raise self.failureException('the wrong type of exception was raised: type [%s] value [%]' %\ |
|
144 (str(type(e)), str(e))) |
|
145 else: |
|
146 raise self.failureException('a ValueError exception was supposed to be raised!') |
|
147 |
|
148 def test_with_invalid_input(self): |
|
149 try: |
|
150 get_tag_list(29) |
|
151 except ValueError, ve: |
|
152 self.assertEquals(str(ve), 'The tag input given was invalid.') |
|
153 except Exception, e: |
|
154 raise self.failureException('the wrong type of exception was raised: type [%s] value [%s]' %\ |
|
155 (str(type(e)), str(e))) |
|
156 else: |
|
157 raise self.failureException('a ValueError exception was supposed to be raised!') |
|
158 |
|
159 def test_with_tag_instance(self): |
|
160 self.assertEquals(get_tag(self.cheese), self.cheese) |
|
161 |
|
162 def test_with_string(self): |
|
163 self.assertEquals(get_tag('cheese'), self.cheese) |
|
164 |
|
165 def test_with_primary_key(self): |
|
166 self.assertEquals(get_tag(self.cheese.id), self.cheese) |
|
167 |
|
168 def test_nonexistent_tag(self): |
|
169 self.assertEquals(get_tag('mouse'), None) |
|
170 |
|
171 class TestCalculateCloud(TestCase): |
|
172 def setUp(self): |
|
173 self.tags = [] |
|
174 for line in open(os.path.join(os.path.dirname(__file__), 'tags.txt')).readlines(): |
|
175 name, count = line.rstrip().split() |
|
176 tag = Tag(name=name) |
|
177 tag.count = int(count) |
|
178 self.tags.append(tag) |
|
179 |
|
180 def test_default_distribution(self): |
|
181 sizes = {} |
|
182 for tag in calculate_cloud(self.tags, steps=5): |
|
183 sizes[tag.font_size] = sizes.get(tag.font_size, 0) + 1 |
|
184 |
|
185 # This isn't a pre-calculated test, just making sure it's consistent |
|
186 self.assertEquals(sizes[1], 48) |
|
187 self.assertEquals(sizes[2], 30) |
|
188 self.assertEquals(sizes[3], 19) |
|
189 self.assertEquals(sizes[4], 15) |
|
190 self.assertEquals(sizes[5], 10) |
|
191 |
|
192 def test_linear_distribution(self): |
|
193 sizes = {} |
|
194 for tag in calculate_cloud(self.tags, steps=5, distribution=LINEAR): |
|
195 sizes[tag.font_size] = sizes.get(tag.font_size, 0) + 1 |
|
196 |
|
197 # This isn't a pre-calculated test, just making sure it's consistent |
|
198 self.assertEquals(sizes[1], 97) |
|
199 self.assertEquals(sizes[2], 12) |
|
200 self.assertEquals(sizes[3], 7) |
|
201 self.assertEquals(sizes[4], 2) |
|
202 self.assertEquals(sizes[5], 4) |
|
203 |
|
204 def test_invalid_distribution(self): |
|
205 try: |
|
206 calculate_cloud(self.tags, steps=5, distribution='cheese') |
|
207 except ValueError, ve: |
|
208 self.assertEquals(str(ve), 'Invalid distribution algorithm specified: cheese.') |
|
209 except Exception, e: |
|
210 raise self.failureException('the wrong type of exception was raised: type [%s] value [%s]' %\ |
|
211 (str(type(e)), str(e))) |
|
212 else: |
|
213 raise self.failureException('a ValueError exception was supposed to be raised!') |
|
214 |
|
215 ########### |
|
216 # Tagging # |
|
217 ########### |
|
218 |
|
219 class TestBasicTagging(TestCase): |
|
220 def setUp(self): |
|
221 self.dead_parrot = Parrot.objects.create(state='dead') |
|
222 |
|
223 def test_update_tags(self): |
|
224 Tag.objects.update_tags(self.dead_parrot, 'foo,bar,"ter"') |
|
225 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
226 self.assertEquals(len(tags), 3) |
|
227 self.failUnless(get_tag('foo') in tags) |
|
228 self.failUnless(get_tag('bar') in tags) |
|
229 self.failUnless(get_tag('ter') in tags) |
|
230 |
|
231 Tag.objects.update_tags(self.dead_parrot, '"foo" bar "baz"') |
|
232 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
233 self.assertEquals(len(tags), 3) |
|
234 self.failUnless(get_tag('bar') in tags) |
|
235 self.failUnless(get_tag('baz') in tags) |
|
236 self.failUnless(get_tag('foo') in tags) |
|
237 |
|
238 def test_add_tag(self): |
|
239 # start off in a known, mildly interesting state |
|
240 Tag.objects.update_tags(self.dead_parrot, 'foo bar baz') |
|
241 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
242 self.assertEquals(len(tags), 3) |
|
243 self.failUnless(get_tag('bar') in tags) |
|
244 self.failUnless(get_tag('baz') in tags) |
|
245 self.failUnless(get_tag('foo') in tags) |
|
246 |
|
247 # try to add a tag that already exists |
|
248 Tag.objects.add_tag(self.dead_parrot, 'foo') |
|
249 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
250 self.assertEquals(len(tags), 3) |
|
251 self.failUnless(get_tag('bar') in tags) |
|
252 self.failUnless(get_tag('baz') in tags) |
|
253 self.failUnless(get_tag('foo') in tags) |
|
254 |
|
255 # now add a tag that doesn't already exist |
|
256 Tag.objects.add_tag(self.dead_parrot, 'zip') |
|
257 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
258 self.assertEquals(len(tags), 4) |
|
259 self.failUnless(get_tag('zip') in tags) |
|
260 self.failUnless(get_tag('bar') in tags) |
|
261 self.failUnless(get_tag('baz') in tags) |
|
262 self.failUnless(get_tag('foo') in tags) |
|
263 |
|
264 def test_add_tag_invalid_input_no_tags_specified(self): |
|
265 # start off in a known, mildly interesting state |
|
266 Tag.objects.update_tags(self.dead_parrot, 'foo bar baz') |
|
267 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
268 self.assertEquals(len(tags), 3) |
|
269 self.failUnless(get_tag('bar') in tags) |
|
270 self.failUnless(get_tag('baz') in tags) |
|
271 self.failUnless(get_tag('foo') in tags) |
|
272 |
|
273 try: |
|
274 Tag.objects.add_tag(self.dead_parrot, ' ') |
|
275 except AttributeError, ae: |
|
276 self.assertEquals(str(ae), 'No tags were given: " ".') |
|
277 except Exception, e: |
|
278 raise self.failureException('the wrong type of exception was raised: type [%s] value [%s]' %\ |
|
279 (str(type(e)), str(e))) |
|
280 else: |
|
281 raise self.failureException('an AttributeError exception was supposed to be raised!') |
|
282 |
|
283 def test_add_tag_invalid_input_multiple_tags_specified(self): |
|
284 # start off in a known, mildly interesting state |
|
285 Tag.objects.update_tags(self.dead_parrot, 'foo bar baz') |
|
286 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
287 self.assertEquals(len(tags), 3) |
|
288 self.failUnless(get_tag('bar') in tags) |
|
289 self.failUnless(get_tag('baz') in tags) |
|
290 self.failUnless(get_tag('foo') in tags) |
|
291 |
|
292 try: |
|
293 Tag.objects.add_tag(self.dead_parrot, 'one two') |
|
294 except AttributeError, ae: |
|
295 self.assertEquals(str(ae), 'Multiple tags were given: "one two".') |
|
296 except Exception, e: |
|
297 raise self.failureException('the wrong type of exception was raised: type [%s] value [%s]' %\ |
|
298 (str(type(e)), str(e))) |
|
299 else: |
|
300 raise self.failureException('an AttributeError exception was supposed to be raised!') |
|
301 |
|
302 def test_update_tags_exotic_characters(self): |
|
303 # start off in a known, mildly interesting state |
|
304 Tag.objects.update_tags(self.dead_parrot, 'foo bar baz') |
|
305 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
306 self.assertEquals(len(tags), 3) |
|
307 self.failUnless(get_tag('bar') in tags) |
|
308 self.failUnless(get_tag('baz') in tags) |
|
309 self.failUnless(get_tag('foo') in tags) |
|
310 |
|
311 Tag.objects.update_tags(self.dead_parrot, u'ŠĐĆŽćžšđ') |
|
312 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
313 self.assertEquals(len(tags), 1) |
|
314 self.assertEquals(tags[0].name, u'ŠĐĆŽćžšđ') |
|
315 |
|
316 Tag.objects.update_tags(self.dead_parrot, u'你好') |
|
317 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
318 self.assertEquals(len(tags), 1) |
|
319 self.assertEquals(tags[0].name, u'你好') |
|
320 |
|
321 def test_update_tags_with_none(self): |
|
322 # start off in a known, mildly interesting state |
|
323 Tag.objects.update_tags(self.dead_parrot, 'foo bar baz') |
|
324 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
325 self.assertEquals(len(tags), 3) |
|
326 self.failUnless(get_tag('bar') in tags) |
|
327 self.failUnless(get_tag('baz') in tags) |
|
328 self.failUnless(get_tag('foo') in tags) |
|
329 |
|
330 Tag.objects.update_tags(self.dead_parrot, None) |
|
331 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
332 self.assertEquals(len(tags), 0) |
|
333 |
|
334 class TestModelTagField(TestCase): |
|
335 """ Test the 'tags' field on models. """ |
|
336 |
|
337 def test_create_with_tags_specified(self): |
|
338 f1 = FormTest.objects.create(tags=u'test3 test2 test1') |
|
339 tags = Tag.objects.get_for_object(f1) |
|
340 test1_tag = get_tag('test1') |
|
341 test2_tag = get_tag('test2') |
|
342 test3_tag = get_tag('test3') |
|
343 self.failUnless(None not in (test1_tag, test2_tag, test3_tag)) |
|
344 self.assertEquals(len(tags), 3) |
|
345 self.failUnless(test1_tag in tags) |
|
346 self.failUnless(test2_tag in tags) |
|
347 self.failUnless(test3_tag in tags) |
|
348 |
|
349 def test_update_via_tags_field(self): |
|
350 f1 = FormTest.objects.create(tags=u'test3 test2 test1') |
|
351 tags = Tag.objects.get_for_object(f1) |
|
352 test1_tag = get_tag('test1') |
|
353 test2_tag = get_tag('test2') |
|
354 test3_tag = get_tag('test3') |
|
355 self.failUnless(None not in (test1_tag, test2_tag, test3_tag)) |
|
356 self.assertEquals(len(tags), 3) |
|
357 self.failUnless(test1_tag in tags) |
|
358 self.failUnless(test2_tag in tags) |
|
359 self.failUnless(test3_tag in tags) |
|
360 |
|
361 f1.tags = u'test4' |
|
362 f1.save() |
|
363 tags = Tag.objects.get_for_object(f1) |
|
364 test4_tag = get_tag('test4') |
|
365 self.assertEquals(len(tags), 1) |
|
366 self.assertEquals(tags[0], test4_tag) |
|
367 |
|
368 f1.tags = '' |
|
369 f1.save() |
|
370 tags = Tag.objects.get_for_object(f1) |
|
371 self.assertEquals(len(tags), 0) |
|
372 |
|
373 class TestSettings(TestCase): |
|
374 def setUp(self): |
|
375 self.original_force_lower_case_tags = settings.FORCE_LOWERCASE_TAGS |
|
376 self.dead_parrot = Parrot.objects.create(state='dead') |
|
377 |
|
378 def tearDown(self): |
|
379 settings.FORCE_LOWERCASE_TAGS = self.original_force_lower_case_tags |
|
380 |
|
381 def test_force_lowercase_tags(self): |
|
382 """ Test forcing tags to lowercase. """ |
|
383 |
|
384 settings.FORCE_LOWERCASE_TAGS = True |
|
385 |
|
386 Tag.objects.update_tags(self.dead_parrot, 'foO bAr Ter') |
|
387 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
388 self.assertEquals(len(tags), 3) |
|
389 foo_tag = get_tag('foo') |
|
390 bar_tag = get_tag('bar') |
|
391 ter_tag = get_tag('ter') |
|
392 self.failUnless(foo_tag in tags) |
|
393 self.failUnless(bar_tag in tags) |
|
394 self.failUnless(ter_tag in tags) |
|
395 |
|
396 Tag.objects.update_tags(self.dead_parrot, 'foO bAr baZ') |
|
397 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
398 baz_tag = get_tag('baz') |
|
399 self.assertEquals(len(tags), 3) |
|
400 self.failUnless(bar_tag in tags) |
|
401 self.failUnless(baz_tag in tags) |
|
402 self.failUnless(foo_tag in tags) |
|
403 |
|
404 Tag.objects.add_tag(self.dead_parrot, 'FOO') |
|
405 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
406 self.assertEquals(len(tags), 3) |
|
407 self.failUnless(bar_tag in tags) |
|
408 self.failUnless(baz_tag in tags) |
|
409 self.failUnless(foo_tag in tags) |
|
410 |
|
411 Tag.objects.add_tag(self.dead_parrot, 'Zip') |
|
412 tags = Tag.objects.get_for_object(self.dead_parrot) |
|
413 self.assertEquals(len(tags), 4) |
|
414 zip_tag = get_tag('zip') |
|
415 self.failUnless(bar_tag in tags) |
|
416 self.failUnless(baz_tag in tags) |
|
417 self.failUnless(foo_tag in tags) |
|
418 self.failUnless(zip_tag in tags) |
|
419 |
|
420 f1 = FormTest.objects.create() |
|
421 f1.tags = u'TEST5' |
|
422 f1.save() |
|
423 tags = Tag.objects.get_for_object(f1) |
|
424 test5_tag = get_tag('test5') |
|
425 self.assertEquals(len(tags), 1) |
|
426 self.failUnless(test5_tag in tags) |
|
427 self.assertEquals(f1.tags, u'test5') |
|
428 |
|
429 class TestTagUsageForModelBaseCase(TestCase): |
|
430 def test_tag_usage_for_model_empty(self): |
|
431 self.assertEquals(Tag.objects.usage_for_model(Parrot), []) |
|
432 |
|
433 class TestTagUsageForModel(TestCase): |
|
434 def setUp(self): |
|
435 parrot_details = ( |
|
436 ('pining for the fjords', 9, True, 'foo bar'), |
|
437 ('passed on', 6, False, 'bar baz ter'), |
|
438 ('no more', 4, True, 'foo ter'), |
|
439 ('late', 2, False, 'bar ter'), |
|
440 ) |
|
441 |
|
442 for state, perch_size, perch_smelly, tags in parrot_details: |
|
443 perch = Perch.objects.create(size=perch_size, smelly=perch_smelly) |
|
444 parrot = Parrot.objects.create(state=state, perch=perch) |
|
445 Tag.objects.update_tags(parrot, tags) |
|
446 |
|
447 def test_tag_usage_for_model(self): |
|
448 tag_usage = Tag.objects.usage_for_model(Parrot, counts=True) |
|
449 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
450 self.assertEquals(len(relevant_attribute_list), 4) |
|
451 self.failUnless((u'bar', 3) in relevant_attribute_list) |
|
452 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
453 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
454 self.failUnless((u'ter', 3) in relevant_attribute_list) |
|
455 |
|
456 def test_tag_usage_for_model_with_min_count(self): |
|
457 tag_usage = Tag.objects.usage_for_model(Parrot, min_count = 2) |
|
458 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
459 self.assertEquals(len(relevant_attribute_list), 3) |
|
460 self.failUnless((u'bar', 3) in relevant_attribute_list) |
|
461 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
462 self.failUnless((u'ter', 3) in relevant_attribute_list) |
|
463 |
|
464 def test_tag_usage_with_filter_on_model_objects(self): |
|
465 tag_usage = Tag.objects.usage_for_model(Parrot, counts=True, filters=dict(state='no more')) |
|
466 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
467 self.assertEquals(len(relevant_attribute_list), 2) |
|
468 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
469 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
470 |
|
471 tag_usage = Tag.objects.usage_for_model(Parrot, counts=True, filters=dict(state__startswith='p')) |
|
472 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
473 self.assertEquals(len(relevant_attribute_list), 4) |
|
474 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
475 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
476 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
477 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
478 |
|
479 tag_usage = Tag.objects.usage_for_model(Parrot, counts=True, filters=dict(perch__size__gt=4)) |
|
480 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
481 self.assertEquals(len(relevant_attribute_list), 4) |
|
482 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
483 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
484 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
485 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
486 |
|
487 tag_usage = Tag.objects.usage_for_model(Parrot, counts=True, filters=dict(perch__smelly=True)) |
|
488 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
489 self.assertEquals(len(relevant_attribute_list), 3) |
|
490 self.failUnless((u'bar', 1) in relevant_attribute_list) |
|
491 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
492 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
493 |
|
494 tag_usage = Tag.objects.usage_for_model(Parrot, min_count=2, filters=dict(perch__smelly=True)) |
|
495 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
496 self.assertEquals(len(relevant_attribute_list), 1) |
|
497 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
498 |
|
499 tag_usage = Tag.objects.usage_for_model(Parrot, filters=dict(perch__size__gt=4)) |
|
500 relevant_attribute_list = [(tag.name, hasattr(tag, 'counts')) for tag in tag_usage] |
|
501 self.assertEquals(len(relevant_attribute_list), 4) |
|
502 self.failUnless((u'bar', False) in relevant_attribute_list) |
|
503 self.failUnless((u'baz', False) in relevant_attribute_list) |
|
504 self.failUnless((u'foo', False) in relevant_attribute_list) |
|
505 self.failUnless((u'ter', False) in relevant_attribute_list) |
|
506 |
|
507 tag_usage = Tag.objects.usage_for_model(Parrot, filters=dict(perch__size__gt=99)) |
|
508 relevant_attribute_list = [(tag.name, hasattr(tag, 'counts')) for tag in tag_usage] |
|
509 self.assertEquals(len(relevant_attribute_list), 0) |
|
510 |
|
511 class TestTagsRelatedForModel(TestCase): |
|
512 def setUp(self): |
|
513 parrot_details = ( |
|
514 ('pining for the fjords', 9, True, 'foo bar'), |
|
515 ('passed on', 6, False, 'bar baz ter'), |
|
516 ('no more', 4, True, 'foo ter'), |
|
517 ('late', 2, False, 'bar ter'), |
|
518 ) |
|
519 |
|
520 for state, perch_size, perch_smelly, tags in parrot_details: |
|
521 perch = Perch.objects.create(size=perch_size, smelly=perch_smelly) |
|
522 parrot = Parrot.objects.create(state=state, perch=perch) |
|
523 Tag.objects.update_tags(parrot, tags) |
|
524 |
|
525 def test_related_for_model_with_tag_query_sets_as_input(self): |
|
526 related_tags = Tag.objects.related_for_model(Tag.objects.filter(name__in=['bar']), Parrot, counts=True) |
|
527 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
528 self.assertEquals(len(relevant_attribute_list), 3) |
|
529 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
530 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
531 self.failUnless((u'ter', 2) in relevant_attribute_list) |
|
532 |
|
533 related_tags = Tag.objects.related_for_model(Tag.objects.filter(name__in=['bar']), Parrot, min_count=2) |
|
534 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
535 self.assertEquals(len(relevant_attribute_list), 1) |
|
536 self.failUnless((u'ter', 2) in relevant_attribute_list) |
|
537 |
|
538 related_tags = Tag.objects.related_for_model(Tag.objects.filter(name__in=['bar']), Parrot, counts=False) |
|
539 relevant_attribute_list = [tag.name for tag in related_tags] |
|
540 self.assertEquals(len(relevant_attribute_list), 3) |
|
541 self.failUnless(u'baz' in relevant_attribute_list) |
|
542 self.failUnless(u'foo' in relevant_attribute_list) |
|
543 self.failUnless(u'ter' in relevant_attribute_list) |
|
544 |
|
545 related_tags = Tag.objects.related_for_model(Tag.objects.filter(name__in=['bar', 'ter']), Parrot, counts=True) |
|
546 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
547 self.assertEquals(len(relevant_attribute_list), 1) |
|
548 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
549 |
|
550 related_tags = Tag.objects.related_for_model(Tag.objects.filter(name__in=['bar', 'ter', 'baz']), Parrot, counts=True) |
|
551 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
552 self.assertEquals(len(relevant_attribute_list), 0) |
|
553 |
|
554 def test_related_for_model_with_tag_strings_as_input(self): |
|
555 # Once again, with feeling (strings) |
|
556 related_tags = Tag.objects.related_for_model('bar', Parrot, counts=True) |
|
557 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
558 self.assertEquals(len(relevant_attribute_list), 3) |
|
559 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
560 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
561 self.failUnless((u'ter', 2) in relevant_attribute_list) |
|
562 |
|
563 related_tags = Tag.objects.related_for_model('bar', Parrot, min_count=2) |
|
564 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
565 self.assertEquals(len(relevant_attribute_list), 1) |
|
566 self.failUnless((u'ter', 2) in relevant_attribute_list) |
|
567 |
|
568 related_tags = Tag.objects.related_for_model('bar', Parrot, counts=False) |
|
569 relevant_attribute_list = [tag.name for tag in related_tags] |
|
570 self.assertEquals(len(relevant_attribute_list), 3) |
|
571 self.failUnless(u'baz' in relevant_attribute_list) |
|
572 self.failUnless(u'foo' in relevant_attribute_list) |
|
573 self.failUnless(u'ter' in relevant_attribute_list) |
|
574 |
|
575 related_tags = Tag.objects.related_for_model(['bar', 'ter'], Parrot, counts=True) |
|
576 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
577 self.assertEquals(len(relevant_attribute_list), 1) |
|
578 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
579 |
|
580 related_tags = Tag.objects.related_for_model(['bar', 'ter', 'baz'], Parrot, counts=True) |
|
581 relevant_attribute_list = [(tag.name, tag.count) for tag in related_tags] |
|
582 self.assertEquals(len(relevant_attribute_list), 0) |
|
583 |
|
584 class TestGetTaggedObjectsByModel(TestCase): |
|
585 def setUp(self): |
|
586 parrot_details = ( |
|
587 ('pining for the fjords', 9, True, 'foo bar'), |
|
588 ('passed on', 6, False, 'bar baz ter'), |
|
589 ('no more', 4, True, 'foo ter'), |
|
590 ('late', 2, False, 'bar ter'), |
|
591 ) |
|
592 |
|
593 for state, perch_size, perch_smelly, tags in parrot_details: |
|
594 perch = Perch.objects.create(size=perch_size, smelly=perch_smelly) |
|
595 parrot = Parrot.objects.create(state=state, perch=perch) |
|
596 Tag.objects.update_tags(parrot, tags) |
|
597 |
|
598 self.foo = Tag.objects.get(name='foo') |
|
599 self.bar = Tag.objects.get(name='bar') |
|
600 self.baz = Tag.objects.get(name='baz') |
|
601 self.ter = Tag.objects.get(name='ter') |
|
602 |
|
603 self.pining_for_the_fjords_parrot = Parrot.objects.get(state='pining for the fjords') |
|
604 self.passed_on_parrot = Parrot.objects.get(state='passed on') |
|
605 self.no_more_parrot = Parrot.objects.get(state='no more') |
|
606 self.late_parrot = Parrot.objects.get(state='late') |
|
607 |
|
608 def test_get_by_model_simple(self): |
|
609 parrots = TaggedItem.objects.get_by_model(Parrot, self.foo) |
|
610 self.assertEquals(len(parrots), 2) |
|
611 self.failUnless(self.no_more_parrot in parrots) |
|
612 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
613 |
|
614 parrots = TaggedItem.objects.get_by_model(Parrot, self.bar) |
|
615 self.assertEquals(len(parrots), 3) |
|
616 self.failUnless(self.late_parrot in parrots) |
|
617 self.failUnless(self.passed_on_parrot in parrots) |
|
618 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
619 |
|
620 def test_get_by_model_intersection(self): |
|
621 parrots = TaggedItem.objects.get_by_model(Parrot, [self.foo, self.baz]) |
|
622 self.assertEquals(len(parrots), 0) |
|
623 |
|
624 parrots = TaggedItem.objects.get_by_model(Parrot, [self.foo, self.bar]) |
|
625 self.assertEquals(len(parrots), 1) |
|
626 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
627 |
|
628 parrots = TaggedItem.objects.get_by_model(Parrot, [self.bar, self.ter]) |
|
629 self.assertEquals(len(parrots), 2) |
|
630 self.failUnless(self.late_parrot in parrots) |
|
631 self.failUnless(self.passed_on_parrot in parrots) |
|
632 |
|
633 # Issue 114 - Intersection with non-existant tags |
|
634 parrots = TaggedItem.objects.get_intersection_by_model(Parrot, []) |
|
635 self.assertEquals(len(parrots), 0) |
|
636 |
|
637 def test_get_by_model_with_tag_querysets_as_input(self): |
|
638 parrots = TaggedItem.objects.get_by_model(Parrot, Tag.objects.filter(name__in=['foo', 'baz'])) |
|
639 self.assertEquals(len(parrots), 0) |
|
640 |
|
641 parrots = TaggedItem.objects.get_by_model(Parrot, Tag.objects.filter(name__in=['foo', 'bar'])) |
|
642 self.assertEquals(len(parrots), 1) |
|
643 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
644 |
|
645 parrots = TaggedItem.objects.get_by_model(Parrot, Tag.objects.filter(name__in=['bar', 'ter'])) |
|
646 self.assertEquals(len(parrots), 2) |
|
647 self.failUnless(self.late_parrot in parrots) |
|
648 self.failUnless(self.passed_on_parrot in parrots) |
|
649 |
|
650 def test_get_by_model_with_strings_as_input(self): |
|
651 parrots = TaggedItem.objects.get_by_model(Parrot, 'foo baz') |
|
652 self.assertEquals(len(parrots), 0) |
|
653 |
|
654 parrots = TaggedItem.objects.get_by_model(Parrot, 'foo bar') |
|
655 self.assertEquals(len(parrots), 1) |
|
656 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
657 |
|
658 parrots = TaggedItem.objects.get_by_model(Parrot, 'bar ter') |
|
659 self.assertEquals(len(parrots), 2) |
|
660 self.failUnless(self.late_parrot in parrots) |
|
661 self.failUnless(self.passed_on_parrot in parrots) |
|
662 |
|
663 def test_get_by_model_with_lists_of_strings_as_input(self): |
|
664 parrots = TaggedItem.objects.get_by_model(Parrot, ['foo', 'baz']) |
|
665 self.assertEquals(len(parrots), 0) |
|
666 |
|
667 parrots = TaggedItem.objects.get_by_model(Parrot, ['foo', 'bar']) |
|
668 self.assertEquals(len(parrots), 1) |
|
669 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
670 |
|
671 parrots = TaggedItem.objects.get_by_model(Parrot, ['bar', 'ter']) |
|
672 self.assertEquals(len(parrots), 2) |
|
673 self.failUnless(self.late_parrot in parrots) |
|
674 self.failUnless(self.passed_on_parrot in parrots) |
|
675 |
|
676 def test_get_by_nonexistent_tag(self): |
|
677 # Issue 50 - Get by non-existent tag |
|
678 parrots = TaggedItem.objects.get_by_model(Parrot, 'argatrons') |
|
679 self.assertEquals(len(parrots), 0) |
|
680 |
|
681 def test_get_union_by_model(self): |
|
682 parrots = TaggedItem.objects.get_union_by_model(Parrot, ['foo', 'ter']) |
|
683 self.assertEquals(len(parrots), 4) |
|
684 self.failUnless(self.late_parrot in parrots) |
|
685 self.failUnless(self.no_more_parrot in parrots) |
|
686 self.failUnless(self.passed_on_parrot in parrots) |
|
687 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
688 |
|
689 parrots = TaggedItem.objects.get_union_by_model(Parrot, ['bar', 'baz']) |
|
690 self.assertEquals(len(parrots), 3) |
|
691 self.failUnless(self.late_parrot in parrots) |
|
692 self.failUnless(self.passed_on_parrot in parrots) |
|
693 self.failUnless(self.pining_for_the_fjords_parrot in parrots) |
|
694 |
|
695 # Issue 114 - Union with non-existant tags |
|
696 parrots = TaggedItem.objects.get_union_by_model(Parrot, []) |
|
697 self.assertEquals(len(parrots), 0) |
|
698 |
|
699 class TestGetRelatedTaggedItems(TestCase): |
|
700 def setUp(self): |
|
701 parrot_details = ( |
|
702 ('pining for the fjords', 9, True, 'foo bar'), |
|
703 ('passed on', 6, False, 'bar baz ter'), |
|
704 ('no more', 4, True, 'foo ter'), |
|
705 ('late', 2, False, 'bar ter'), |
|
706 ) |
|
707 |
|
708 for state, perch_size, perch_smelly, tags in parrot_details: |
|
709 perch = Perch.objects.create(size=perch_size, smelly=perch_smelly) |
|
710 parrot = Parrot.objects.create(state=state, perch=perch) |
|
711 Tag.objects.update_tags(parrot, tags) |
|
712 |
|
713 self.l1 = Link.objects.create(name='link 1') |
|
714 Tag.objects.update_tags(self.l1, 'tag1 tag2 tag3 tag4 tag5') |
|
715 self.l2 = Link.objects.create(name='link 2') |
|
716 Tag.objects.update_tags(self.l2, 'tag1 tag2 tag3') |
|
717 self.l3 = Link.objects.create(name='link 3') |
|
718 Tag.objects.update_tags(self.l3, 'tag1') |
|
719 self.l4 = Link.objects.create(name='link 4') |
|
720 |
|
721 self.a1 = Article.objects.create(name='article 1') |
|
722 Tag.objects.update_tags(self.a1, 'tag1 tag2 tag3 tag4') |
|
723 |
|
724 def test_get_related_objects_of_same_model(self): |
|
725 related_objects = TaggedItem.objects.get_related(self.l1, Link) |
|
726 self.assertEquals(len(related_objects), 2) |
|
727 self.failUnless(self.l2 in related_objects) |
|
728 self.failUnless(self.l3 in related_objects) |
|
729 |
|
730 related_objects = TaggedItem.objects.get_related(self.l4, Link) |
|
731 self.assertEquals(len(related_objects), 0) |
|
732 |
|
733 def test_get_related_objects_of_same_model_limited_number_of_results(self): |
|
734 # This fails on Oracle because it has no support for a 'LIMIT' clause. |
|
735 # See http://asktom.oracle.com/pls/asktom/f?p=100:11:0::::P11_QUESTION_ID:127412348064 |
|
736 |
|
737 # ask for no more than 1 result |
|
738 related_objects = TaggedItem.objects.get_related(self.l1, Link, num=1) |
|
739 self.assertEquals(len(related_objects), 1) |
|
740 self.failUnless(self.l2 in related_objects) |
|
741 |
|
742 def test_get_related_objects_of_same_model_limit_related_items(self): |
|
743 related_objects = TaggedItem.objects.get_related(self.l1, Link.objects.exclude(name='link 3')) |
|
744 self.assertEquals(len(related_objects), 1) |
|
745 self.failUnless(self.l2 in related_objects) |
|
746 |
|
747 def test_get_related_objects_of_different_model(self): |
|
748 related_objects = TaggedItem.objects.get_related(self.a1, Link) |
|
749 self.assertEquals(len(related_objects), 3) |
|
750 self.failUnless(self.l1 in related_objects) |
|
751 self.failUnless(self.l2 in related_objects) |
|
752 self.failUnless(self.l3 in related_objects) |
|
753 |
|
754 Tag.objects.update_tags(self.a1, 'tag6') |
|
755 related_objects = TaggedItem.objects.get_related(self.a1, Link) |
|
756 self.assertEquals(len(related_objects), 0) |
|
757 |
|
758 class TestTagUsageForQuerySet(TestCase): |
|
759 def setUp(self): |
|
760 parrot_details = ( |
|
761 ('pining for the fjords', 9, True, 'foo bar'), |
|
762 ('passed on', 6, False, 'bar baz ter'), |
|
763 ('no more', 4, True, 'foo ter'), |
|
764 ('late', 2, False, 'bar ter'), |
|
765 ) |
|
766 |
|
767 for state, perch_size, perch_smelly, tags in parrot_details: |
|
768 perch = Perch.objects.create(size=perch_size, smelly=perch_smelly) |
|
769 parrot = Parrot.objects.create(state=state, perch=perch) |
|
770 Tag.objects.update_tags(parrot, tags) |
|
771 |
|
772 def test_tag_usage_for_queryset(self): |
|
773 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(state='no more'), counts=True) |
|
774 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
775 self.assertEquals(len(relevant_attribute_list), 2) |
|
776 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
777 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
778 |
|
779 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(state__startswith='p'), counts=True) |
|
780 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
781 self.assertEquals(len(relevant_attribute_list), 4) |
|
782 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
783 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
784 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
785 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
786 |
|
787 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(perch__size__gt=4), counts=True) |
|
788 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
789 self.assertEquals(len(relevant_attribute_list), 4) |
|
790 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
791 self.failUnless((u'baz', 1) in relevant_attribute_list) |
|
792 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
793 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
794 |
|
795 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(perch__smelly=True), counts=True) |
|
796 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
797 self.assertEquals(len(relevant_attribute_list), 3) |
|
798 self.failUnless((u'bar', 1) in relevant_attribute_list) |
|
799 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
800 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
801 |
|
802 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(perch__smelly=True), min_count=2) |
|
803 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
804 self.assertEquals(len(relevant_attribute_list), 1) |
|
805 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
806 |
|
807 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(perch__size__gt=4)) |
|
808 relevant_attribute_list = [(tag.name, hasattr(tag, 'counts')) for tag in tag_usage] |
|
809 self.assertEquals(len(relevant_attribute_list), 4) |
|
810 self.failUnless((u'bar', False) in relevant_attribute_list) |
|
811 self.failUnless((u'baz', False) in relevant_attribute_list) |
|
812 self.failUnless((u'foo', False) in relevant_attribute_list) |
|
813 self.failUnless((u'ter', False) in relevant_attribute_list) |
|
814 |
|
815 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(perch__size__gt=99)) |
|
816 relevant_attribute_list = [(tag.name, hasattr(tag, 'counts')) for tag in tag_usage] |
|
817 self.assertEquals(len(relevant_attribute_list), 0) |
|
818 |
|
819 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(Q(perch__size__gt=6) | Q(state__startswith='l')), counts=True) |
|
820 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
821 self.assertEquals(len(relevant_attribute_list), 3) |
|
822 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
823 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
824 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
825 |
|
826 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(Q(perch__size__gt=6) | Q(state__startswith='l')), min_count=2) |
|
827 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
828 self.assertEquals(len(relevant_attribute_list), 1) |
|
829 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
830 |
|
831 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.filter(Q(perch__size__gt=6) | Q(state__startswith='l'))) |
|
832 relevant_attribute_list = [(tag.name, hasattr(tag, 'counts')) for tag in tag_usage] |
|
833 self.assertEquals(len(relevant_attribute_list), 3) |
|
834 self.failUnless((u'bar', False) in relevant_attribute_list) |
|
835 self.failUnless((u'foo', False) in relevant_attribute_list) |
|
836 self.failUnless((u'ter', False) in relevant_attribute_list) |
|
837 |
|
838 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.exclude(state='passed on'), counts=True) |
|
839 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
840 self.assertEquals(len(relevant_attribute_list), 3) |
|
841 self.failUnless((u'bar', 2) in relevant_attribute_list) |
|
842 self.failUnless((u'foo', 2) in relevant_attribute_list) |
|
843 self.failUnless((u'ter', 2) in relevant_attribute_list) |
|
844 |
|
845 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.exclude(state__startswith='p'), min_count=2) |
|
846 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
847 self.assertEquals(len(relevant_attribute_list), 1) |
|
848 self.failUnless((u'ter', 2) in relevant_attribute_list) |
|
849 |
|
850 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.exclude(Q(perch__size__gt=6) | Q(perch__smelly=False)), counts=True) |
|
851 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
852 self.assertEquals(len(relevant_attribute_list), 2) |
|
853 self.failUnless((u'foo', 1) in relevant_attribute_list) |
|
854 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
855 |
|
856 tag_usage = Tag.objects.usage_for_queryset(Parrot.objects.exclude(perch__smelly=True).filter(state__startswith='l'), counts=True) |
|
857 relevant_attribute_list = [(tag.name, tag.count) for tag in tag_usage] |
|
858 self.assertEquals(len(relevant_attribute_list), 2) |
|
859 self.failUnless((u'bar', 1) in relevant_attribute_list) |
|
860 self.failUnless((u'ter', 1) in relevant_attribute_list) |
|
861 |
|
862 ################ |
|
863 # Model Fields # |
|
864 ################ |
|
865 |
|
866 class TestTagFieldInForms(TestCase): |
|
867 def test_tag_field_in_modelform(self): |
|
868 # Ensure that automatically created forms use TagField |
|
869 class TestForm(forms.ModelForm): |
|
870 class Meta: |
|
871 model = FormTest |
|
872 |
|
873 form = TestForm() |
|
874 self.assertEquals(form.fields['tags'].__class__.__name__, 'TagField') |
|
875 |
|
876 def test_recreation_of_tag_list_string_representations(self): |
|
877 plain = Tag.objects.create(name='plain') |
|
878 spaces = Tag.objects.create(name='spa ces') |
|
879 comma = Tag.objects.create(name='com,ma') |
|
880 self.assertEquals(edit_string_for_tags([plain]), u'plain') |
|
881 self.assertEquals(edit_string_for_tags([plain, spaces]), u'plain, spa ces') |
|
882 self.assertEquals(edit_string_for_tags([plain, spaces, comma]), u'plain, spa ces, "com,ma"') |
|
883 self.assertEquals(edit_string_for_tags([plain, comma]), u'plain "com,ma"') |
|
884 self.assertEquals(edit_string_for_tags([comma, spaces]), u'"com,ma", spa ces') |
|
885 |
|
886 def test_tag_d_validation(self): |
|
887 t = TagField() |
|
888 self.assertEquals(t.clean('foo'), u'foo') |
|
889 self.assertEquals(t.clean('foo bar baz'), u'foo bar baz') |
|
890 self.assertEquals(t.clean('foo,bar,baz'), u'foo,bar,baz') |
|
891 self.assertEquals(t.clean('foo, bar, baz'), u'foo, bar, baz') |
|
892 self.assertEquals(t.clean('foo qwertyuiopasdfghjklzxcvbnmqwertyuiopasdfghjklzxcvb bar'), |
|
893 u'foo qwertyuiopasdfghjklzxcvbnmqwertyuiopasdfghjklzxcvb bar') |
|
894 try: |
|
895 t.clean('foo qwertyuiopasdfghjklzxcvbnmqwertyuiopasdfghjklzxcvbn bar') |
|
896 except forms.ValidationError, ve: |
|
897 self.assertEquals(str(ve), "[u'Each tag may be no more than 50 characters long.']") |
|
898 except Exception, e: |
|
899 raise e |
|
900 else: |
|
901 raise self.failureException('a ValidationError exception was supposed to have been raised.') |