Add testing for the 'random' page

diff --git a/quotes/tests.py b/quotes/tests.py
index 3836f4e5e829dce28360b52e25235f0b2215dc40..772d4e3491d90f143de4b1cdab5d22107dd81c32 100644 (file)
--- a/quotes/tests.py
+++ b/quotes/tests.py
@@ -1,8 +1,8 @@
-import sys
 from django.test import TestCase, Client
 
 # Create your tests here.
 from .models import Tag, Author, Work, Quote
+import re
 
 class QuoteTest(TestCase):
     def setUp(self):
@@ -53,6 +53,24 @@ class ViewsTest(TestCase):
             self.assertTrue(q.text in content, content)
             self.assertTrue(q.notes in content, content)
 
+    def test_random(self):
+        seen = {}
+        for q in Quote.objects.all():
+            seen[q.id] = False
+
+        # <a href="{% url 'quotes:onequote' quote.id %}">Permalink</a>
+        permalinkre = re.compile(r'([0-9]+).*Permalink')
+
+        for i in xrange(100):
+            content = self.getPage('random')
+            m = permalinkre.search(content)
+            self.assertIsNotNone(m, content)
+            quoteid = int(m.group(1))
+            self.assertIn(quoteid, seen)
+            seen[quoteid] = True
+        for q in Quote.objects.all():
+            self.assertTrue(seen[q.id])
+
     def test_views_all_data(self):
         q = Quote.objects.filter(text__startswith="Quote01")
         self.assertEqual(q.count(), 1)

diff --git a/quotes/urls.py b/quotes/urls.py
index 3e905bc5393db6a99318fa8b9be39570922412dd..8864db0ce36ce98f2fea743bf80a776adb939919 100644 (file)
--- a/quotes/urls.py
+++ b/quotes/urls.py
@@ -5,6 +5,7 @@ from . import views
 urlpatterns = [
     url(r'^$', views.random, name='random'),
     url(r'^random$', views.random, name='random'),
+    url(r'^random/$', views.random, name='random'),
     url(r'^show/(?P<quote_id>[0-9]+)/$', views.onequote, name="onequote"),
     url(r'^tag/(?P<tag_id>[0-9]+)/$', views.tags, name="tags"),
     url(r'^author/(?P<author_id>[0-9]+)/$', views.author, name="author"),