twisted/python/test/test_hashlib.py

   1 # Copyright (c) Twisted Matrix Laboratories.
   2 # See LICENSE for details.
   3
   4 """
   5 Tests for L{twisted.python.hashlib}
   6 """
   7
   8 from twisted.trial.unittest import TestCase
   9
  10 from twisted.python.hashlib import md5, sha1
  11
  12
  13 class HashObjectTests(TestCase):
  14     """
  15     Tests for the hash object APIs presented by L{hashlib}, C{md5} and C{sha1}.
  16     """
  17     def test_md5(self):
  18         """
  19         L{hashlib.md5} returns an object which can be used to compute an MD5
  20         hash as defined by U{RFC 1321<http://www.ietf.org/rfc/rfc1321.txt>}.
  21         """
  22         # Test the result using values from section A.5 of the RFC.
  23         self.assertEqual(
  24             md5().hexdigest(), "d41d8cd98f00b204e9800998ecf8427e")
  25         self.assertEqual(
  26             md5("a").hexdigest(), "0cc175b9c0f1b6a831c399e269772661")
  27         self.assertEqual(
  28             md5("abc").hexdigest(), "900150983cd24fb0d6963f7d28e17f72")
  29         self.assertEqual(
  30             md5("message digest").hexdigest(),
  31             "f96b697d7cb7938d525a2f31aaf161d0")
  32         self.assertEqual(
  33             md5("abcdefghijklmnopqrstuvwxyz").hexdigest(),
  34             "c3fcd3d76192e4007dfb496cca67e13b")
  35         self.assertEqual(
  36             md5("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
  37                 "0123456789").hexdigest(),
  38             "d174ab98d277d9f5a5611c2c9f419d9f")
  39         self.assertEqual(
  40             md5("1234567890123456789012345678901234567890123456789012345678901"
  41                 "2345678901234567890").hexdigest(),
  42             "57edf4a22be3c955ac49da2e2107b67a")
  43
  44         # It should have digest and update methods, too.
  45         self.assertEqual(
  46             md5().digest().encode('hex'),
  47             "d41d8cd98f00b204e9800998ecf8427e")
  48         hash = md5()
  49         hash.update("a")
  50         self.assertEqual(
  51             hash.digest().encode('hex'),
  52             "0cc175b9c0f1b6a831c399e269772661")
  53
  54         # Instances of it should have a digest_size attribute
  55         self.assertEqual(md5().digest_size, 16)
  56
  57
  58     def test_sha1(self):
  59         """
  60         L{hashlib.sha1} returns an object which can be used to compute a SHA1
  61         hash as defined by U{RFC 3174<http://tools.ietf.org/rfc/rfc3174.txt>}.
  62         """
  63         def format(s):
  64             return ''.join(s.split()).lower()
  65         # Test the result using values from section 7.3 of the RFC.
  66         self.assertEqual(
  67             sha1("abc").hexdigest(),
  68             format(
  69                 "A9 99 3E 36 47 06 81 6A BA 3E 25 71 78 50 C2 6C 9C D0 D8 9D"))
  70         self.assertEqual(
  71             sha1("abcdbcdecdefdefgefghfghighijhi"
  72                  "jkijkljklmklmnlmnomnopnopq").hexdigest(),
  73             format(
  74                 "84 98 3E 44 1C 3B D2 6E BA AE 4A A1 F9 51 29 E5 E5 46 70 F1"))
  75
  76         # It should have digest and update methods, too.
  77         self.assertEqual(
  78             sha1("abc").digest().encode('hex'),
  79             format(
  80                 "A9 99 3E 36 47 06 81 6A BA 3E 25 71 78 50 C2 6C 9C D0 D8 9D"))
  81         hash = sha1()
  82         hash.update("abc")
  83         self.assertEqual(
  84             hash.digest().encode('hex'),
  85             format(
  86                 "A9 99 3E 36 47 06 81 6A BA 3E 25 71 78 50 C2 6C 9C D0 D8 9D"))
  87
  88         # Instances of it should have a digest_size attribute.
  89         self.assertEqual(
  90             sha1().digest_size, 20)