backport the atexit test for r66563

diff --git a/Lib/test/test_atexit.py b/Lib/test/test_atexit.py
index bea482f9dd0131e698c3539c17c29af370a88ad6..a60dfa85b6bd953b3c5ffafeabb4e8752d02b8f3 100644 (file)
--- a/Lib/test/test_atexit.py
+++ b/Lib/test/test_atexit.py
@@ -22,6 +22,19 @@ class TestCase(unittest.TestCase):
             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()