Mercurial > public > sg101

annotate potd/tests/test_tools.py @ 895:e7c549e4dbf7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add counter and timer.
author Brian Neal <bgneal@gmail.com>
date Thu, 19 Feb 2015 21:02:21 -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]))