Commit | Line | Data |
---|---|---|
920dae64 AT |
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') |