libs/openid/test/oidutil.py

   1 import unittest
   2 import codecs
   3 import string
   4 import random
   5 from openid import oidutil
   6
   7 def test_base64():
   8     allowed_s = string.ascii_letters + string.digits + '+/='
   9     allowed_d = {}
  10     for c in allowed_s:
  11         allowed_d[c] = None
  12     isAllowed = allowed_d.has_key
  13
  14     def checkEncoded(s):
  15         for c in s:
  16             assert isAllowed(c), s
  17
  18     cases = [
  19         '',
  20         'x',
  21         '\x00',
  22         '\x01',
  23         '\x00' * 100,
  24         ''.join(map(chr, range(256))),
  25         ]
  26
  27     for s in cases:
  28         b64 = oidutil.toBase64(s)
  29         checkEncoded(b64)
  30         s_prime = oidutil.fromBase64(b64)
  31         assert s_prime == s, (s, b64, s_prime)
  32
  33     # Randomized test
  34     for _ in xrange(50):
  35         n = random.randrange(2048)
  36         s = ''.join(map(chr, map(lambda _: random.randrange(256), range(n))))
  37         b64 = oidutil.toBase64(s)
  38         checkEncoded(b64)
  39         s_prime = oidutil.fromBase64(b64)
  40         assert s_prime == s, (s, b64, s_prime)
  41
  42 class AppendArgsTest(unittest.TestCase):
  43     def __init__(self, desc, args, expected):
  44         unittest.TestCase.__init__(self)
  45         self.desc = desc
  46         self.args = args
  47         self.expected = expected
  48
  49     def runTest(self):
  50         result = oidutil.appendArgs(*self.args)
  51         self.assertEqual(self.expected, result, self.args)
  52
  53     def shortDescription(self):
  54         return self.desc
  55
  56
  57
  58 class TestSymbol(unittest.TestCase):
  59     def testCopyHash(self):
  60         import copy
  61         s = oidutil.Symbol("Foo")
  62         d = {s: 1}
  63         d_prime = copy.deepcopy(d)
  64         self.failUnless(s in d_prime, "%r isn't in %r" % (s, d_prime))
  65
  66         t = oidutil.Symbol("Bar")
  67         self.failIfEqual(hash(s), hash(t))
  68
  69
  70 def buildAppendTests():
  71     simple = 'http://www.example.com/'
  72     cases = [
  73         ('empty list',
  74          (simple, []),
  75          simple),
  76
  77         ('empty dict',
  78          (simple, {}),
  79          simple),
  80
  81         ('one list',
  82          (simple, [('a', 'b')]),
  83          simple + '?a=b'),
  84
  85         ('one dict',
  86          (simple, {'a':'b'}),
  87          simple + '?a=b'),
  88
  89         ('two list (same)',
  90          (simple, [('a', 'b'), ('a', 'c')]),
  91          simple + '?a=b&a=c'),
  92
  93         ('two list',
  94          (simple, [('a', 'b'), ('b', 'c')]),
  95          simple + '?a=b&b=c'),
  96
  97         ('two list (order)',
  98          (simple, [('b', 'c'), ('a', 'b')]),
  99          simple + '?b=c&a=b'),
 100
 101         ('two dict (order)',
 102          (simple, {'b':'c', 'a':'b'}),
 103          simple + '?a=b&b=c'),
 104
 105         ('escape',
 106          (simple, [('=', '=')]),
 107          simple + '?%3D=%3D'),
 108
 109         ('escape (URL)',
 110          (simple, [('this_url', simple)]),
 111          simple + '?this_url=http%3A%2F%2Fwww.example.com%2F'),
 112
 113         ('use dots',
 114          (simple, [('openid.stuff', 'bother')]),
 115          simple + '?openid.stuff=bother'),
 116
 117         ('args exist (empty)',
 118          (simple + '?stuff=bother', []),
 119          simple + '?stuff=bother'),
 120
 121         ('args exist',
 122          (simple + '?stuff=bother', [('ack', 'ack')]),
 123          simple + '?stuff=bother&ack=ack'),
 124
 125         ('args exist',
 126          (simple + '?stuff=bother', [('ack', 'ack')]),
 127          simple + '?stuff=bother&ack=ack'),
 128
 129         ('args exist (dict)',
 130          (simple + '?stuff=bother', {'ack': 'ack'}),
 131          simple + '?stuff=bother&ack=ack'),
 132
 133         ('args exist (dict 2)',
 134          (simple + '?stuff=bother', {'ack': 'ack', 'zebra':'lion'}),
 135          simple + '?stuff=bother&ack=ack&zebra=lion'),
 136
 137         ('three args (dict)',
 138          (simple, {'stuff': 'bother', 'ack': 'ack', 'zebra':'lion'}),
 139          simple + '?ack=ack&stuff=bother&zebra=lion'),
 140
 141         ('three args (list)',
 142          (simple, [('stuff', 'bother'), ('ack', 'ack'), ('zebra', 'lion')]),
 143          simple + '?stuff=bother&ack=ack&zebra=lion'),
 144         ]
 145
 146     tests = []
 147
 148     for name, args, expected in cases:
 149         test = AppendArgsTest(name, args, expected)
 150         tests.append(test)
 151
 152     return unittest.TestSuite(tests)
 153
 154 def pyUnitTests():
 155     some = buildAppendTests()
 156     some.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(TestSymbol))
 157     return some
 158
 159 def test_appendArgs():
 160     suite = buildAppendTests()
 161     suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(TestSymbol))
 162     runner = unittest.TextTestRunner()
 163     result = runner.run(suite)
 164     assert result.wasSuccessful()
 165
 166 # XXX: there are more functions that could benefit from being better
 167 # specified and tested in oidutil.py These include, but are not
 168 # limited to appendArgs
 169
 170 def test(skipPyUnit=True):
 171     test_base64()
 172     if not skipPyUnit:
 173         test_appendArgs()
 174
 175 if __name__ == '__main__':
 176     test(skipPyUnit=False)