1# Testing sha module (NIST's Secure Hash Algorithm) 2 3# use the three examples from Federal Information Processing Standards 4# Publication 180-1, Secure Hash Standard, 1995 April 17 5# http://www.itl.nist.gov/div897/pubs/fip180-1.htm 6 7import warnings 8warnings.filterwarnings("ignore", "the sha module is deprecated.*", 9 DeprecationWarning) 10 11import sha 12import unittest 13from test import test_support 14 15 16class SHATestCase(unittest.TestCase): 17 def check(self, data, digest): 18 # Check digest matches the expected value 19 obj = sha.new(data) 20 computed = obj.hexdigest() 21 self.assertTrue(computed == digest) 22 23 # Verify that the value doesn't change between two consecutive 24 # digest operations. 25 computed_again = obj.hexdigest() 26 self.assertTrue(computed == computed_again) 27 28 # Check hexdigest() output matches digest()'s output 29 digest = obj.digest() 30 hexd = "" 31 for c in digest: 32 hexd += '%02x' % ord(c) 33 self.assertTrue(computed == hexd) 34 35 def test_case_1(self): 36 self.check("abc", 37 "a9993e364706816aba3e25717850c26c9cd0d89d") 38 39 def test_case_2(self): 40 self.check("abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 41 "84983e441c3bd26ebaae4aa1f95129e5e54670f1") 42 43 def test_case_3(self): 44 self.check("a" * 1000000, 45 "34aa973cd4c4daa4f61eeb2bdbad27316534016f") 46 47 def test_case_4(self): 48 self.check(chr(0xAA) * 80, 49 '4ca0ef38f1794b28a8f8ee110ee79d48ce13be25') 50 51def test_main(): 52 test_support.run_unittest(SHATestCase) 53 54 55if __name__ == "__main__": 56 test_main() 57