home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2011 June / maximum-cd-2011-06.iso / DiscContents / LibO_3.3.1_Win_x86_install_multi.exe / libreoffice1.cab / test_pep247.py < prev    next >
Encoding:
Python Source  |  2011-02-15  |  2.0 KB  |  75 lines
  1. """
  2. Test suite to check compilance with PEP 247, the standard API
  3. for hashing algorithms
  4. """
  5.  
  6. import warnings
  7. warnings.filterwarnings('ignore', 'the md5 module is deprecated.*',
  8.                         DeprecationWarning)
  9. warnings.filterwarnings('ignore', 'the sha module is deprecated.*',
  10.                         DeprecationWarning)
  11.  
  12. import hmac
  13. import md5
  14. import sha
  15.  
  16. import unittest
  17. from test import test_support
  18.  
  19. class Pep247Test(unittest.TestCase):
  20.  
  21.     def check_module(self, module, key=None):
  22.         self.assert_(hasattr(module, 'digest_size'))
  23.         self.assert_(module.digest_size is None or module.digest_size > 0)
  24.  
  25.         if not key is None:
  26.             obj1 = module.new(key)
  27.             obj2 = module.new(key, 'string')
  28.  
  29.             h1 = module.new(key, 'string').digest()
  30.             obj3 = module.new(key)
  31.             obj3.update('string')
  32.             h2 = obj3.digest()
  33.         else:
  34.             obj1 = module.new()
  35.             obj2 = module.new('string')
  36.  
  37.             h1 = module.new('string').digest()
  38.             obj3 = module.new()
  39.             obj3.update('string')
  40.             h2 = obj3.digest()
  41.  
  42.         self.assertEquals(h1, h2)
  43.  
  44.         self.assert_(hasattr(obj1, 'digest_size'))
  45.  
  46.         if not module.digest_size is None:
  47.             self.assertEquals(obj1.digest_size, module.digest_size)
  48.  
  49.         self.assertEquals(obj1.digest_size, len(h1))
  50.         obj1.update('string')
  51.         obj_copy = obj1.copy()
  52.         self.assertEquals(obj1.digest(), obj_copy.digest())
  53.         self.assertEquals(obj1.hexdigest(), obj_copy.hexdigest())
  54.  
  55.         digest, hexdigest = obj1.digest(), obj1.hexdigest()
  56.         hd2 = ""
  57.         for byte in digest:
  58.             hd2 += '%02x' % ord(byte)
  59.         self.assertEquals(hd2, hexdigest)
  60.  
  61.     def test_md5(self):
  62.         self.check_module(md5)
  63.  
  64.     def test_sha(self):
  65.         self.check_module(sha)
  66.  
  67.     def test_hmac(self):
  68.         self.check_module(hmac, key='abc')
  69.  
  70. def test_main():
  71.     test_support.run_unittest(Pep247Test)
  72.  
  73. if __name__ == '__main__':
  74.     test_main()
  75.