Mercurial > public > sg101
annotate potd/tests/test_tools.py @ 917:0365fdbb4d78
Fix app conflict with messages.
Django's messages app label conflicts with our messages app.
We can't easily rename our label as that will make us rename database tables.
Since our app came first we'll just customize Django messages label.
For Django 1.7.7 upgrade.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Mon, 06 Apr 2015 20:02:25 -0500 |
| parents | 8789299c75b1 |
| children |
| rev | line source |
|---|---|
| bgneal@515 | 1 """ |
| bgneal@515 | 2 Tests for the pick_potd() function. |
| bgneal@515 | 3 |
| bgneal@515 | 4 """ |
| bgneal@515 | 5 from django.test import TestCase |
| bgneal@515 | 6 |
| bgneal@515 | 7 from potd.models import Current, Photo, Sequence |
| bgneal@515 | 8 from potd.tools import pick_potd |
| bgneal@515 | 9 |
| bgneal@515 | 10 |
| bgneal@515 | 11 class PickPotdTest(TestCase): |
| bgneal@515 | 12 |
| bgneal@515 | 13 fixtures = ['potd_test.json'] |
| bgneal@515 | 14 |
| bgneal@515 | 15 def test_pick(self): |
| bgneal@515 | 16 |
| bgneal@515 | 17 pick_potd() |
| bgneal@515 | 18 |
| bgneal@515 | 19 curr = Current.objects.get(pk=1) |
| bgneal@515 | 20 self.assertEqual(curr.potd.pk, 2) |
| bgneal@515 | 21 |
| bgneal@515 | 22 def test_shuffle(self): |
| bgneal@515 | 23 |
| bgneal@515 | 24 photo = Photo.objects.get(pk=3) |
| bgneal@515 | 25 curr = Current.objects.get(pk=1) |
| bgneal@515 | 26 curr.potd = photo |
| bgneal@515 | 27 curr.save() |
| bgneal@515 | 28 |
| bgneal@515 | 29 pick_potd() |
| bgneal@515 | 30 |
| bgneal@515 | 31 ids = Sequence.objects.get(pk=1).seq.split(',') |
| bgneal@515 | 32 curr = Current.objects.get(pk=1) |
| bgneal@515 | 33 self.assertEqual(len(ids), 3) |
| bgneal@515 | 34 self.assertEqual(curr.potd.pk, int(ids[0])) |