Mercurial > public > sg101
annotate potd/tests/test_tools.py @ 1160:b84ce2d7e93d
New template for displaying django messages.
New common template for rendering django messages in the new V3 Foundation
style.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Tue, 24 Jan 2017 19:22:29 -0600 |
| 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])) |