Mercurial > public > sg101
annotate potd/tests/test_tools.py @ 887:9a15f7c27526
Actually save model object upon change.
This commit was tested on the comments model.
Additional logging added.
Added check for Markdown image references.
Added TODOs after observing behavior on comments.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Tue, 03 Feb 2015 21:09:44 -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])) |