tests/testdigest.py

   1 from ctypescrypto.oid import Oid
   2 from ctypescrypto import digest
   3 from base64 import b16decode,b16encode
   4 import unittest
   5
   6 class TestDigestType(unittest.TestCase):
   7         def test_md4(self):
   8                 d=digest.DigestType("md4")
   9                 self.assertEqual(d.digest_size(),16)
  10                 self.assertEqual(d.block_size(),64)
  11                 self.assertEqual(d.oid(),Oid("md4"))
  12         def test_md5(self):
  13                 d=digest.DigestType("md5")
  14                 self.assertEqual(d.digest_size(),16)
  15                 self.assertEqual(d.block_size(),64)
  16                 self.assertEqual(d.oid(),Oid("md5"))
  17         def test_sha1(self):
  18                 d=digest.DigestType("sha1")
  19                 self.assertEqual(d.digest_size(),20)
  20                 self.assertEqual(d.block_size(),64)
  21                 self.assertEqual(d.oid(),Oid("sha1"))
  22         def test_sha256(self):
  23                 d=digest.DigestType("sha256")
  24                 self.assertEqual(d.digest_size(),32)
  25                 self.assertEqual(d.block_size(),64)
  26                 self.assertEqual(d.oid(),Oid("sha256"))
  27         def test_sha384(self):
  28                 d=digest.DigestType("sha384")
  29                 self.assertEqual(d.digest_size(),48)
  30                 self.assertEqual(d.block_size(),128)
  31                 self.assertEqual(d.oid(),Oid("sha384"))
  32         def test_sha512(self):
  33                 d=digest.DigestType("sha512")
  34                 self.assertEqual(d.digest_size(),64)
  35                 self.assertEqual(d.block_size(),128)
  36                 self.assertEqual(d.oid(),Oid("sha512"))
  37
  38
  39 class TestIface(unittest.TestCase):
  40         """ Test all methods with one algorithms """
  41         def test_cons(self):
  42                 md=digest.DigestType("sha1")
  43                 dgst=digest.Digest(md)
  44                 dgst.update("A quick brown fox jumps over the lazy dog.")
  45                 self.assertEqual(dgst.digest_size,20)
  46                 self.assertEqual(dgst.hexdigest(),"00CFFE7312BF9CA73584F24BDF7DF1D028340397")
  47         def test_bindigest(self):
  48                 dgst=digest.new("sha1")
  49                 dgst.update("A quick brown fox jumps over the lazy dog.")
  50                 self.assertEqual(dgst.digest_size,20)
  51                 self.assertEqual(dgst.digest(),b16decode("00CFFE7312BF9CA73584F24BDF7DF1D028340397",True))
  52         def test_duplicatedigest(self):
  53                 dgst=digest.new("sha1")
  54                 dgst.update("A quick brown fox jumps over the lazy dog.")
  55                 v1=dgst.digest()
  56                 v2=dgst.digest()
  57                 self.assertEqual(v1,v2)
  58         def test_copy(self):
  59                 dgst=digest.new("sha1")
  60                 dgst.update("A quick brown fox jumps over ")
  61                 d2=dgst.copy()
  62                 dgst.update("the lazy dog.")
  63                 value1=dgst.hexdigest()
  64                 d2.update("the fat pig.")
  65                 value2=d2.hexdigest()
  66                 self.assertEqual(value1,"00CFFE7312BF9CA73584F24BDF7DF1D028340397")
  67                 self.assertEqual(value2,"5328F33739BEC2A15B6A30F17D3BC13CC11A7C78")
  68 class TestAlgo(unittest.TestCase):
  69         """ Test all statdard algorithms """
  70         def test_md5(self):
  71                 d=digest.new("md5")
  72                 self.assertEqual(d.digest_size,16)
  73                 d.update("A quick brown fox jumps over the lazy dog.")
  74                 self.assertEqual(d.hexdigest(),"DF756A3769FCAB0A261880957590C768")
  75
  76         def test_md4(self):
  77                 d=digest.new("md4")
  78                 d.update("A quick brown fox jumps over the lazy dog.")
  79                 self.assertEqual(d.digest_size,16)
  80                 self.assertEqual(d.hexdigest(),"FAAED595A3E38BBF0D9B4B98021D200F")
  81         def test_sha256(self):
  82                 d=digest.new("sha256")
  83                 d.update("A quick brown fox jumps over the lazy dog.")
  84                 self.assertEqual(d.digest_size,32)
  85                 self.assertEqual(d.hexdigest(),"FFCA2587CFD4846E4CB975B503C9EB940F94566AA394E8BD571458B9DA5097D5")
  86         def test_sha384(self):
  87                 d=digest.new("sha384")
  88                 d.update("A quick brown fox jumps over the lazy dog.")
  89                 self.assertEqual(d.digest_size,48)
  90                 self.assertEqual(d.hexdigest(),"C7D71B1BA81D0DD028E79C7E75CF2F83169C14BA732CA5A2AD731151584E9DE843C1A314077D62B96B03367F72E126D8")
  91         def test_sha512(self):
  92                 d=digest.new("sha512")
  93                 self.assertEqual(d.digest_size,64)
  94                 d.update("A quick brown fox jumps over the lazy dog.")
  95                 self.assertEqual(d.hexdigest(),"3045575CF3B873DD656F5F3426E04A4ACD11950BB2538772EE14867002B408E21FF18EF7F7B2CAB484A3C1C0BE3F8ACC4AED536A427353C7748DC365FC1A8646")
  96
  97 if __name__ == "__main__":
  98         unittest.main()