libs/openid/test/test_nonce.py

   1 from openid.test import datadriven
   2 import time
   3 import unittest
   4 import re
   5
   6 from openid.store.nonce import \
   7      mkNonce, \
   8      split as splitNonce, \
   9      checkTimestamp
  10
  11 nonce_re = re.compile(r'\A\d{4}-\d\d-\d\dT\d\d:\d\d:\d\dZ')
  12
  13 class NonceTest(unittest.TestCase):
  14     def test_mkNonce(self):
  15         nonce = mkNonce()
  16         self.failUnless(nonce_re.match(nonce))
  17         self.failUnless(len(nonce) == 26)
  18
  19     def test_mkNonce_when(self):
  20         nonce = mkNonce(0)
  21         self.failUnless(nonce_re.match(nonce))
  22         self.failUnless(nonce.startswith('1970-01-01T00:00:00Z'))
  23         self.failUnless(len(nonce) == 26)
  24
  25     def test_splitNonce(self):
  26         s = '1970-01-01T00:00:00Z'
  27         expected_t = 0
  28         expected_salt = ''
  29         actual_t, actual_salt = splitNonce(s)
  30         self.failUnlessEqual(expected_t, actual_t)
  31         self.failUnlessEqual(expected_salt, actual_salt)
  32
  33     def test_mkSplit(self):
  34         t = 42
  35         nonce_str = mkNonce(t)
  36         self.failUnless(nonce_re.match(nonce_str))
  37         et, salt = splitNonce(nonce_str)
  38         self.failUnlessEqual(len(salt), 6)
  39         self.failUnlessEqual(et, t)
  40
  41 class BadSplitTest(datadriven.DataDrivenTestCase):
  42     cases = [
  43         '',
  44         '1970-01-01T00:00:00+1:00',
  45         '1969-01-01T00:00:00Z',
  46         '1970-00-01T00:00:00Z',
  47         '1970.01-01T00:00:00Z',
  48         'Thu Sep  7 13:29:31 PDT 2006',
  49         'monkeys',
  50         ]
  51
  52     def __init__(self, nonce_str):
  53         datadriven.DataDrivenTestCase.__init__(self, nonce_str)
  54         self.nonce_str = nonce_str
  55
  56     def runOneTest(self):
  57         self.failUnlessRaises(ValueError, splitNonce, self.nonce_str)
  58
  59 class CheckTimestampTest(datadriven.DataDrivenTestCase):
  60     cases = [
  61         # exact, no allowed skew
  62         ('1970-01-01T00:00:00Z', 0, 0, True),
  63
  64         # exact, large skew
  65         ('1970-01-01T00:00:00Z', 1000, 0, True),
  66
  67         # no allowed skew, one second old
  68         ('1970-01-01T00:00:00Z', 0, 1, False),
  69
  70         # many seconds old, outside of skew
  71         ('1970-01-01T00:00:00Z', 10, 50, False),
  72
  73         # one second old, one second skew allowed
  74         ('1970-01-01T00:00:00Z', 1, 1, True),
  75
  76         # One second in the future, one second skew allowed
  77         ('1970-01-01T00:00:02Z', 1, 1, True),
  78
  79         # two seconds in the future, one second skew allowed
  80         ('1970-01-01T00:00:02Z', 1, 0, False),
  81
  82         # malformed nonce string
  83         ('monkeys', 0, 0, False),
  84         ]
  85
  86     def __init__(self, nonce_string, allowed_skew, now, expected):
  87         datadriven.DataDrivenTestCase.__init__(
  88             self, repr((nonce_string, allowed_skew, now)))
  89         self.nonce_string = nonce_string
  90         self.allowed_skew = allowed_skew
  91         self.now = now
  92         self.expected = expected
  93
  94     def runOneTest(self):
  95         actual = checkTimestamp(self.nonce_string, self.allowed_skew, self.now)
  96         self.failUnlessEqual(bool(self.expected), bool(actual))
  97
  98 def pyUnitTests():
  99     return datadriven.loadTests(__name__)
 100
 101 if __name__ == '__main__':
 102     suite = pyUnitTests()
 103     runner = unittest.TextTestRunner()
 104     runner.run(suite)