diff potd/tests/tools_tests.py @ 581:ee87ea74d46b

For Django 1.4, rearranged project structure for new manage.py.
author Brian Neal <bgneal@gmail.com>
date Sat, 05 May 2012 17:10:48 -0500
parents gpp/potd/tests/tools_tests.py@ae89ba801e8b
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/potd/tests/tools_tests.py	Sat May 05 17:10:48 2012 -0500
@@ -0,0 +1,35 @@
+"""
+Tests for the pick_potd() function.
+
+"""
+from django.test import TestCase
+from django.contrib.auth.models import User
+
+from potd.models import Current, Photo, Sequence
+from potd.tools import pick_potd
+
+
+class PickPotdTest(TestCase):
+
+    fixtures = ['potd_test.json']
+
+    def test_pick(self):
+
+        pick_potd()
+
+        curr = Current.objects.get(pk=1)
+        self.assertEqual(curr.potd.pk, 2)
+
+    def test_shuffle(self):
+
+        photo = Photo.objects.get(pk=3)
+        curr = Current.objects.get(pk=1)
+        curr.potd = photo
+        curr.save()
+
+        pick_potd()
+
+        ids = Sequence.objects.get(pk=1).seq.split(',')
+        curr = Current.objects.get(pk=1)
+        self.assertEqual(len(ids), 3)
+        self.assertEqual(curr.potd.pk, int(ids[0]))