home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / test_pep247.py < prev    next >
Text File  |  2003-12-30  |  2KB  |  51 lines
  1. #
  2. # Test suite to check compliance with PEP 247, the standard API for
  3. # hashing algorithms.
  4. #
  5.  
  6. import md5, sha, hmac
  7.  
  8. def check_hash_module(module, key=None):
  9.     assert hasattr(module, 'digest_size'), "Must have digest_size"
  10.     assert (module.digest_size is None or
  11.             module.digest_size > 0), "digest_size must be None or positive"
  12.  
  13.     if key is not None:
  14.         obj1 = module.new(key)
  15.         obj2 = module.new(key, "string")
  16.  
  17.         h1 = module.new(key, "string").digest()
  18.         obj3 = module.new(key) ; obj3.update("string") ; h2 = obj3.digest()
  19.         assert h1 == h2, "Hashes must match"
  20.  
  21.     else:
  22.         obj1 = module.new()
  23.         obj2 = module.new("string")
  24.  
  25.         h1 = module.new("string").digest()
  26.         obj3 = module.new() ; obj3.update("string") ; h2 = obj3.digest()
  27.         assert h1 == h2, "Hashes must match"
  28.  
  29.     assert hasattr(obj1, 'digest_size'), "Objects must have digest_size attr"
  30.     if module.digest_size is not None:
  31.         assert obj1.digest_size == module.digest_size, "digest_size must match"
  32.     assert obj1.digest_size == len(h1), "digest_size must match actual size"
  33.     obj1.update("string")
  34.     obj_copy = obj1.copy()
  35.     assert obj1.digest() == obj_copy.digest(), "Copied objects must match"
  36.     assert obj1.hexdigest() == obj_copy.hexdigest(), \
  37.            "Copied objects must match"
  38.     digest, hexdigest = obj1.digest(), obj1.hexdigest()
  39.     hd2 = ""
  40.     for byte in digest:
  41.         hd2 += "%02x" % ord(byte)
  42.     assert hd2 == hexdigest, "hexdigest doesn't appear correct"
  43.  
  44.     print 'Module', module.__name__, 'seems to comply with PEP 247'
  45.  
  46.  
  47. if __name__ == '__main__':
  48.     check_hash_module(md5)
  49.     check_hash_module(sha)
  50.     check_hash_module(hmac, key='abc')
  51.