Converted contrib/auth/tokens doctests to unittests. We've always said "no more" to doctests.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@14100 bcc190cf-cafb-0310-a4f2-bffc1f526a37

django/contrib/auth/tests/__init__.py

@@ -5,12 +5,8 @@ from django.contrib.auth.tests.forms import UserCreationFormTest, Authentication
 from django.contrib.auth.tests.remote_user \
         import RemoteUserTest, RemoteUserNoCreateTest, RemoteUserCustomTest
 from django.contrib.auth.tests.models import ProfileTestCase
-from django.contrib.auth.tests.tokens import TOKEN_GENERATOR_TESTS
+from django.contrib.auth.tests.tokens import TokenGeneratorTest
 from django.contrib.auth.tests.views \
         import PasswordResetTest, ChangePasswordTest, LoginTest, LogoutTest
 
 # The password for the fixture data users is 'password'
-
-__test__ = {
-    'TOKEN_GENERATOR_TESTS': TOKEN_GENERATOR_TESTS,
-}

django/contrib/auth/tests/tokens.py

@@ -1,37 +1,52 @@
-TOKEN_GENERATOR_TESTS = """
+from datetime import date, timedelta
->>> from django.contrib.auth.models import User, AnonymousUser
->>> from django.contrib.auth.tokens import PasswordResetTokenGenerator
->>> from django.conf import settings
->>> u = User.objects.create_user('tokentestuser', 'test2@example.com', 'testpw')
->>> p0 = PasswordResetTokenGenerator()
->>> tk1 = p0.make_token(u)
->>> p0.check_token(u, tk1)
-True
 
->>> u = User.objects.create_user('comebackkid', 'test3@example.com', 'testpw')
+from django.conf import settings
->>> p0 = PasswordResetTokenGenerator()
+from django.contrib.auth.models import User, AnonymousUser
->>> tk1 = p0.make_token(u)
+from django.contrib.auth.tokens import PasswordResetTokenGenerator
->>> reload = User.objects.get(username='comebackkid')
+from django.test import TestCase
->>> tk2 = p0.make_token(reload)
->>> tk1 == tk2
-True
 
-Tests to ensure we can use the token after n days, but no greater.
-Use a mocked version of PasswordResetTokenGenerator so we can change
-the value of 'today'
 
->>> class Mocked(PasswordResetTokenGenerator):
+class TokenGeneratorTest(TestCase):
-...     def __init__(self, today):
-...         self._today_val = today
-...     def _today(self):
-...         return self._today_val
 
->>> from datetime import date, timedelta
+    def test_make_token(self):
->>> p1 = Mocked(date.today() + timedelta(settings.PASSWORD_RESET_TIMEOUT_DAYS))
+        """
->>> p1.check_token(u, tk1)
+        Ensure that we can make a token and that it is valid
-True
+        """
->>> p2 = Mocked(date.today() + timedelta(settings.PASSWORD_RESET_TIMEOUT_DAYS + 1))
+        user = User.objects.create_user('tokentestuser', 'test2@example.com', 'testpw')
->>> p2.check_token(u, tk1)
+        p0 = PasswordResetTokenGenerator()
-False
+        tk1 = p0.make_token(user)
+        self.assertTrue(p0.check_token(user, tk1))
 
-"""
+    def test_10265(self):
+        """
+        Ensure that the token generated for a user created in the same request
+        will work correctly.
+        """
+        # See ticket #10265
+        user = User.objects.create_user('comebackkid', 'test3@example.com', 'testpw')
+        p0 = PasswordResetTokenGenerator()
+        tk1 = p0.make_token(user)
+        reload = User.objects.get(username='comebackkid')
+        tk2 = p0.make_token(reload)
+        self.assertEqual(tk1, tk2)
+
+    def test_timeout(self):
+        """
+        Ensure we can use the token after n days, but no greater.
+        """
+        # Uses a mocked version of PasswordResetTokenGenerator so we can change
+        # the value of 'today'
+        class Mocked(PasswordResetTokenGenerator):
+            def __init__(self, today):
+                self._today_val = today
+            def _today(self):
+                return self._today_val
+
+        user = User.objects.create_user('tokentestuser', 'test2@example.com', 'testpw')
+        p0 = PasswordResetTokenGenerator()
+        tk1 = p0.make_token(user)
+        p1 = Mocked(date.today() + timedelta(settings.PASSWORD_RESET_TIMEOUT_DAYS))
+        self.assertTrue(p1.check_token(user, tk1))
+
+        p2 = Mocked(date.today() + timedelta(settings.PASSWORD_RESET_TIMEOUT_DAYS + 1))
+        self.assertFalse(p2.check_token(user, tk1))