Fixed #16182: Increase timestamp precision on TimestampSigner. Thanks to Eric Florenzano.

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

1 files changed, 17 insertions, 17 deletions

diff --git a/tests/regressiontests/signing/tests.py b/tests/regressiontests/signing/tests.py
index 0b0cacf10e..bc16214600 100644
--- a/tests/regressiontests/signing/tests.py
+++ b/tests/regressiontests/signing/tests.py
@@ -1,5 +1,3 @@
-import time
-
 from django.core import signing
 from django.test import TestCase
 from django.utils.encoding import force_unicode
@@ -98,19 +96,21 @@ class TestTimestampSigner(TestCase):
 
     def test_timestamp_signer(self):
         value = u'hello'
-        _time = time.time
-        time.time = lambda: 123456789
-        try:
-            signer = signing.TimestampSigner('predictable-key')
-            ts = signer.sign(value)
-            self.assertNotEqual(ts,
-                signing.Signer('predictable-key').sign(value))
+        signer = signing.TimestampSigner('predictable-key',
+            time=lambda: 123456789)
+        ts = signer.sign(value)
+        self.assertNotEqual(ts,
+            signing.Signer('predictable-key').sign(value))
 
-            self.assertEqual(signer.unsign(ts), value)
-            time.time = lambda: 123456800
-            self.assertEqual(signer.unsign(ts, max_age=12), value)
-            self.assertEqual(signer.unsign(ts, max_age=11), value)
-            self.assertRaises(
-                signing.SignatureExpired, signer.unsign, ts, max_age=10)
-        finally:
-            time.time = _time
+        self.assertEqual(signer.unsign(ts), value)
+        signer = signing.TimestampSigner('predictable-key',
+            time=lambda: 123456800)
+        self.assertEqual(signer.unsign(ts, max_age=12), value)
+        self.assertEqual(signer.unsign(ts, max_age=11), value)
+        self.assertRaises(
+            signing.SignatureExpired, signer.unsign, ts, max_age=10)
+    
+    def test_timestamp_precision(self):
+        one = signing.TimestampSigner('key', time=lambda: 123.4567).sign('v')
+        two = signing.TimestampSigner('key', time=lambda: 123.4568).sign('v')
+        self.assertNotEqual(one, two)