blob: 7a2dd0c8f17fa86d0c6a0c366a142c4938cb66f3 [file] [log] [blame]
Guido van Rossum3bead091992-01-27 17:00:37 +00001# Python test set -- part 5, built-in exceptions
2
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00003import os
4import sys
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00005import unittest
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00006import pickle
Barry Warsaw8d109cb2008-05-08 04:26:35 +00007import weakref
Antoine Pitroua7622852011-09-01 21:37:43 +02008import errno
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00009
Benjamin Petersonae5f2f42010-03-07 17:10:51 +000010from test.support import (TESTFN, unlink, run_unittest, captured_output,
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +000011 gc_collect, cpython_only)
Guido van Rossum83b120d2001-08-23 03:23:03 +000012
Guido van Rossum3bead091992-01-27 17:00:37 +000013# XXX This is not really enough, each *operation* should be tested!
14
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000015class ExceptionTests(unittest.TestCase):
Barry Warsawb9c1d3d2001-08-13 23:07:00 +000016
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000017 def raise_catch(self, exc, excname):
18 try:
Collin Winter828f04a2007-08-31 00:04:24 +000019 raise exc("spam")
Guido van Rossumb940e112007-01-10 16:19:56 +000020 except exc as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000021 buf1 = str(err)
22 try:
23 raise exc("spam")
Guido van Rossumb940e112007-01-10 16:19:56 +000024 except exc as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000025 buf2 = str(err)
Ezio Melottib3aedd42010-11-20 19:04:17 +000026 self.assertEqual(buf1, buf2)
27 self.assertEqual(exc.__name__, excname)
Guido van Rossum3bead091992-01-27 17:00:37 +000028
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000029 def testRaising(self):
30 self.raise_catch(AttributeError, "AttributeError")
31 self.assertRaises(AttributeError, getattr, sys, "undefined_attribute")
Guido van Rossum3bead091992-01-27 17:00:37 +000032
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000033 self.raise_catch(EOFError, "EOFError")
34 fp = open(TESTFN, 'w')
35 fp.close()
36 fp = open(TESTFN, 'r')
37 savestdin = sys.stdin
38 try:
39 try:
40 import marshal
Antoine Pitrou4a90ef02012-03-03 02:35:32 +010041 marshal.loads(b'')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000042 except EOFError:
43 pass
44 finally:
45 sys.stdin = savestdin
46 fp.close()
47 unlink(TESTFN)
Guido van Rossum3bead091992-01-27 17:00:37 +000048
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000049 self.raise_catch(IOError, "IOError")
50 self.assertRaises(IOError, open, 'this file does not exist', 'r')
Guido van Rossum3bead091992-01-27 17:00:37 +000051
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000052 self.raise_catch(ImportError, "ImportError")
53 self.assertRaises(ImportError, __import__, "undefined_module")
Guido van Rossum3bead091992-01-27 17:00:37 +000054
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000055 self.raise_catch(IndexError, "IndexError")
56 x = []
57 self.assertRaises(IndexError, x.__getitem__, 10)
Guido van Rossum3bead091992-01-27 17:00:37 +000058
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000059 self.raise_catch(KeyError, "KeyError")
60 x = {}
61 self.assertRaises(KeyError, x.__getitem__, 'key')
Guido van Rossum3bead091992-01-27 17:00:37 +000062
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000063 self.raise_catch(KeyboardInterrupt, "KeyboardInterrupt")
Guido van Rossum3bead091992-01-27 17:00:37 +000064
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000065 self.raise_catch(MemoryError, "MemoryError")
Guido van Rossum3bead091992-01-27 17:00:37 +000066
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000067 self.raise_catch(NameError, "NameError")
68 try: x = undefined_variable
69 except NameError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000070
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000071 self.raise_catch(OverflowError, "OverflowError")
72 x = 1
73 for dummy in range(128):
74 x += x # this simply shouldn't blow up
Guido van Rossum3bead091992-01-27 17:00:37 +000075
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000076 self.raise_catch(RuntimeError, "RuntimeError")
Guido van Rossum3bead091992-01-27 17:00:37 +000077
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000078 self.raise_catch(SyntaxError, "SyntaxError")
Georg Brandl7cae87c2006-09-06 06:51:57 +000079 try: exec('/\n')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000080 except SyntaxError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000081
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000082 self.raise_catch(IndentationError, "IndentationError")
Fred Drake72e48bd2000-09-08 16:32:34 +000083
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000084 self.raise_catch(TabError, "TabError")
Georg Brandle1b5ac62008-06-04 13:06:58 +000085 try: compile("try:\n\t1/0\n \t1/0\nfinally:\n pass\n",
86 '<string>', 'exec')
87 except TabError: pass
88 else: self.fail("TabError not raised")
Fred Drake72e48bd2000-09-08 16:32:34 +000089
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000090 self.raise_catch(SystemError, "SystemError")
Fred Drake72e48bd2000-09-08 16:32:34 +000091
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000092 self.raise_catch(SystemExit, "SystemExit")
93 self.assertRaises(SystemExit, sys.exit, 0)
Fred Drake85f36392000-07-11 17:53:00 +000094
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000095 self.raise_catch(TypeError, "TypeError")
96 try: [] + ()
97 except TypeError: pass
Fred Drake85f36392000-07-11 17:53:00 +000098
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000099 self.raise_catch(ValueError, "ValueError")
Guido van Rossume63bae62007-07-17 00:34:25 +0000100 self.assertRaises(ValueError, chr, 17<<16)
Guido van Rossum3bead091992-01-27 17:00:37 +0000101
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000102 self.raise_catch(ZeroDivisionError, "ZeroDivisionError")
103 try: x = 1/0
104 except ZeroDivisionError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000105
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000106 self.raise_catch(Exception, "Exception")
107 try: x = 1/0
Guido van Rossumb940e112007-01-10 16:19:56 +0000108 except Exception as e: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000109
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000110 def testSyntaxErrorMessage(self):
111 # make sure the right exception message is raised for each of
112 # these code fragments
Guido van Rossum3bead091992-01-27 17:00:37 +0000113
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000114 def ckmsg(src, msg):
115 try:
116 compile(src, '<fragment>', 'exec')
Guido van Rossumb940e112007-01-10 16:19:56 +0000117 except SyntaxError as e:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000118 if e.msg != msg:
119 self.fail("expected %s, got %s" % (msg, e.msg))
120 else:
121 self.fail("failed to get expected SyntaxError")
Guido van Rossum3bead091992-01-27 17:00:37 +0000122
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000123 s = '''while 1:
124 try:
125 pass
126 finally:
127 continue'''
Barry Warsaw992cb8a2000-05-25 23:16:54 +0000128
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000129 if not sys.platform.startswith('java'):
130 ckmsg(s, "'continue' not supported inside 'finally' clause")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000131
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000132 s = '''if 1:
133 try:
134 continue
135 except:
136 pass'''
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000137
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000138 ckmsg(s, "'continue' not properly in loop")
139 ckmsg("continue\n", "'continue' not properly in loop")
Thomas Wouters303de6a2006-04-20 22:42:37 +0000140
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000141 @cpython_only
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000142 def testSettingException(self):
143 # test that setting an exception at the C level works even if the
144 # exception object can't be constructed.
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000145
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000146 class BadException(Exception):
147 def __init__(self_):
Collin Winter828f04a2007-08-31 00:04:24 +0000148 raise RuntimeError("can't instantiate BadException")
Finn Bockaa3dc452001-12-08 10:15:48 +0000149
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000150 class InvalidException:
151 pass
Thomas Wouters303de6a2006-04-20 22:42:37 +0000152
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000153 def test_capi1():
154 import _testcapi
155 try:
156 _testcapi.raise_exception(BadException, 1)
Guido van Rossumb940e112007-01-10 16:19:56 +0000157 except TypeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000158 exc, err, tb = sys.exc_info()
159 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000160 self.assertEqual(co.co_name, "test_capi1")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000161 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000162 else:
163 self.fail("Expected exception")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000164
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000165 def test_capi2():
166 import _testcapi
167 try:
168 _testcapi.raise_exception(BadException, 0)
Guido van Rossumb940e112007-01-10 16:19:56 +0000169 except RuntimeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000170 exc, err, tb = sys.exc_info()
171 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000172 self.assertEqual(co.co_name, "__init__")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000173 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000174 co2 = tb.tb_frame.f_back.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000175 self.assertEqual(co2.co_name, "test_capi2")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000176 else:
177 self.fail("Expected exception")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000178
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000179 def test_capi3():
180 import _testcapi
181 self.assertRaises(SystemError, _testcapi.raise_exception,
182 InvalidException, 1)
183
184 if not sys.platform.startswith('java'):
185 test_capi1()
186 test_capi2()
187 test_capi3()
188
Thomas Wouters89f507f2006-12-13 04:49:30 +0000189 def test_WindowsError(self):
190 try:
191 WindowsError
192 except NameError:
193 pass
194 else:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000195 self.assertEqual(str(WindowsError(1001)), "1001")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000196 self.assertEqual(str(WindowsError(1001, "message")),
Ezio Melottib3aedd42010-11-20 19:04:17 +0000197 "[Error 1001] message")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000198 self.assertEqual(WindowsError(1001, "message").errno, 22)
199 self.assertEqual(WindowsError(1001, "message").winerror, 1001)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000200
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000201 def testAttributes(self):
202 # test that exception attributes are happy
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000203
204 exceptionList = [
Guido van Rossumebe3e162007-05-17 18:20:34 +0000205 (BaseException, (), {'args' : ()}),
206 (BaseException, (1, ), {'args' : (1,)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000207 (BaseException, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000208 {'args' : ('foo',)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000209 (BaseException, ('foo', 1),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000210 {'args' : ('foo', 1)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000211 (SystemExit, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000212 {'args' : ('foo',), 'code' : 'foo'}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000213 (IOError, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000214 {'args' : ('foo',), 'filename' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000215 'errno' : None, 'strerror' : None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000216 (IOError, ('foo', 'bar'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000217 {'args' : ('foo', 'bar'), 'filename' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000218 'errno' : 'foo', 'strerror' : 'bar'}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000219 (IOError, ('foo', 'bar', 'baz'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000220 {'args' : ('foo', 'bar'), 'filename' : 'baz',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000221 'errno' : 'foo', 'strerror' : 'bar'}),
222 (IOError, ('foo', 'bar', 'baz', 'quux'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000223 {'args' : ('foo', 'bar', 'baz', 'quux')}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000224 (EnvironmentError, ('errnoStr', 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000225 {'args' : ('errnoStr', 'strErrorStr'),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000226 'strerror' : 'strErrorStr', 'errno' : 'errnoStr',
227 'filename' : 'filenameStr'}),
228 (EnvironmentError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000229 {'args' : (1, 'strErrorStr'), 'errno' : 1,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000230 'strerror' : 'strErrorStr', 'filename' : 'filenameStr'}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000231 (SyntaxError, (), {'msg' : None, 'text' : None,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000232 'filename' : None, 'lineno' : None, 'offset' : None,
233 'print_file_and_line' : None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000234 (SyntaxError, ('msgStr',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000235 {'args' : ('msgStr',), 'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000236 'print_file_and_line' : None, 'msg' : 'msgStr',
237 'filename' : None, 'lineno' : None, 'offset' : None}),
238 (SyntaxError, ('msgStr', ('filenameStr', 'linenoStr', 'offsetStr',
239 'textStr')),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000240 {'offset' : 'offsetStr', 'text' : 'textStr',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000241 'args' : ('msgStr', ('filenameStr', 'linenoStr',
242 'offsetStr', 'textStr')),
243 'print_file_and_line' : None, 'msg' : 'msgStr',
244 'filename' : 'filenameStr', 'lineno' : 'linenoStr'}),
245 (SyntaxError, ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
246 'textStr', 'print_file_and_lineStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000247 {'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000248 'args' : ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
249 'textStr', 'print_file_and_lineStr'),
250 'print_file_and_line' : None, 'msg' : 'msgStr',
251 'filename' : None, 'lineno' : None, 'offset' : None}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000252 (UnicodeError, (), {'args' : (),}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000253 (UnicodeEncodeError, ('ascii', 'a', 0, 1,
254 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000255 {'args' : ('ascii', 'a', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000256 'ordinal not in range'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000257 'encoding' : 'ascii', 'object' : 'a',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000258 'start' : 0, 'reason' : 'ordinal not in range'}),
Guido van Rossum254348e2007-11-21 19:29:53 +0000259 (UnicodeDecodeError, ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000260 'ordinal not in range'),
Guido van Rossum254348e2007-11-21 19:29:53 +0000261 {'args' : ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000262 'ordinal not in range'),
263 'encoding' : 'ascii', 'object' : b'\xff',
264 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000265 (UnicodeDecodeError, ('ascii', b'\xff', 0, 1,
266 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000267 {'args' : ('ascii', b'\xff', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000268 'ordinal not in range'),
Guido van Rossumb8142c32007-05-08 17:49:10 +0000269 'encoding' : 'ascii', 'object' : b'\xff',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000270 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000271 (UnicodeTranslateError, ("\u3042", 0, 1, "ouch"),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000272 {'args' : ('\u3042', 0, 1, 'ouch'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000273 'object' : '\u3042', 'reason' : 'ouch',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000274 'start' : 0, 'end' : 1}),
Thomas Wouters477c8d52006-05-27 19:21:47 +0000275 ]
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000276 try:
277 exceptionList.append(
278 (WindowsError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000279 {'args' : (1, 'strErrorStr'),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000280 'strerror' : 'strErrorStr', 'winerror' : 1,
281 'errno' : 22, 'filename' : 'filenameStr'})
282 )
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000283 except NameError:
284 pass
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000285
Guido van Rossumebe3e162007-05-17 18:20:34 +0000286 for exc, args, expected in exceptionList:
287 try:
288 e = exc(*args)
289 except:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000290 print("\nexc=%r, args=%r" % (exc, args), file=sys.stderr)
Guido van Rossumebe3e162007-05-17 18:20:34 +0000291 raise
292 else:
293 # Verify module name
Ezio Melottib3aedd42010-11-20 19:04:17 +0000294 self.assertEqual(type(e).__module__, 'builtins')
Guido van Rossumebe3e162007-05-17 18:20:34 +0000295 # Verify no ref leaks in Exc_str()
296 s = str(e)
297 for checkArgName in expected:
298 value = getattr(e, checkArgName)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000299 self.assertEqual(repr(value),
300 repr(expected[checkArgName]),
301 '%r.%s == %r, expected %r' % (
302 e, checkArgName,
303 value, expected[checkArgName]))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000304
Guido van Rossumebe3e162007-05-17 18:20:34 +0000305 # test for pickling support
Guido van Rossum99603b02007-07-20 00:22:32 +0000306 for p in [pickle]:
Guido van Rossumebe3e162007-05-17 18:20:34 +0000307 for protocol in range(p.HIGHEST_PROTOCOL + 1):
308 s = p.dumps(e, protocol)
309 new = p.loads(s)
310 for checkArgName in expected:
311 got = repr(getattr(new, checkArgName))
312 want = repr(expected[checkArgName])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000313 self.assertEqual(got, want,
314 'pickled "%r", attribute "%s' %
315 (e, checkArgName))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000316
Collin Winter828f04a2007-08-31 00:04:24 +0000317 def testWithTraceback(self):
318 try:
319 raise IndexError(4)
320 except:
321 tb = sys.exc_info()[2]
322
323 e = BaseException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000324 self.assertIsInstance(e, BaseException)
Collin Winter828f04a2007-08-31 00:04:24 +0000325 self.assertEqual(e.__traceback__, tb)
326
327 e = IndexError(5).with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000328 self.assertIsInstance(e, IndexError)
Collin Winter828f04a2007-08-31 00:04:24 +0000329 self.assertEqual(e.__traceback__, tb)
330
331 class MyException(Exception):
332 pass
333
334 e = MyException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000335 self.assertIsInstance(e, MyException)
Collin Winter828f04a2007-08-31 00:04:24 +0000336 self.assertEqual(e.__traceback__, tb)
337
338 def testInvalidTraceback(self):
339 try:
340 Exception().__traceback__ = 5
341 except TypeError as e:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000342 self.assertIn("__traceback__ must be a traceback", str(e))
Collin Winter828f04a2007-08-31 00:04:24 +0000343 else:
344 self.fail("No exception raised")
345
Georg Brandlab6f2f62009-03-31 04:16:10 +0000346 def testInvalidAttrs(self):
347 self.assertRaises(TypeError, setattr, Exception(), '__cause__', 1)
348 self.assertRaises(TypeError, delattr, Exception(), '__cause__')
349 self.assertRaises(TypeError, setattr, Exception(), '__context__', 1)
350 self.assertRaises(TypeError, delattr, Exception(), '__context__')
351
Collin Winter828f04a2007-08-31 00:04:24 +0000352 def testNoneClearsTracebackAttr(self):
353 try:
354 raise IndexError(4)
355 except:
356 tb = sys.exc_info()[2]
357
358 e = Exception()
359 e.__traceback__ = tb
360 e.__traceback__ = None
361 self.assertEqual(e.__traceback__, None)
362
363 def testChainingAttrs(self):
364 e = Exception()
365 self.assertEqual(e.__context__, None)
366 self.assertEqual(e.__cause__, None)
367
368 e = TypeError()
369 self.assertEqual(e.__context__, None)
370 self.assertEqual(e.__cause__, None)
371
372 class MyException(EnvironmentError):
373 pass
374
375 e = MyException()
376 self.assertEqual(e.__context__, None)
377 self.assertEqual(e.__cause__, None)
378
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000379 def testKeywordArgs(self):
380 # test that builtin exception don't take keyword args,
381 # but user-defined subclasses can if they want
382 self.assertRaises(TypeError, BaseException, a=1)
383
384 class DerivedException(BaseException):
385 def __init__(self, fancy_arg):
386 BaseException.__init__(self)
387 self.fancy_arg = fancy_arg
388
389 x = DerivedException(fancy_arg=42)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000390 self.assertEqual(x.fancy_arg, 42)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000391
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000392 def testInfiniteRecursion(self):
393 def f():
394 return f()
395 self.assertRaises(RuntimeError, f)
396
397 def g():
398 try:
399 return g()
400 except ValueError:
401 return -1
402 self.assertRaises(RuntimeError, g)
403
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000404 def test_str(self):
405 # Make sure both instances and classes have a str representation.
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000406 self.assertTrue(str(Exception))
407 self.assertTrue(str(Exception('a')))
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000408 self.assertTrue(str(Exception('a', 'b')))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000409
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000410 def testExceptionCleanupNames(self):
411 # Make sure the local variable bound to the exception instance by
412 # an "except" statement is only visible inside the except block.
Guido van Rossumb940e112007-01-10 16:19:56 +0000413 try:
414 raise Exception()
415 except Exception as e:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000416 self.assertTrue(e)
Guido van Rossumb940e112007-01-10 16:19:56 +0000417 del e
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000418 self.assertNotIn('e', locals())
Guido van Rossumb940e112007-01-10 16:19:56 +0000419
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000420 def testExceptionCleanupState(self):
421 # Make sure exception state is cleaned up as soon as the except
422 # block is left. See #2507
423
424 class MyException(Exception):
425 def __init__(self, obj):
426 self.obj = obj
427 class MyObj:
428 pass
429
430 def inner_raising_func():
431 # Create some references in exception value and traceback
432 local_ref = obj
433 raise MyException(obj)
434
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000435 # Qualified "except" with "as"
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000436 obj = MyObj()
437 wr = weakref.ref(obj)
438 try:
439 inner_raising_func()
440 except MyException as e:
441 pass
442 obj = None
443 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000444 self.assertTrue(obj is None, "%s" % obj)
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000445
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000446 # Qualified "except" without "as"
447 obj = MyObj()
448 wr = weakref.ref(obj)
449 try:
450 inner_raising_func()
451 except MyException:
452 pass
453 obj = None
454 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000455 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000456
457 # Bare "except"
458 obj = MyObj()
459 wr = weakref.ref(obj)
460 try:
461 inner_raising_func()
462 except:
463 pass
464 obj = None
465 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000466 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000467
468 # "except" with premature block leave
469 obj = MyObj()
470 wr = weakref.ref(obj)
471 for i in [0]:
472 try:
473 inner_raising_func()
474 except:
475 break
476 obj = None
477 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000478 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000479
480 # "except" block raising another exception
481 obj = MyObj()
482 wr = weakref.ref(obj)
483 try:
484 try:
485 inner_raising_func()
486 except:
487 raise KeyError
Guido van Rossumb4fb6e42008-06-14 20:20:24 +0000488 except KeyError as e:
489 # We want to test that the except block above got rid of
490 # the exception raised in inner_raising_func(), but it
491 # also ends up in the __context__ of the KeyError, so we
492 # must clear the latter manually for our test to succeed.
493 e.__context__ = None
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000494 obj = None
495 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000496 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000497
498 # Some complicated construct
499 obj = MyObj()
500 wr = weakref.ref(obj)
501 try:
502 inner_raising_func()
503 except MyException:
504 try:
505 try:
506 raise
507 finally:
508 raise
509 except MyException:
510 pass
511 obj = None
512 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000513 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000514
515 # Inside an exception-silencing "with" block
516 class Context:
517 def __enter__(self):
518 return self
519 def __exit__ (self, exc_type, exc_value, exc_tb):
520 return True
521 obj = MyObj()
522 wr = weakref.ref(obj)
523 with Context():
524 inner_raising_func()
525 obj = None
526 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000527 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000528
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000529 def test_exception_target_in_nested_scope(self):
530 # issue 4617: This used to raise a SyntaxError
531 # "can not delete variable 'e' referenced in nested scope"
532 def print_error():
533 e
534 try:
535 something
536 except Exception as e:
537 print_error()
538 # implicit "del e" here
539
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000540 def test_generator_leaking(self):
541 # Test that generator exception state doesn't leak into the calling
542 # frame
543 def yield_raise():
544 try:
545 raise KeyError("caught")
546 except KeyError:
547 yield sys.exc_info()[0]
548 yield sys.exc_info()[0]
549 yield sys.exc_info()[0]
550 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000551 self.assertEqual(next(g), KeyError)
552 self.assertEqual(sys.exc_info()[0], None)
553 self.assertEqual(next(g), KeyError)
554 self.assertEqual(sys.exc_info()[0], None)
555 self.assertEqual(next(g), None)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000556
557 # Same test, but inside an exception handler
558 try:
559 raise TypeError("foo")
560 except TypeError:
561 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000562 self.assertEqual(next(g), KeyError)
563 self.assertEqual(sys.exc_info()[0], TypeError)
564 self.assertEqual(next(g), KeyError)
565 self.assertEqual(sys.exc_info()[0], TypeError)
566 self.assertEqual(next(g), TypeError)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000567 del g
Ezio Melottib3aedd42010-11-20 19:04:17 +0000568 self.assertEqual(sys.exc_info()[0], TypeError)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000569
Benjamin Peterson83195c32011-07-03 13:44:00 -0500570 def test_generator_leaking2(self):
571 # See issue 12475.
572 def g():
573 yield
574 try:
575 raise RuntimeError
576 except RuntimeError:
577 it = g()
578 next(it)
579 try:
580 next(it)
581 except StopIteration:
582 pass
583 self.assertEqual(sys.exc_info(), (None, None, None))
584
Benjamin Petersonac913412011-07-03 16:25:11 -0500585 def test_generator_doesnt_retain_old_exc(self):
586 def g():
587 self.assertIsInstance(sys.exc_info()[1], RuntimeError)
588 yield
589 self.assertEqual(sys.exc_info(), (None, None, None))
590 it = g()
591 try:
592 raise RuntimeError
593 except RuntimeError:
594 next(it)
595 self.assertRaises(StopIteration, next, it)
596
Benjamin Petersonae5f2f42010-03-07 17:10:51 +0000597 def test_generator_finalizing_and_exc_info(self):
598 # See #7173
599 def simple_gen():
600 yield 1
601 def run_gen():
602 gen = simple_gen()
603 try:
604 raise RuntimeError
605 except RuntimeError:
606 return next(gen)
607 run_gen()
608 gc_collect()
609 self.assertEqual(sys.exc_info(), (None, None, None))
610
Antoine Pitroua370fcf2011-08-20 14:15:03 +0200611 def _check_generator_cleanup_exc_state(self, testfunc):
612 # Issue #12791: exception state is cleaned up as soon as a generator
613 # is closed (reference cycles are broken).
614 class MyException(Exception):
615 def __init__(self, obj):
616 self.obj = obj
617 class MyObj:
618 pass
619
620 def raising_gen():
621 try:
622 raise MyException(obj)
623 except MyException:
624 yield
625
626 obj = MyObj()
627 wr = weakref.ref(obj)
628 g = raising_gen()
629 next(g)
630 testfunc(g)
631 g = obj = None
632 obj = wr()
633 self.assertIs(obj, None)
634
635 def test_generator_throw_cleanup_exc_state(self):
636 def do_throw(g):
637 try:
638 g.throw(RuntimeError())
639 except RuntimeError:
640 pass
641 self._check_generator_cleanup_exc_state(do_throw)
642
643 def test_generator_close_cleanup_exc_state(self):
644 def do_close(g):
645 g.close()
646 self._check_generator_cleanup_exc_state(do_close)
647
648 def test_generator_del_cleanup_exc_state(self):
649 def do_del(g):
650 g = None
651 self._check_generator_cleanup_exc_state(do_del)
652
653 def test_generator_next_cleanup_exc_state(self):
654 def do_next(g):
655 try:
656 next(g)
657 except StopIteration:
658 pass
659 else:
660 self.fail("should have raised StopIteration")
661 self._check_generator_cleanup_exc_state(do_next)
662
663 def test_generator_send_cleanup_exc_state(self):
664 def do_send(g):
665 try:
666 g.send(None)
667 except StopIteration:
668 pass
669 else:
670 self.fail("should have raised StopIteration")
671 self._check_generator_cleanup_exc_state(do_send)
672
Benjamin Peterson27d63672008-06-15 20:09:12 +0000673 def test_3114(self):
674 # Bug #3114: in its destructor, MyObject retrieves a pointer to
675 # obsolete and/or deallocated objects.
Benjamin Peterson979f3112008-06-15 00:05:44 +0000676 class MyObject:
677 def __del__(self):
678 nonlocal e
679 e = sys.exc_info()
680 e = ()
681 try:
682 raise Exception(MyObject())
683 except:
684 pass
Ezio Melottib3aedd42010-11-20 19:04:17 +0000685 self.assertEqual(e, (None, None, None))
Benjamin Peterson979f3112008-06-15 00:05:44 +0000686
Eric Smith0facd772010-02-24 15:42:29 +0000687 def testUnicodeChangeAttributes(self):
688 # See issue 7309. This was a crasher.
689
690 u = UnicodeEncodeError('baz', 'xxxxx', 1, 5, 'foo')
691 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: foo")
692 u.end = 2
693 self.assertEqual(str(u), "'baz' codec can't encode character '\\x78' in position 1: foo")
694 u.end = 5
695 u.reason = 0x345345345345345345
696 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: 965230951443685724997")
697 u.encoding = 4000
698 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1-4: 965230951443685724997")
699 u.start = 1000
700 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1000-4: 965230951443685724997")
701
702 u = UnicodeDecodeError('baz', b'xxxxx', 1, 5, 'foo')
703 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: foo")
704 u.end = 2
705 self.assertEqual(str(u), "'baz' codec can't decode byte 0x78 in position 1: foo")
706 u.end = 5
707 u.reason = 0x345345345345345345
708 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: 965230951443685724997")
709 u.encoding = 4000
710 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1-4: 965230951443685724997")
711 u.start = 1000
712 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1000-4: 965230951443685724997")
713
714 u = UnicodeTranslateError('xxxx', 1, 5, 'foo')
715 self.assertEqual(str(u), "can't translate characters in position 1-4: foo")
716 u.end = 2
717 self.assertEqual(str(u), "can't translate character '\\x78' in position 1: foo")
718 u.end = 5
719 u.reason = 0x345345345345345345
720 self.assertEqual(str(u), "can't translate characters in position 1-4: 965230951443685724997")
721 u.start = 1000
722 self.assertEqual(str(u), "can't translate characters in position 1000-4: 965230951443685724997")
Benjamin Peterson6e7740c2008-08-20 23:23:34 +0000723
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000724 def test_badisinstance(self):
725 # Bug #2542: if issubclass(e, MyException) raises an exception,
726 # it should be ignored
727 class Meta(type):
728 def __subclasscheck__(cls, subclass):
729 raise ValueError()
730 class MyException(Exception, metaclass=Meta):
731 pass
732
733 with captured_output("stderr") as stderr:
734 try:
735 raise KeyError()
736 except MyException as e:
737 self.fail("exception should not be a MyException")
738 except KeyError:
739 pass
740 except:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000741 self.fail("Should have raised KeyError")
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000742 else:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000743 self.fail("Should have raised KeyError")
744
745 def g():
746 try:
747 return g()
748 except RuntimeError:
749 return sys.exc_info()
750 e, v, tb = g()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000751 self.assertTrue(isinstance(v, RuntimeError), type(v))
Benjamin Peterson577473f2010-01-19 00:09:57 +0000752 self.assertIn("maximum recursion depth exceeded", str(v))
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000753
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000754
755 def test_MemoryError(self):
756 # PyErr_NoMemory always raises the same exception instance.
757 # Check that the traceback is not doubled.
758 import traceback
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000759 from _testcapi import raise_memoryerror
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000760 def raiseMemError():
761 try:
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000762 raise_memoryerror()
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000763 except MemoryError as e:
764 tb = e.__traceback__
765 else:
766 self.fail("Should have raises a MemoryError")
767 return traceback.format_tb(tb)
768
769 tb1 = raiseMemError()
770 tb2 = raiseMemError()
771 self.assertEqual(tb1, tb2)
772
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000773 @cpython_only
Georg Brandl1e28a272009-12-28 08:41:01 +0000774 def test_exception_with_doc(self):
775 import _testcapi
776 doc2 = "This is a test docstring."
777 doc4 = "This is another test docstring."
778
779 self.assertRaises(SystemError, _testcapi.make_exception_with_doc,
780 "error1")
781
782 # test basic usage of PyErr_NewException
783 error1 = _testcapi.make_exception_with_doc("_testcapi.error1")
784 self.assertIs(type(error1), type)
785 self.assertTrue(issubclass(error1, Exception))
786 self.assertIsNone(error1.__doc__)
787
788 # test with given docstring
789 error2 = _testcapi.make_exception_with_doc("_testcapi.error2", doc2)
790 self.assertEqual(error2.__doc__, doc2)
791
792 # test with explicit base (without docstring)
793 error3 = _testcapi.make_exception_with_doc("_testcapi.error3",
794 base=error2)
795 self.assertTrue(issubclass(error3, error2))
796
797 # test with explicit base tuple
798 class C(object):
799 pass
800 error4 = _testcapi.make_exception_with_doc("_testcapi.error4", doc4,
801 (error3, C))
802 self.assertTrue(issubclass(error4, error3))
803 self.assertTrue(issubclass(error4, C))
804 self.assertEqual(error4.__doc__, doc4)
805
806 # test with explicit dictionary
807 error5 = _testcapi.make_exception_with_doc("_testcapi.error5", "",
808 error4, {'a': 1})
809 self.assertTrue(issubclass(error5, error4))
810 self.assertEqual(error5.a, 1)
811 self.assertEqual(error5.__doc__, "")
812
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000813 def test_memory_error_cleanup(self):
814 # Issue #5437: preallocated MemoryError instances should not keep
815 # traceback objects alive.
816 from _testcapi import raise_memoryerror
817 class C:
818 pass
819 wr = None
820 def inner():
821 nonlocal wr
822 c = C()
823 wr = weakref.ref(c)
824 raise_memoryerror()
825 # We cannot use assertRaises since it manually deletes the traceback
826 try:
827 inner()
828 except MemoryError as e:
829 self.assertNotEqual(wr(), None)
830 else:
831 self.fail("MemoryError not raised")
832 self.assertEqual(wr(), None)
833
834 def test_recursion_error_cleanup(self):
835 # Same test as above, but with "recursion exceeded" errors
836 class C:
837 pass
838 wr = None
839 def inner():
840 nonlocal wr
841 c = C()
842 wr = weakref.ref(c)
843 inner()
844 # We cannot use assertRaises since it manually deletes the traceback
845 try:
846 inner()
847 except RuntimeError as e:
848 self.assertNotEqual(wr(), None)
849 else:
850 self.fail("RuntimeError not raised")
851 self.assertEqual(wr(), None)
Georg Brandl1e28a272009-12-28 08:41:01 +0000852
Antoine Pitroua7622852011-09-01 21:37:43 +0200853 def test_errno_ENOTDIR(self):
854 # Issue #12802: "not a directory" errors are ENOTDIR even on Windows
855 with self.assertRaises(OSError) as cm:
856 os.listdir(__file__)
857 self.assertEqual(cm.exception.errno, errno.ENOTDIR, cm.exception)
858
859
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000860def test_main():
861 run_unittest(ExceptionTests)
862
863if __name__ == '__main__':
Guido van Rossumb8142c32007-05-08 17:49:10 +0000864 unittest.main()