[Python-checkins] r66569 - python/trunk/Lib/test/test_atexit.py

[benjamin.peterson]

Author: benjamin.peterson
Date: Tue Sep 23 22:43:09 2008
New Revision: 66569

Log:
backport the atexit test for r66563

Modified:
   python/trunk/Lib/test/test_atexit.py

Modified: python/trunk/Lib/test/test_atexit.py
==============================================================================
--- python/trunk/Lib/test/test_atexit.py	(original)
+++ python/trunk/Lib/test/test_atexit.py	Tue Sep 23 22:43:09 2008
@@ -22,6 +22,19 @@
             atexit._exithandlers = save_handlers
         self.assertEqual(s.getvalue(), "h4 (4,) {'kw': 'abc'}\nh4 () {}\nh1\n")
 
+    def test_badargs(self):
+        s = StringIO.StringIO()
+        sys.stdout = sys.stderr = s
+        save_handlers = atexit._exithandlers
+        atexit._exithandlers = []
+        try:
+            atexit.register(lambda: 1, 0, 0, (x for x in (1,2)), 0, 0)
+            self.assertRaises(TypeError, atexit._run_exitfuncs)
+        finally:
+            sys.stdout = sys.__stdout__
+            sys.stderr = sys.__stderr__
+            atexit._exithandlers = save_handlers
+
     def test_order(self):
         # be sure handlers are executed in reverse order
         s = StringIO.StringIO()