sg101: potd/tests/test_tools.py annotate

Remove affiliate links on donation index page.

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]))

Mercurial > public > sg101