59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
|
# Testing md5 module
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from md5 import md5
|
||
|
|
from test import test_support
|
||
|
|
|
||
|
|
def hexstr(s):
|
||
|
|
import string
|
||
|
|
h = string.hexdigits
|
||
|
|
r = ''
|
||
|
|
for c in s:
|
||
|
|
i = ord(c)
|
||
|
|
r = r + h[(i >> 4) & 0xF] + h[i & 0xF]
|
||
|
|
return r
|
||
|
|
|
||
|
|
class MD5_Test(unittest.TestCase):
|
||
|
|
|
||
|
|
def md5test(self, s, expected):
|
||
|
|
self.assertEqual(hexstr(md5(s).digest()), expected)
|
||
|
|
self.assertEqual(md5(s).hexdigest(), expected)
|
||
|
|
|
||
|
|
def test_basics(self):
|
||
|
|
eq = self.md5test
|
||
|
|
eq('', 'd41d8cd98f00b204e9800998ecf8427e')
|
||
|
|
eq('a', '0cc175b9c0f1b6a831c399e269772661')
|
||
|
|
eq('abc', '900150983cd24fb0d6963f7d28e17f72')
|
||
|
|
eq('message digest', 'f96b697d7cb7938d525a2f31aaf161d0')
|
||
|
|
eq('abcdefghijklmnopqrstuvwxyz', 'c3fcd3d76192e4007dfb496cca67e13b')
|
||
|
|
eq('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789',
|
||
|
|
'd174ab98d277d9f5a5611c2c9f419d9f')
|
||
|
|
eq('12345678901234567890123456789012345678901234567890123456789012345678901234567890',
|
||
|
|
'57edf4a22be3c955ac49da2e2107b67a')
|
||
|
|
|
||
|
|
def test_hexdigest(self):
|
||
|
|
# hexdigest is new with Python 2.0
|
||
|
|
m = md5('testing the hexdigest method')
|
||
|
|
h = m.hexdigest()
|
||
|
|
self.assertEqual(hexstr(m.digest()), h)
|
||
|
|
|
||
|
|
def test_large_update(self):
|
||
|
|
aas = 'a' * 64
|
||
|
|
bees = 'b' * 64
|
||
|
|
cees = 'c' * 64
|
||
|
|
|
||
|
|
m1 = md5()
|
||
|
|
m1.update(aas)
|
||
|
|
m1.update(bees)
|
||
|
|
m1.update(cees)
|
||
|
|
|
||
|
|
m2 = md5()
|
||
|
|
m2.update(aas + bees + cees)
|
||
|
|
self.assertEqual(m1.digest(), m2.digest())
|
||
|
|
|
||
|
|
def test_main():
|
||
|
|
test_support.run_unittest(MD5_Test)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
test_main()
|