1 # Testing md5 module

     2 import warnings

     3 warnings.filterwarnings("ignore", "the md5 module is deprecated.*",

     4                         DeprecationWarning)

     5 

     6 import unittest

     7 from md5 import md5

     8 from test import test_support

     9 

    10 def hexstr(s):

    11     import string

    12     h = string.hexdigits

    13     r = ''

    14     for c in s:

    15         i = ord(c)

    16         r = r + h[(i >> 4) & 0xF] + h[i & 0xF]

    17     return r

    18 

    19 class MD5_Test(unittest.TestCase):

    20 

    21     def md5test(self, s, expected):

    22         self.assertEqual(hexstr(md5(s).digest()), expected)

    23         self.assertEqual(md5(s).hexdigest(), expected)

    24 

    25     def test_basics(self):

    26         eq = self.md5test

    27         eq('', 'd41d8cd98f00b204e9800998ecf8427e')

    28         eq('a', '0cc175b9c0f1b6a831c399e269772661')

    29         eq('abc', '900150983cd24fb0d6963f7d28e17f72')

    30         eq('message digest', 'f96b697d7cb7938d525a2f31aaf161d0')

    31         eq('abcdefghijklmnopqrstuvwxyz', 'c3fcd3d76192e4007dfb496cca67e13b')

    32         eq('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789',

    33            'd174ab98d277d9f5a5611c2c9f419d9f')

    34         eq('12345678901234567890123456789012345678901234567890123456789012345678901234567890',

    35            '57edf4a22be3c955ac49da2e2107b67a')

    36 

    37     def test_hexdigest(self):

    38         # hexdigest is new with Python 2.0

    39         m = md5('testing the hexdigest method')

    40         h = m.hexdigest()

    41         self.assertEqual(hexstr(m.digest()), h)

    42 

    43     def test_large_update(self):

    44         aas = 'a' * 64

    45         bees = 'b' * 64

    46         cees = 'c' * 64

    47 

    48         m1 = md5()

    49         m1.update(aas)

    50         m1.update(bees)

    51         m1.update(cees)

    52 

    53         m2 = md5()

    54         m2.update(aas + bees + cees)

    55         self.assertEqual(m1.digest(), m2.digest())

    56 

    57 def test_main():

    58     test_support.run_unittest(MD5_Test)

    59 

    60 if __name__ == '__main__':

    61     test_main()