blob: b35a5e4a57ace234fdda4a138468526a828ff401 [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
Philip Jenveyb37ac8e2012-11-14 14:37:24 -080010from test.support import (TESTFN, captured_output, check_impl_detail,
Philip Jenvey8f2b6ad2012-11-14 14:51:44 -080011 check_warnings, cpython_only, gc_collect, run_unittest,
12 no_tracing, unlink)
Guido van Rossum83b120d2001-08-23 03:23:03 +000013
Richard Oudkerk5562d9d2012-07-28 17:45:28 +010014class NaiveException(Exception):
15 def __init__(self, x):
16 self.x = x
17
18class SlottedNaiveException(Exception):
19 __slots__ = ('x',)
20 def __init__(self, x):
21 self.x = x
22
Guido van Rossum3bead091992-01-27 17:00:37 +000023# XXX This is not really enough, each *operation* should be tested!
24
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000025class ExceptionTests(unittest.TestCase):
Barry Warsawb9c1d3d2001-08-13 23:07:00 +000026
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000027 def raise_catch(self, exc, excname):
28 try:
Collin Winter828f04a2007-08-31 00:04:24 +000029 raise exc("spam")
Guido van Rossumb940e112007-01-10 16:19:56 +000030 except exc as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000031 buf1 = str(err)
32 try:
33 raise exc("spam")
Guido van Rossumb940e112007-01-10 16:19:56 +000034 except exc as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000035 buf2 = str(err)
Ezio Melottib3aedd42010-11-20 19:04:17 +000036 self.assertEqual(buf1, buf2)
37 self.assertEqual(exc.__name__, excname)
Guido van Rossum3bead091992-01-27 17:00:37 +000038
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000039 def testRaising(self):
40 self.raise_catch(AttributeError, "AttributeError")
41 self.assertRaises(AttributeError, getattr, sys, "undefined_attribute")
Guido van Rossum3bead091992-01-27 17:00:37 +000042
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000043 self.raise_catch(EOFError, "EOFError")
44 fp = open(TESTFN, 'w')
45 fp.close()
46 fp = open(TESTFN, 'r')
47 savestdin = sys.stdin
48 try:
49 try:
50 import marshal
Antoine Pitrou4a90ef02012-03-03 02:35:32 +010051 marshal.loads(b'')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000052 except EOFError:
53 pass
54 finally:
55 sys.stdin = savestdin
56 fp.close()
57 unlink(TESTFN)
Guido van Rossum3bead091992-01-27 17:00:37 +000058
Antoine Pitrou6b4883d2011-10-12 02:54:14 +020059 self.raise_catch(OSError, "OSError")
60 self.assertRaises(OSError, open, 'this file does not exist', 'r')
Guido van Rossum3bead091992-01-27 17:00:37 +000061
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000062 self.raise_catch(ImportError, "ImportError")
63 self.assertRaises(ImportError, __import__, "undefined_module")
Guido van Rossum3bead091992-01-27 17:00:37 +000064
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000065 self.raise_catch(IndexError, "IndexError")
66 x = []
67 self.assertRaises(IndexError, x.__getitem__, 10)
Guido van Rossum3bead091992-01-27 17:00:37 +000068
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000069 self.raise_catch(KeyError, "KeyError")
70 x = {}
71 self.assertRaises(KeyError, x.__getitem__, 'key')
Guido van Rossum3bead091992-01-27 17:00:37 +000072
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000073 self.raise_catch(KeyboardInterrupt, "KeyboardInterrupt")
Guido van Rossum3bead091992-01-27 17:00:37 +000074
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000075 self.raise_catch(MemoryError, "MemoryError")
Guido van Rossum3bead091992-01-27 17:00:37 +000076
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000077 self.raise_catch(NameError, "NameError")
78 try: x = undefined_variable
79 except NameError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000080
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000081 self.raise_catch(OverflowError, "OverflowError")
82 x = 1
83 for dummy in range(128):
84 x += x # this simply shouldn't blow up
Guido van Rossum3bead091992-01-27 17:00:37 +000085
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000086 self.raise_catch(RuntimeError, "RuntimeError")
Guido van Rossum3bead091992-01-27 17:00:37 +000087
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000088 self.raise_catch(SyntaxError, "SyntaxError")
Georg Brandl7cae87c2006-09-06 06:51:57 +000089 try: exec('/\n')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000090 except SyntaxError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000091
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000092 self.raise_catch(IndentationError, "IndentationError")
Fred Drake72e48bd2000-09-08 16:32:34 +000093
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000094 self.raise_catch(TabError, "TabError")
Georg Brandle1b5ac62008-06-04 13:06:58 +000095 try: compile("try:\n\t1/0\n \t1/0\nfinally:\n pass\n",
96 '<string>', 'exec')
97 except TabError: pass
98 else: self.fail("TabError not raised")
Fred Drake72e48bd2000-09-08 16:32:34 +000099
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000100 self.raise_catch(SystemError, "SystemError")
Fred Drake72e48bd2000-09-08 16:32:34 +0000101
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000102 self.raise_catch(SystemExit, "SystemExit")
103 self.assertRaises(SystemExit, sys.exit, 0)
Fred Drake85f36392000-07-11 17:53:00 +0000104
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000105 self.raise_catch(TypeError, "TypeError")
106 try: [] + ()
107 except TypeError: pass
Fred Drake85f36392000-07-11 17:53:00 +0000108
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000109 self.raise_catch(ValueError, "ValueError")
Guido van Rossume63bae62007-07-17 00:34:25 +0000110 self.assertRaises(ValueError, chr, 17<<16)
Guido van Rossum3bead091992-01-27 17:00:37 +0000111
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000112 self.raise_catch(ZeroDivisionError, "ZeroDivisionError")
113 try: x = 1/0
114 except ZeroDivisionError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000115
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000116 self.raise_catch(Exception, "Exception")
117 try: x = 1/0
Guido van Rossumb940e112007-01-10 16:19:56 +0000118 except Exception as e: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000119
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000120 def testSyntaxErrorMessage(self):
121 # make sure the right exception message is raised for each of
122 # these code fragments
Guido van Rossum3bead091992-01-27 17:00:37 +0000123
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000124 def ckmsg(src, msg):
125 try:
126 compile(src, '<fragment>', 'exec')
Guido van Rossumb940e112007-01-10 16:19:56 +0000127 except SyntaxError as e:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000128 if e.msg != msg:
129 self.fail("expected %s, got %s" % (msg, e.msg))
130 else:
131 self.fail("failed to get expected SyntaxError")
Guido van Rossum3bead091992-01-27 17:00:37 +0000132
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000133 s = '''while 1:
134 try:
135 pass
136 finally:
137 continue'''
Barry Warsaw992cb8a2000-05-25 23:16:54 +0000138
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000139 if not sys.platform.startswith('java'):
140 ckmsg(s, "'continue' not supported inside 'finally' clause")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000141
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000142 s = '''if 1:
143 try:
144 continue
145 except:
146 pass'''
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000147
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000148 ckmsg(s, "'continue' not properly in loop")
149 ckmsg("continue\n", "'continue' not properly in loop")
Thomas Wouters303de6a2006-04-20 22:42:37 +0000150
Serhiy Storchaka65fd0592014-01-21 22:26:52 +0200151 def testSyntaxErrorOffset(self):
152 def check(src, lineno, offset):
153 with self.assertRaises(SyntaxError) as cm:
154 compile(src, '<fragment>', 'exec')
155 self.assertEqual(cm.exception.lineno, lineno)
156 self.assertEqual(cm.exception.offset, offset)
157
158 check('def fact(x):\n\treturn x!\n', 2, 10)
159 check('1 +\n', 1, 4)
160 check('def spam():\n print(1)\n print(2)', 3, 10)
161 check('Python = "Python" +', 1, 20)
162 check('Python = "\u1e54\xfd\u0163\u0125\xf2\xf1" +', 1, 20)
163
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000164 @cpython_only
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000165 def testSettingException(self):
166 # test that setting an exception at the C level works even if the
167 # exception object can't be constructed.
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000168
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000169 class BadException(Exception):
170 def __init__(self_):
Collin Winter828f04a2007-08-31 00:04:24 +0000171 raise RuntimeError("can't instantiate BadException")
Finn Bockaa3dc452001-12-08 10:15:48 +0000172
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000173 class InvalidException:
174 pass
Thomas Wouters303de6a2006-04-20 22:42:37 +0000175
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000176 def test_capi1():
177 import _testcapi
178 try:
179 _testcapi.raise_exception(BadException, 1)
Guido van Rossumb940e112007-01-10 16:19:56 +0000180 except TypeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000181 exc, err, tb = sys.exc_info()
182 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000183 self.assertEqual(co.co_name, "test_capi1")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000184 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000185 else:
186 self.fail("Expected exception")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000187
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000188 def test_capi2():
189 import _testcapi
190 try:
191 _testcapi.raise_exception(BadException, 0)
Guido van Rossumb940e112007-01-10 16:19:56 +0000192 except RuntimeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000193 exc, err, tb = sys.exc_info()
194 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000195 self.assertEqual(co.co_name, "__init__")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000196 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000197 co2 = tb.tb_frame.f_back.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000198 self.assertEqual(co2.co_name, "test_capi2")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000199 else:
200 self.fail("Expected exception")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000201
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000202 def test_capi3():
203 import _testcapi
204 self.assertRaises(SystemError, _testcapi.raise_exception,
205 InvalidException, 1)
206
207 if not sys.platform.startswith('java'):
208 test_capi1()
209 test_capi2()
210 test_capi3()
211
Thomas Wouters89f507f2006-12-13 04:49:30 +0000212 def test_WindowsError(self):
213 try:
214 WindowsError
215 except NameError:
216 pass
217 else:
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200218 self.assertIs(WindowsError, OSError)
219 self.assertEqual(str(OSError(1001)), "1001")
220 self.assertEqual(str(OSError(1001, "message")),
221 "[Errno 1001] message")
222 # POSIX errno (9 aka EBADF) is untranslated
223 w = OSError(9, 'foo', 'bar')
224 self.assertEqual(w.errno, 9)
225 self.assertEqual(w.winerror, None)
226 self.assertEqual(str(w), "[Errno 9] foo: 'bar'")
227 # ERROR_PATH_NOT_FOUND (win error 3) becomes ENOENT (2)
228 w = OSError(0, 'foo', 'bar', 3)
229 self.assertEqual(w.errno, 2)
230 self.assertEqual(w.winerror, 3)
231 self.assertEqual(w.strerror, 'foo')
232 self.assertEqual(w.filename, 'bar')
Richard Oudkerk30147712012-08-28 19:33:26 +0100233 self.assertEqual(str(w), "[WinError 3] foo: 'bar'")
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200234 # Unknown win error becomes EINVAL (22)
235 w = OSError(0, 'foo', None, 1001)
236 self.assertEqual(w.errno, 22)
237 self.assertEqual(w.winerror, 1001)
238 self.assertEqual(w.strerror, 'foo')
239 self.assertEqual(w.filename, None)
Richard Oudkerk30147712012-08-28 19:33:26 +0100240 self.assertEqual(str(w), "[WinError 1001] foo")
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200241 # Non-numeric "errno"
242 w = OSError('bar', 'foo')
243 self.assertEqual(w.errno, 'bar')
244 self.assertEqual(w.winerror, None)
245 self.assertEqual(w.strerror, 'foo')
246 self.assertEqual(w.filename, None)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000247
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000248 def testAttributes(self):
249 # test that exception attributes are happy
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000250
251 exceptionList = [
Guido van Rossumebe3e162007-05-17 18:20:34 +0000252 (BaseException, (), {'args' : ()}),
253 (BaseException, (1, ), {'args' : (1,)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000254 (BaseException, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000255 {'args' : ('foo',)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000256 (BaseException, ('foo', 1),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000257 {'args' : ('foo', 1)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000258 (SystemExit, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000259 {'args' : ('foo',), 'code' : 'foo'}),
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200260 (OSError, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000261 {'args' : ('foo',), 'filename' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000262 'errno' : None, 'strerror' : None}),
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200263 (OSError, ('foo', 'bar'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000264 {'args' : ('foo', 'bar'), 'filename' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000265 'errno' : 'foo', 'strerror' : 'bar'}),
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200266 (OSError, ('foo', 'bar', 'baz'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000267 {'args' : ('foo', 'bar'), 'filename' : 'baz',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000268 'errno' : 'foo', 'strerror' : 'bar'}),
Larry Hastingsb0827312014-02-09 22:05:19 -0800269 (OSError, ('foo', 'bar', 'baz', None, 'quux'),
270 {'args' : ('foo', 'bar'), 'filename' : 'baz', 'filename2': 'quux'}),
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200271 (OSError, ('errnoStr', 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000272 {'args' : ('errnoStr', 'strErrorStr'),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000273 'strerror' : 'strErrorStr', 'errno' : 'errnoStr',
274 'filename' : 'filenameStr'}),
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200275 (OSError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000276 {'args' : (1, 'strErrorStr'), 'errno' : 1,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000277 'strerror' : 'strErrorStr', 'filename' : 'filenameStr'}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000278 (SyntaxError, (), {'msg' : None, 'text' : None,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000279 'filename' : None, 'lineno' : None, 'offset' : None,
280 'print_file_and_line' : None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000281 (SyntaxError, ('msgStr',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000282 {'args' : ('msgStr',), 'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000283 'print_file_and_line' : None, 'msg' : 'msgStr',
284 'filename' : None, 'lineno' : None, 'offset' : None}),
285 (SyntaxError, ('msgStr', ('filenameStr', 'linenoStr', 'offsetStr',
286 'textStr')),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000287 {'offset' : 'offsetStr', 'text' : 'textStr',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000288 'args' : ('msgStr', ('filenameStr', 'linenoStr',
289 'offsetStr', 'textStr')),
290 'print_file_and_line' : None, 'msg' : 'msgStr',
291 'filename' : 'filenameStr', 'lineno' : 'linenoStr'}),
292 (SyntaxError, ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
293 'textStr', 'print_file_and_lineStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000294 {'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000295 'args' : ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
296 'textStr', 'print_file_and_lineStr'),
297 'print_file_and_line' : None, 'msg' : 'msgStr',
298 'filename' : None, 'lineno' : None, 'offset' : None}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000299 (UnicodeError, (), {'args' : (),}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000300 (UnicodeEncodeError, ('ascii', 'a', 0, 1,
301 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000302 {'args' : ('ascii', 'a', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000303 'ordinal not in range'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000304 'encoding' : 'ascii', 'object' : 'a',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000305 'start' : 0, 'reason' : 'ordinal not in range'}),
Guido van Rossum254348e2007-11-21 19:29:53 +0000306 (UnicodeDecodeError, ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000307 'ordinal not in range'),
Guido van Rossum254348e2007-11-21 19:29:53 +0000308 {'args' : ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000309 'ordinal not in range'),
310 'encoding' : 'ascii', 'object' : b'\xff',
311 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000312 (UnicodeDecodeError, ('ascii', b'\xff', 0, 1,
313 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000314 {'args' : ('ascii', b'\xff', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000315 'ordinal not in range'),
Guido van Rossumb8142c32007-05-08 17:49:10 +0000316 'encoding' : 'ascii', 'object' : b'\xff',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000317 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000318 (UnicodeTranslateError, ("\u3042", 0, 1, "ouch"),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000319 {'args' : ('\u3042', 0, 1, 'ouch'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000320 'object' : '\u3042', 'reason' : 'ouch',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000321 'start' : 0, 'end' : 1}),
Richard Oudkerk5562d9d2012-07-28 17:45:28 +0100322 (NaiveException, ('foo',),
323 {'args': ('foo',), 'x': 'foo'}),
324 (SlottedNaiveException, ('foo',),
325 {'args': ('foo',), 'x': 'foo'}),
Thomas Wouters477c8d52006-05-27 19:21:47 +0000326 ]
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000327 try:
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200328 # More tests are in test_WindowsError
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000329 exceptionList.append(
330 (WindowsError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000331 {'args' : (1, 'strErrorStr'),
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200332 'strerror' : 'strErrorStr', 'winerror' : None,
333 'errno' : 1, 'filename' : 'filenameStr'})
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000334 )
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000335 except NameError:
336 pass
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000337
Guido van Rossumebe3e162007-05-17 18:20:34 +0000338 for exc, args, expected in exceptionList:
339 try:
340 e = exc(*args)
341 except:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000342 print("\nexc=%r, args=%r" % (exc, args), file=sys.stderr)
Guido van Rossumebe3e162007-05-17 18:20:34 +0000343 raise
344 else:
345 # Verify module name
Richard Oudkerk5562d9d2012-07-28 17:45:28 +0100346 if not type(e).__name__.endswith('NaiveException'):
347 self.assertEqual(type(e).__module__, 'builtins')
Guido van Rossumebe3e162007-05-17 18:20:34 +0000348 # Verify no ref leaks in Exc_str()
349 s = str(e)
350 for checkArgName in expected:
351 value = getattr(e, checkArgName)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000352 self.assertEqual(repr(value),
353 repr(expected[checkArgName]),
354 '%r.%s == %r, expected %r' % (
355 e, checkArgName,
356 value, expected[checkArgName]))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000357
Guido van Rossumebe3e162007-05-17 18:20:34 +0000358 # test for pickling support
Guido van Rossum99603b02007-07-20 00:22:32 +0000359 for p in [pickle]:
Guido van Rossumebe3e162007-05-17 18:20:34 +0000360 for protocol in range(p.HIGHEST_PROTOCOL + 1):
361 s = p.dumps(e, protocol)
362 new = p.loads(s)
363 for checkArgName in expected:
364 got = repr(getattr(new, checkArgName))
365 want = repr(expected[checkArgName])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000366 self.assertEqual(got, want,
367 'pickled "%r", attribute "%s' %
368 (e, checkArgName))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000369
Collin Winter828f04a2007-08-31 00:04:24 +0000370 def testWithTraceback(self):
371 try:
372 raise IndexError(4)
373 except:
374 tb = sys.exc_info()[2]
375
376 e = BaseException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000377 self.assertIsInstance(e, BaseException)
Collin Winter828f04a2007-08-31 00:04:24 +0000378 self.assertEqual(e.__traceback__, tb)
379
380 e = IndexError(5).with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000381 self.assertIsInstance(e, IndexError)
Collin Winter828f04a2007-08-31 00:04:24 +0000382 self.assertEqual(e.__traceback__, tb)
383
384 class MyException(Exception):
385 pass
386
387 e = MyException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000388 self.assertIsInstance(e, MyException)
Collin Winter828f04a2007-08-31 00:04:24 +0000389 self.assertEqual(e.__traceback__, tb)
390
391 def testInvalidTraceback(self):
392 try:
393 Exception().__traceback__ = 5
394 except TypeError as e:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000395 self.assertIn("__traceback__ must be a traceback", str(e))
Collin Winter828f04a2007-08-31 00:04:24 +0000396 else:
397 self.fail("No exception raised")
398
Georg Brandlab6f2f62009-03-31 04:16:10 +0000399 def testInvalidAttrs(self):
400 self.assertRaises(TypeError, setattr, Exception(), '__cause__', 1)
401 self.assertRaises(TypeError, delattr, Exception(), '__cause__')
402 self.assertRaises(TypeError, setattr, Exception(), '__context__', 1)
403 self.assertRaises(TypeError, delattr, Exception(), '__context__')
404
Collin Winter828f04a2007-08-31 00:04:24 +0000405 def testNoneClearsTracebackAttr(self):
406 try:
407 raise IndexError(4)
408 except:
409 tb = sys.exc_info()[2]
410
411 e = Exception()
412 e.__traceback__ = tb
413 e.__traceback__ = None
414 self.assertEqual(e.__traceback__, None)
415
416 def testChainingAttrs(self):
417 e = Exception()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000418 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700419 self.assertIsNone(e.__cause__)
Collin Winter828f04a2007-08-31 00:04:24 +0000420
421 e = TypeError()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000422 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700423 self.assertIsNone(e.__cause__)
Collin Winter828f04a2007-08-31 00:04:24 +0000424
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200425 class MyException(OSError):
Collin Winter828f04a2007-08-31 00:04:24 +0000426 pass
427
428 e = MyException()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000429 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700430 self.assertIsNone(e.__cause__)
Nick Coghlanab7bf212012-02-26 17:49:52 +1000431
432 def testChainingDescriptors(self):
433 try:
434 raise Exception()
435 except Exception as exc:
436 e = exc
437
438 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700439 self.assertIsNone(e.__cause__)
440 self.assertFalse(e.__suppress_context__)
Nick Coghlanab7bf212012-02-26 17:49:52 +1000441
442 e.__context__ = NameError()
443 e.__cause__ = None
444 self.assertIsInstance(e.__context__, NameError)
445 self.assertIsNone(e.__cause__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700446 self.assertTrue(e.__suppress_context__)
447 e.__suppress_context__ = False
448 self.assertFalse(e.__suppress_context__)
Collin Winter828f04a2007-08-31 00:04:24 +0000449
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000450 def testKeywordArgs(self):
451 # test that builtin exception don't take keyword args,
452 # but user-defined subclasses can if they want
453 self.assertRaises(TypeError, BaseException, a=1)
454
455 class DerivedException(BaseException):
456 def __init__(self, fancy_arg):
457 BaseException.__init__(self)
458 self.fancy_arg = fancy_arg
459
460 x = DerivedException(fancy_arg=42)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000461 self.assertEqual(x.fancy_arg, 42)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000462
Brett Cannon31f59292011-02-21 19:29:56 +0000463 @no_tracing
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000464 def testInfiniteRecursion(self):
465 def f():
466 return f()
467 self.assertRaises(RuntimeError, f)
468
469 def g():
470 try:
471 return g()
472 except ValueError:
473 return -1
474 self.assertRaises(RuntimeError, g)
475
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000476 def test_str(self):
477 # Make sure both instances and classes have a str representation.
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000478 self.assertTrue(str(Exception))
479 self.assertTrue(str(Exception('a')))
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000480 self.assertTrue(str(Exception('a', 'b')))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000481
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000482 def testExceptionCleanupNames(self):
483 # Make sure the local variable bound to the exception instance by
484 # an "except" statement is only visible inside the except block.
Guido van Rossumb940e112007-01-10 16:19:56 +0000485 try:
486 raise Exception()
487 except Exception as e:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000488 self.assertTrue(e)
Guido van Rossumb940e112007-01-10 16:19:56 +0000489 del e
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000490 self.assertNotIn('e', locals())
Guido van Rossumb940e112007-01-10 16:19:56 +0000491
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000492 def testExceptionCleanupState(self):
493 # Make sure exception state is cleaned up as soon as the except
494 # block is left. See #2507
495
496 class MyException(Exception):
497 def __init__(self, obj):
498 self.obj = obj
499 class MyObj:
500 pass
501
502 def inner_raising_func():
503 # Create some references in exception value and traceback
504 local_ref = obj
505 raise MyException(obj)
506
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000507 # Qualified "except" with "as"
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000508 obj = MyObj()
509 wr = weakref.ref(obj)
510 try:
511 inner_raising_func()
512 except MyException as e:
513 pass
514 obj = None
515 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000516 self.assertTrue(obj is None, "%s" % obj)
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000517
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000518 # Qualified "except" without "as"
519 obj = MyObj()
520 wr = weakref.ref(obj)
521 try:
522 inner_raising_func()
523 except MyException:
524 pass
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
529 # Bare "except"
530 obj = MyObj()
531 wr = weakref.ref(obj)
532 try:
533 inner_raising_func()
534 except:
535 pass
536 obj = None
537 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000538 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000539
540 # "except" with premature block leave
541 obj = MyObj()
542 wr = weakref.ref(obj)
543 for i in [0]:
544 try:
545 inner_raising_func()
546 except:
547 break
548 obj = None
549 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000550 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000551
552 # "except" block raising another exception
553 obj = MyObj()
554 wr = weakref.ref(obj)
555 try:
556 try:
557 inner_raising_func()
558 except:
559 raise KeyError
Guido van Rossumb4fb6e42008-06-14 20:20:24 +0000560 except KeyError as e:
561 # We want to test that the except block above got rid of
562 # the exception raised in inner_raising_func(), but it
563 # also ends up in the __context__ of the KeyError, so we
564 # must clear the latter manually for our test to succeed.
565 e.__context__ = None
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000566 obj = None
567 obj = wr()
Philip Jenveyb37ac8e2012-11-14 14:37:24 -0800568 # guarantee no ref cycles on CPython (don't gc_collect)
569 if check_impl_detail(cpython=False):
570 gc_collect()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000571 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000572
573 # Some complicated construct
574 obj = MyObj()
575 wr = weakref.ref(obj)
576 try:
577 inner_raising_func()
578 except MyException:
579 try:
580 try:
581 raise
582 finally:
583 raise
584 except MyException:
585 pass
586 obj = None
Philip Jenveyb37ac8e2012-11-14 14:37:24 -0800587 if check_impl_detail(cpython=False):
588 gc_collect()
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000589 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000590 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000591
592 # Inside an exception-silencing "with" block
593 class Context:
594 def __enter__(self):
595 return self
596 def __exit__ (self, exc_type, exc_value, exc_tb):
597 return True
598 obj = MyObj()
599 wr = weakref.ref(obj)
600 with Context():
601 inner_raising_func()
602 obj = None
Philip Jenveyb37ac8e2012-11-14 14:37:24 -0800603 if check_impl_detail(cpython=False):
604 gc_collect()
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000605 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000606 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000607
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000608 def test_exception_target_in_nested_scope(self):
609 # issue 4617: This used to raise a SyntaxError
610 # "can not delete variable 'e' referenced in nested scope"
611 def print_error():
612 e
613 try:
614 something
615 except Exception as e:
616 print_error()
617 # implicit "del e" here
618
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000619 def test_generator_leaking(self):
620 # Test that generator exception state doesn't leak into the calling
621 # frame
622 def yield_raise():
623 try:
624 raise KeyError("caught")
625 except KeyError:
626 yield sys.exc_info()[0]
627 yield sys.exc_info()[0]
628 yield sys.exc_info()[0]
629 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000630 self.assertEqual(next(g), KeyError)
631 self.assertEqual(sys.exc_info()[0], None)
632 self.assertEqual(next(g), KeyError)
633 self.assertEqual(sys.exc_info()[0], None)
634 self.assertEqual(next(g), None)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000635
636 # Same test, but inside an exception handler
637 try:
638 raise TypeError("foo")
639 except TypeError:
640 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000641 self.assertEqual(next(g), KeyError)
642 self.assertEqual(sys.exc_info()[0], TypeError)
643 self.assertEqual(next(g), KeyError)
644 self.assertEqual(sys.exc_info()[0], TypeError)
645 self.assertEqual(next(g), TypeError)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000646 del g
Ezio Melottib3aedd42010-11-20 19:04:17 +0000647 self.assertEqual(sys.exc_info()[0], TypeError)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000648
Benjamin Peterson83195c32011-07-03 13:44:00 -0500649 def test_generator_leaking2(self):
650 # See issue 12475.
651 def g():
652 yield
653 try:
654 raise RuntimeError
655 except RuntimeError:
656 it = g()
657 next(it)
658 try:
659 next(it)
660 except StopIteration:
661 pass
662 self.assertEqual(sys.exc_info(), (None, None, None))
663
Antoine Pitrouc4c19b32015-03-18 22:22:46 +0100664 def test_generator_leaking3(self):
665 # See issue #23353. When gen.throw() is called, the caller's
666 # exception state should be save and restored.
667 def g():
668 try:
669 yield
670 except ZeroDivisionError:
671 yield sys.exc_info()[1]
672 it = g()
673 next(it)
674 try:
675 1/0
676 except ZeroDivisionError as e:
677 self.assertIs(sys.exc_info()[1], e)
678 gen_exc = it.throw(e)
679 self.assertIs(sys.exc_info()[1], e)
680 self.assertIs(gen_exc, e)
681 self.assertEqual(sys.exc_info(), (None, None, None))
682
683 def test_generator_leaking4(self):
684 # See issue #23353. When an exception is raised by a generator,
685 # the caller's exception state should still be restored.
686 def g():
687 try:
688 1/0
689 except ZeroDivisionError:
690 yield sys.exc_info()[0]
691 raise
692 it = g()
693 try:
694 raise TypeError
695 except TypeError:
696 # The caller's exception state (TypeError) is temporarily
697 # saved in the generator.
698 tp = next(it)
699 self.assertIs(tp, ZeroDivisionError)
700 try:
701 next(it)
702 # We can't check it immediately, but while next() returns
703 # with an exception, it shouldn't have restored the old
704 # exception state (TypeError).
705 except ZeroDivisionError as e:
706 self.assertIs(sys.exc_info()[1], e)
707 # We used to find TypeError here.
708 self.assertEqual(sys.exc_info(), (None, None, None))
709
Benjamin Petersonac913412011-07-03 16:25:11 -0500710 def test_generator_doesnt_retain_old_exc(self):
711 def g():
712 self.assertIsInstance(sys.exc_info()[1], RuntimeError)
713 yield
714 self.assertEqual(sys.exc_info(), (None, None, None))
715 it = g()
716 try:
717 raise RuntimeError
718 except RuntimeError:
719 next(it)
720 self.assertRaises(StopIteration, next, it)
721
Benjamin Petersonae5f2f42010-03-07 17:10:51 +0000722 def test_generator_finalizing_and_exc_info(self):
723 # See #7173
724 def simple_gen():
725 yield 1
726 def run_gen():
727 gen = simple_gen()
728 try:
729 raise RuntimeError
730 except RuntimeError:
731 return next(gen)
732 run_gen()
733 gc_collect()
734 self.assertEqual(sys.exc_info(), (None, None, None))
735
Antoine Pitroua370fcf2011-08-20 14:15:03 +0200736 def _check_generator_cleanup_exc_state(self, testfunc):
737 # Issue #12791: exception state is cleaned up as soon as a generator
738 # is closed (reference cycles are broken).
739 class MyException(Exception):
740 def __init__(self, obj):
741 self.obj = obj
742 class MyObj:
743 pass
744
745 def raising_gen():
746 try:
747 raise MyException(obj)
748 except MyException:
749 yield
750
751 obj = MyObj()
752 wr = weakref.ref(obj)
753 g = raising_gen()
754 next(g)
755 testfunc(g)
756 g = obj = None
757 obj = wr()
758 self.assertIs(obj, None)
759
760 def test_generator_throw_cleanup_exc_state(self):
761 def do_throw(g):
762 try:
763 g.throw(RuntimeError())
764 except RuntimeError:
765 pass
766 self._check_generator_cleanup_exc_state(do_throw)
767
768 def test_generator_close_cleanup_exc_state(self):
769 def do_close(g):
770 g.close()
771 self._check_generator_cleanup_exc_state(do_close)
772
773 def test_generator_del_cleanup_exc_state(self):
774 def do_del(g):
775 g = None
776 self._check_generator_cleanup_exc_state(do_del)
777
778 def test_generator_next_cleanup_exc_state(self):
779 def do_next(g):
780 try:
781 next(g)
782 except StopIteration:
783 pass
784 else:
785 self.fail("should have raised StopIteration")
786 self._check_generator_cleanup_exc_state(do_next)
787
788 def test_generator_send_cleanup_exc_state(self):
789 def do_send(g):
790 try:
791 g.send(None)
792 except StopIteration:
793 pass
794 else:
795 self.fail("should have raised StopIteration")
796 self._check_generator_cleanup_exc_state(do_send)
797
Benjamin Peterson27d63672008-06-15 20:09:12 +0000798 def test_3114(self):
799 # Bug #3114: in its destructor, MyObject retrieves a pointer to
800 # obsolete and/or deallocated objects.
Benjamin Peterson979f3112008-06-15 00:05:44 +0000801 class MyObject:
802 def __del__(self):
803 nonlocal e
804 e = sys.exc_info()
805 e = ()
806 try:
807 raise Exception(MyObject())
808 except:
809 pass
Ezio Melottib3aedd42010-11-20 19:04:17 +0000810 self.assertEqual(e, (None, None, None))
Benjamin Peterson979f3112008-06-15 00:05:44 +0000811
Benjamin Peterson24dfb052014-04-02 12:05:35 -0400812 def test_unicode_change_attributes(self):
Eric Smith0facd772010-02-24 15:42:29 +0000813 # See issue 7309. This was a crasher.
814
815 u = UnicodeEncodeError('baz', 'xxxxx', 1, 5, 'foo')
816 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: foo")
817 u.end = 2
818 self.assertEqual(str(u), "'baz' codec can't encode character '\\x78' in position 1: foo")
819 u.end = 5
820 u.reason = 0x345345345345345345
821 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: 965230951443685724997")
822 u.encoding = 4000
823 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1-4: 965230951443685724997")
824 u.start = 1000
825 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1000-4: 965230951443685724997")
826
827 u = UnicodeDecodeError('baz', b'xxxxx', 1, 5, 'foo')
828 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: foo")
829 u.end = 2
830 self.assertEqual(str(u), "'baz' codec can't decode byte 0x78 in position 1: foo")
831 u.end = 5
832 u.reason = 0x345345345345345345
833 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: 965230951443685724997")
834 u.encoding = 4000
835 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1-4: 965230951443685724997")
836 u.start = 1000
837 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1000-4: 965230951443685724997")
838
839 u = UnicodeTranslateError('xxxx', 1, 5, 'foo')
840 self.assertEqual(str(u), "can't translate characters in position 1-4: foo")
841 u.end = 2
842 self.assertEqual(str(u), "can't translate character '\\x78' in position 1: foo")
843 u.end = 5
844 u.reason = 0x345345345345345345
845 self.assertEqual(str(u), "can't translate characters in position 1-4: 965230951443685724997")
846 u.start = 1000
847 self.assertEqual(str(u), "can't translate characters in position 1000-4: 965230951443685724997")
Benjamin Peterson6e7740c2008-08-20 23:23:34 +0000848
Benjamin Peterson9b09ba12014-04-02 12:15:06 -0400849 def test_unicode_errors_no_object(self):
850 # See issue #21134.
Benjamin Petersone3311212014-04-02 15:51:38 -0400851 klasses = UnicodeEncodeError, UnicodeDecodeError, UnicodeTranslateError
Benjamin Peterson9b09ba12014-04-02 12:15:06 -0400852 for klass in klasses:
853 self.assertEqual(str(klass.__new__(klass)), "")
854
Brett Cannon31f59292011-02-21 19:29:56 +0000855 @no_tracing
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000856 def test_badisinstance(self):
857 # Bug #2542: if issubclass(e, MyException) raises an exception,
858 # it should be ignored
859 class Meta(type):
860 def __subclasscheck__(cls, subclass):
861 raise ValueError()
862 class MyException(Exception, metaclass=Meta):
863 pass
864
865 with captured_output("stderr") as stderr:
866 try:
867 raise KeyError()
868 except MyException as e:
869 self.fail("exception should not be a MyException")
870 except KeyError:
871 pass
872 except:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000873 self.fail("Should have raised KeyError")
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000874 else:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000875 self.fail("Should have raised KeyError")
876
877 def g():
878 try:
879 return g()
880 except RuntimeError:
881 return sys.exc_info()
882 e, v, tb = g()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000883 self.assertTrue(isinstance(v, RuntimeError), type(v))
Benjamin Peterson577473f2010-01-19 00:09:57 +0000884 self.assertIn("maximum recursion depth exceeded", str(v))
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000885
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000886
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +0200887 @cpython_only
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000888 def test_MemoryError(self):
889 # PyErr_NoMemory always raises the same exception instance.
890 # Check that the traceback is not doubled.
891 import traceback
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000892 from _testcapi import raise_memoryerror
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000893 def raiseMemError():
894 try:
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000895 raise_memoryerror()
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000896 except MemoryError as e:
897 tb = e.__traceback__
898 else:
899 self.fail("Should have raises a MemoryError")
900 return traceback.format_tb(tb)
901
902 tb1 = raiseMemError()
903 tb2 = raiseMemError()
904 self.assertEqual(tb1, tb2)
905
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000906 @cpython_only
Georg Brandl1e28a272009-12-28 08:41:01 +0000907 def test_exception_with_doc(self):
908 import _testcapi
909 doc2 = "This is a test docstring."
910 doc4 = "This is another test docstring."
911
912 self.assertRaises(SystemError, _testcapi.make_exception_with_doc,
913 "error1")
914
915 # test basic usage of PyErr_NewException
916 error1 = _testcapi.make_exception_with_doc("_testcapi.error1")
917 self.assertIs(type(error1), type)
918 self.assertTrue(issubclass(error1, Exception))
919 self.assertIsNone(error1.__doc__)
920
921 # test with given docstring
922 error2 = _testcapi.make_exception_with_doc("_testcapi.error2", doc2)
923 self.assertEqual(error2.__doc__, doc2)
924
925 # test with explicit base (without docstring)
926 error3 = _testcapi.make_exception_with_doc("_testcapi.error3",
927 base=error2)
928 self.assertTrue(issubclass(error3, error2))
929
930 # test with explicit base tuple
931 class C(object):
932 pass
933 error4 = _testcapi.make_exception_with_doc("_testcapi.error4", doc4,
934 (error3, C))
935 self.assertTrue(issubclass(error4, error3))
936 self.assertTrue(issubclass(error4, C))
937 self.assertEqual(error4.__doc__, doc4)
938
939 # test with explicit dictionary
940 error5 = _testcapi.make_exception_with_doc("_testcapi.error5", "",
941 error4, {'a': 1})
942 self.assertTrue(issubclass(error5, error4))
943 self.assertEqual(error5.a, 1)
944 self.assertEqual(error5.__doc__, "")
945
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +0200946 @cpython_only
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000947 def test_memory_error_cleanup(self):
948 # Issue #5437: preallocated MemoryError instances should not keep
949 # traceback objects alive.
950 from _testcapi import raise_memoryerror
951 class C:
952 pass
953 wr = None
954 def inner():
955 nonlocal wr
956 c = C()
957 wr = weakref.ref(c)
958 raise_memoryerror()
959 # We cannot use assertRaises since it manually deletes the traceback
960 try:
961 inner()
962 except MemoryError as e:
963 self.assertNotEqual(wr(), None)
964 else:
965 self.fail("MemoryError not raised")
966 self.assertEqual(wr(), None)
967
Brett Cannon31f59292011-02-21 19:29:56 +0000968 @no_tracing
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000969 def test_recursion_error_cleanup(self):
970 # Same test as above, but with "recursion exceeded" errors
971 class C:
972 pass
973 wr = None
974 def inner():
975 nonlocal wr
976 c = C()
977 wr = weakref.ref(c)
978 inner()
979 # We cannot use assertRaises since it manually deletes the traceback
980 try:
981 inner()
982 except RuntimeError as e:
983 self.assertNotEqual(wr(), None)
984 else:
985 self.fail("RuntimeError not raised")
986 self.assertEqual(wr(), None)
Georg Brandl1e28a272009-12-28 08:41:01 +0000987
Antoine Pitroua7622852011-09-01 21:37:43 +0200988 def test_errno_ENOTDIR(self):
989 # Issue #12802: "not a directory" errors are ENOTDIR even on Windows
990 with self.assertRaises(OSError) as cm:
991 os.listdir(__file__)
992 self.assertEqual(cm.exception.errno, errno.ENOTDIR, cm.exception)
993
994
Brett Cannon79ec55e2012-04-12 20:24:54 -0400995class ImportErrorTests(unittest.TestCase):
996
997 def test_attributes(self):
998 # Setting 'name' and 'path' should not be a problem.
999 exc = ImportError('test')
1000 self.assertIsNone(exc.name)
1001 self.assertIsNone(exc.path)
1002
1003 exc = ImportError('test', name='somemodule')
1004 self.assertEqual(exc.name, 'somemodule')
1005 self.assertIsNone(exc.path)
1006
1007 exc = ImportError('test', path='somepath')
1008 self.assertEqual(exc.path, 'somepath')
1009 self.assertIsNone(exc.name)
1010
1011 exc = ImportError('test', path='somepath', name='somename')
1012 self.assertEqual(exc.name, 'somename')
1013 self.assertEqual(exc.path, 'somepath')
1014
Brett Cannon07c6e712012-08-24 13:05:09 -04001015 def test_non_str_argument(self):
1016 # Issue #15778
Nadeem Vawda6d708702012-10-14 01:42:32 +02001017 with check_warnings(('', BytesWarning), quiet=True):
1018 arg = b'abc'
1019 exc = ImportError(arg)
1020 self.assertEqual(str(arg), str(exc))
Brett Cannon79ec55e2012-04-12 20:24:54 -04001021
1022
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001023if __name__ == '__main__':
Guido van Rossumb8142c32007-05-08 17:49:10 +00001024 unittest.main()