From: Ezio Melotti <ezio.melotti@gmail.com>
Date: Fri, 11 Jan 2013 03:18:45 +0000 (+0200)
Subject: #16919: test_crypt now works with unittest test discovery. Patch by Zachary Ware.
X-Git-Tag: v3.3.1rc1~391
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=90bbbd11648e8d64013369a595b982788fa89819;p=python

#16919: test_crypt now works with unittest test discovery. Patch by Zachary Ware.
---

diff --git a/Lib/test/test_crypt.py b/Lib/test/test_crypt.py
index dc107d8970..cfb7341e20 100644
--- a/Lib/test/test_crypt.py
+++ b/Lib/test/test_crypt.py
@@ -1,7 +1,11 @@
 from test import support
 import unittest
 
-crypt = support.import_module('crypt')
+def setUpModule():
+    # this import will raise unittest.SkipTest if _crypt doesn't exist,
+    # so it has to be done in setUpModule for test discovery to work
+    global crypt
+    crypt = support.import_module('crypt')
 
 class CryptTestCase(unittest.TestCase):
 
@@ -29,8 +33,5 @@ class CryptTestCase(unittest.TestCase):
         self.assertTrue(len(crypt.methods) >= 1)
         self.assertEqual(crypt.METHOD_CRYPT, crypt.methods[-1])
 
-def test_main():
-    support.run_unittest(CryptTestCase)
-
 if __name__ == "__main__":
-    test_main()
+    unittest.main()
diff --git a/Misc/NEWS b/Misc/NEWS
index 1ed665a803..7360af4436 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -421,6 +421,9 @@ Library
 Tests
 -----
 
+- Issue #16919: test_crypt now works with unittest test discovery.
+  Patch by Zachary Ware.
+
 - Issue #16910: test_bytes, test_unicode, and test_userstring now work with
   unittest test discovery.  Patch by Zachary Ware.