]> granicus.if.org Git - python/commitdiff
add a test for bad atexit arguments
authorBenjamin Peterson <benjamin@python.org>
Tue, 23 Sep 2008 03:14:49 +0000 (03:14 +0000)
committerBenjamin Peterson <benjamin@python.org>
Tue, 23 Sep 2008 03:14:49 +0000 (03:14 +0000)
Lib/test/test_atexit.py

index 6856042edc60a1040b9e7f29b53d9691fe79e65e..04af9fb143d0312ea17cf7705915ef03a45b3822 100644 (file)
@@ -44,6 +44,10 @@ class TestCase(unittest.TestCase):
         self.assertEqual(self.stream.getvalue(),
                             "h4 (4,) {'kw': 'abc'}\nh4 () {}\nh1\n")
 
+    def test_badargs(self):
+        atexit.register(lambda: 1, 0, 0, (x for x in (1,2)), 0, 0)
+        self.assertRaises(TypeError, atexit._run_exitfuncs)
+
     def test_order(self):
         # be sure handlers are executed in reverse order
         atexit.register(h1)