Mercurial > public > sg101
annotate potd/tests/test_tools.py @ 1195:7fc6c42b2f5b
Adding a local user photo upload option.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Sun, 07 May 2023 16:22:13 -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])) |