diff --git a/Lib/test/test_hmac.py b/Lib/test/test_hmac.py index f9cd73e6420..f36db4ea636 100644 --- a/Lib/test/test_hmac.py +++ b/Lib/test/test_hmac.py @@ -3,8 +3,9 @@ import unittest from test import test_support class TestVectorsTestCase(unittest.TestCase): + def test_vectors(self): - """Test the HMAC module against test vectors from the RFC.""" + # Test the HMAC module against test vectors from the RFC. def md5test(key, data, digest): h = hmac.HMAC(key, data) @@ -23,8 +24,9 @@ class TestVectorsTestCase(unittest.TestCase): "56be34521d144c88dbb8c733f0e8b3f6") class ConstructorTestCase(unittest.TestCase): + def test_normal(self): - """Standard constructor call.""" + # Standard constructor call. failed = 0 try: h = hmac.HMAC("key") @@ -32,14 +34,14 @@ class ConstructorTestCase(unittest.TestCase): self.fail("Standard constructor call raised exception.") def test_withtext(self): - """Constructor call with text.""" + # Constructor call with text. try: h = hmac.HMAC("key", "hash this!") except: self.fail("Constructor call with text argument raised exception.") def test_withmodule(self): - """Constructor call with text and digest module.""" + # Constructor call with text and digest module. import sha try: h = hmac.HMAC("key", "", sha) @@ -47,14 +49,15 @@ class ConstructorTestCase(unittest.TestCase): self.fail("Constructor call with sha module raised exception.") class SanityTestCase(unittest.TestCase): + def test_default_is_md5(self): - """Testing if HMAC defaults to MD5 algorithm.""" + # Testing if HMAC defaults to MD5 algorithm. import md5 h = hmac.HMAC("key") self.failUnless(h.digestmod == md5) def test_exercise_all_methods(self): - """Exercising all methods once.""" + # Exercising all methods once. # This must not raise any exceptions try: h = hmac.HMAC("my secret key") @@ -66,8 +69,9 @@ class SanityTestCase(unittest.TestCase): self.fail("Exception raised during normal usage of HMAC class.") class CopyTestCase(unittest.TestCase): + def test_attributes(self): - """Testing if attributes are of same type.""" + # Testing if attributes are of same type. h1 = hmac.HMAC("key") h2 = h1.copy() self.failUnless(h1.digestmod == h2.digestmod, @@ -78,7 +82,7 @@ class CopyTestCase(unittest.TestCase): "Types of outer don't match.") def test_realcopy(self): - """Testing if the copy method created a real copy.""" + # Testing if the copy method created a real copy. h1 = hmac.HMAC("key") h2 = h1.copy() # Using id() in case somebody has overridden __cmp__. @@ -89,7 +93,7 @@ class CopyTestCase(unittest.TestCase): "No real copy of the attribute 'outer'.") def test_equality(self): - """Testing if the copy has the same digests.""" + # Testing if the copy has the same digests. h1 = hmac.HMAC("key") h1.update("some random text") h2 = h1.copy() @@ -99,10 +103,12 @@ class CopyTestCase(unittest.TestCase): "Hexdigest of copy doesn't match original hexdigest.") def test_main(): - test_support.run_unittest(TestVectorsTestCase) - test_support.run_unittest(ConstructorTestCase) - test_support.run_unittest(SanityTestCase) - test_support.run_unittest(CopyTestCase) + suite = unittest.TestSuite() + suite.addTest(unittest.makeSuite(TestVectorsTestCase)) + suite.addTest(unittest.makeSuite(ConstructorTestCase)) + suite.addTest(unittest.makeSuite(SanityTestCase)) + suite.addTest(unittest.makeSuite(CopyTestCase)) + test_support.run_suite(suite) if __name__ == "__main__": test_main()