From: Benjamin Peterson <benjamin@python.org>
Date: Tue, 23 Sep 2008 20:43:09 +0000 (+0000)
Subject: backport the atexit test for r66563
X-Git-Tag: v2.6~46
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=a74d675facda18cca2a821be56d49446184b6ea1;p=thirdparty%2FPython%2Fcpython.git

backport the atexit test for r66563
---

diff --git a/Lib/test/test_atexit.py b/Lib/test/test_atexit.py
index bea482f9dd01..a60dfa85b6bd 100644
--- 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()