| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 1 | # Python test set -- part 5, built-in exceptions | 
 | 2 |  | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 3 | import os | 
 | 4 | import sys | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 5 | import unittest | 
| Guido van Rossum | bf12cdb | 2006-08-17 20:24:18 +0000 | [diff] [blame] | 6 | import pickle | 
| Barry Warsaw | 8d109cb | 2008-05-08 04:26:35 +0000 | [diff] [blame] | 7 | import weakref | 
| Antoine Pitrou | a762285 | 2011-09-01 21:37:43 +0200 | [diff] [blame] | 8 | import errno | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 9 |  | 
| Philip Jenvey | b37ac8e | 2012-11-14 14:37:24 -0800 | [diff] [blame] | 10 | from test.support import (TESTFN, captured_output, check_impl_detail, | 
| Philip Jenvey | 8f2b6ad | 2012-11-14 14:51:44 -0800 | [diff] [blame] | 11 |                           check_warnings, cpython_only, gc_collect, run_unittest, | 
 | 12 |                           no_tracing, unlink) | 
| Guido van Rossum | 83b120d | 2001-08-23 03:23:03 +0000 | [diff] [blame] | 13 |  | 
| Richard Oudkerk | 5562d9d | 2012-07-28 17:45:28 +0100 | [diff] [blame] | 14 | class NaiveException(Exception): | 
 | 15 |     def __init__(self, x): | 
 | 16 |         self.x = x | 
 | 17 |  | 
 | 18 | class SlottedNaiveException(Exception): | 
 | 19 |     __slots__ = ('x',) | 
 | 20 |     def __init__(self, x): | 
 | 21 |         self.x = x | 
 | 22 |  | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 23 | # XXX This is not really enough, each *operation* should be tested! | 
 | 24 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 25 | class ExceptionTests(unittest.TestCase): | 
| Barry Warsaw | b9c1d3d | 2001-08-13 23:07:00 +0000 | [diff] [blame] | 26 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 27 |     def raise_catch(self, exc, excname): | 
 | 28 |         try: | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 29 |             raise exc("spam") | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 30 |         except exc as err: | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 31 |             buf1 = str(err) | 
 | 32 |         try: | 
 | 33 |             raise exc("spam") | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 34 |         except exc as err: | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 35 |             buf2 = str(err) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 36 |         self.assertEqual(buf1, buf2) | 
 | 37 |         self.assertEqual(exc.__name__, excname) | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 38 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 39 |     def testRaising(self): | 
 | 40 |         self.raise_catch(AttributeError, "AttributeError") | 
 | 41 |         self.assertRaises(AttributeError, getattr, sys, "undefined_attribute") | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 42 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 43 |         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 Pitrou | 4a90ef0 | 2012-03-03 02:35:32 +0100 | [diff] [blame] | 51 |                 marshal.loads(b'') | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 52 |             except EOFError: | 
 | 53 |                 pass | 
 | 54 |         finally: | 
 | 55 |             sys.stdin = savestdin | 
 | 56 |             fp.close() | 
 | 57 |             unlink(TESTFN) | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 58 |  | 
| Antoine Pitrou | 6b4883d | 2011-10-12 02:54:14 +0200 | [diff] [blame] | 59 |         self.raise_catch(OSError, "OSError") | 
 | 60 |         self.assertRaises(OSError, open, 'this file does not exist', 'r') | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 61 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 62 |         self.raise_catch(ImportError, "ImportError") | 
 | 63 |         self.assertRaises(ImportError, __import__, "undefined_module") | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 64 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 65 |         self.raise_catch(IndexError, "IndexError") | 
 | 66 |         x = [] | 
 | 67 |         self.assertRaises(IndexError, x.__getitem__, 10) | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 68 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 69 |         self.raise_catch(KeyError, "KeyError") | 
 | 70 |         x = {} | 
 | 71 |         self.assertRaises(KeyError, x.__getitem__, 'key') | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 72 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 73 |         self.raise_catch(KeyboardInterrupt, "KeyboardInterrupt") | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 74 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 75 |         self.raise_catch(MemoryError, "MemoryError") | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 76 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 77 |         self.raise_catch(NameError, "NameError") | 
 | 78 |         try: x = undefined_variable | 
 | 79 |         except NameError: pass | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 80 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 81 |         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 Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 85 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 86 |         self.raise_catch(RuntimeError, "RuntimeError") | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 87 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 88 |         self.raise_catch(SyntaxError, "SyntaxError") | 
| Georg Brandl | 7cae87c | 2006-09-06 06:51:57 +0000 | [diff] [blame] | 89 |         try: exec('/\n') | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 90 |         except SyntaxError: pass | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 91 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 92 |         self.raise_catch(IndentationError, "IndentationError") | 
| Fred Drake | 72e48bd | 2000-09-08 16:32:34 +0000 | [diff] [blame] | 93 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 94 |         self.raise_catch(TabError, "TabError") | 
| Georg Brandl | e1b5ac6 | 2008-06-04 13:06:58 +0000 | [diff] [blame] | 95 |         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 Drake | 72e48bd | 2000-09-08 16:32:34 +0000 | [diff] [blame] | 99 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 100 |         self.raise_catch(SystemError, "SystemError") | 
| Fred Drake | 72e48bd | 2000-09-08 16:32:34 +0000 | [diff] [blame] | 101 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 102 |         self.raise_catch(SystemExit, "SystemExit") | 
 | 103 |         self.assertRaises(SystemExit, sys.exit, 0) | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 104 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 105 |         self.raise_catch(TypeError, "TypeError") | 
 | 106 |         try: [] + () | 
 | 107 |         except TypeError: pass | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 108 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 109 |         self.raise_catch(ValueError, "ValueError") | 
| Guido van Rossum | e63bae6 | 2007-07-17 00:34:25 +0000 | [diff] [blame] | 110 |         self.assertRaises(ValueError, chr, 17<<16) | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 111 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 112 |         self.raise_catch(ZeroDivisionError, "ZeroDivisionError") | 
 | 113 |         try: x = 1/0 | 
 | 114 |         except ZeroDivisionError: pass | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 115 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 116 |         self.raise_catch(Exception, "Exception") | 
 | 117 |         try: x = 1/0 | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 118 |         except Exception as e: pass | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 119 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 120 |     def testSyntaxErrorMessage(self): | 
 | 121 |         # make sure the right exception message is raised for each of | 
 | 122 |         # these code fragments | 
| Guido van Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 123 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 124 |         def ckmsg(src, msg): | 
 | 125 |             try: | 
 | 126 |                 compile(src, '<fragment>', 'exec') | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 127 |             except SyntaxError as e: | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 128 |                 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 Rossum | 3bead09 | 1992-01-27 17:00:37 +0000 | [diff] [blame] | 132 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 133 |         s = '''while 1: | 
 | 134 |             try: | 
 | 135 |                 pass | 
 | 136 |             finally: | 
 | 137 |                 continue''' | 
| Barry Warsaw | 992cb8a | 2000-05-25 23:16:54 +0000 | [diff] [blame] | 138 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 139 |         if not sys.platform.startswith('java'): | 
 | 140 |             ckmsg(s, "'continue' not supported inside 'finally' clause") | 
| Jeremy Hylton | ede049b | 2001-09-26 20:01:13 +0000 | [diff] [blame] | 141 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 142 |         s = '''if 1: | 
 | 143 |         try: | 
 | 144 |             continue | 
 | 145 |         except: | 
 | 146 |             pass''' | 
| Jeremy Hylton | ede049b | 2001-09-26 20:01:13 +0000 | [diff] [blame] | 147 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 148 |         ckmsg(s, "'continue' not properly in loop") | 
 | 149 |         ckmsg("continue\n", "'continue' not properly in loop") | 
| Thomas Wouters | 303de6a | 2006-04-20 22:42:37 +0000 | [diff] [blame] | 150 |  | 
| Serhiy Storchaka | 65fd059 | 2014-01-21 22:26:52 +0200 | [diff] [blame] | 151 |     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 Peterson | 17e0bbc | 2010-06-28 15:39:55 +0000 | [diff] [blame] | 164 |     @cpython_only | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 165 |     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 Hylton | ede049b | 2001-09-26 20:01:13 +0000 | [diff] [blame] | 168 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 169 |         class BadException(Exception): | 
 | 170 |             def __init__(self_): | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 171 |                 raise RuntimeError("can't instantiate BadException") | 
| Finn Bock | aa3dc45 | 2001-12-08 10:15:48 +0000 | [diff] [blame] | 172 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 173 |         class InvalidException: | 
 | 174 |             pass | 
| Thomas Wouters | 303de6a | 2006-04-20 22:42:37 +0000 | [diff] [blame] | 175 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 176 |         def test_capi1(): | 
 | 177 |             import _testcapi | 
 | 178 |             try: | 
 | 179 |                 _testcapi.raise_exception(BadException, 1) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 180 |             except TypeError as err: | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 181 |                 exc, err, tb = sys.exc_info() | 
 | 182 |                 co = tb.tb_frame.f_code | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 183 |                 self.assertEqual(co.co_name, "test_capi1") | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 184 |                 self.assertTrue(co.co_filename.endswith('test_exceptions.py')) | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 185 |             else: | 
 | 186 |                 self.fail("Expected exception") | 
| Jeremy Hylton | ede049b | 2001-09-26 20:01:13 +0000 | [diff] [blame] | 187 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 188 |         def test_capi2(): | 
 | 189 |             import _testcapi | 
 | 190 |             try: | 
 | 191 |                 _testcapi.raise_exception(BadException, 0) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 192 |             except RuntimeError as err: | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 193 |                 exc, err, tb = sys.exc_info() | 
 | 194 |                 co = tb.tb_frame.f_code | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 195 |                 self.assertEqual(co.co_name, "__init__") | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 196 |                 self.assertTrue(co.co_filename.endswith('test_exceptions.py')) | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 197 |                 co2 = tb.tb_frame.f_back.f_code | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 198 |                 self.assertEqual(co2.co_name, "test_capi2") | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 199 |             else: | 
 | 200 |                 self.fail("Expected exception") | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 201 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 202 |         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 Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 212 |     def test_WindowsError(self): | 
 | 213 |         try: | 
 | 214 |             WindowsError | 
 | 215 |         except NameError: | 
 | 216 |             pass | 
 | 217 |         else: | 
| Antoine Pitrou | 6b4883d | 2011-10-12 02:54:14 +0200 | [diff] [blame] | 218 |             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 Oudkerk | 3014771 | 2012-08-28 19:33:26 +0100 | [diff] [blame] | 233 |             self.assertEqual(str(w), "[WinError 3] foo: 'bar'") | 
| Antoine Pitrou | 6b4883d | 2011-10-12 02:54:14 +0200 | [diff] [blame] | 234 |             # 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 Oudkerk | 3014771 | 2012-08-28 19:33:26 +0100 | [diff] [blame] | 240 |             self.assertEqual(str(w), "[WinError 1001] foo") | 
| Antoine Pitrou | 6b4883d | 2011-10-12 02:54:14 +0200 | [diff] [blame] | 241 |             # 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 Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 247 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 248 |     def testAttributes(self): | 
 | 249 |         # test that exception attributes are happy | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 250 |  | 
 | 251 |         exceptionList = [ | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 252 |             (BaseException, (), {'args' : ()}), | 
 | 253 |             (BaseException, (1, ), {'args' : (1,)}), | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 254 |             (BaseException, ('foo',), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 255 |                 {'args' : ('foo',)}), | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 256 |             (BaseException, ('foo', 1), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 257 |                 {'args' : ('foo', 1)}), | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 258 |             (SystemExit, ('foo',), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 259 |                 {'args' : ('foo',), 'code' : 'foo'}), | 
| Andrew Svetlov | f7a17b4 | 2012-12-25 16:47:37 +0200 | [diff] [blame] | 260 |             (OSError, ('foo',), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 261 |                 {'args' : ('foo',), 'filename' : None, | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 262 |                  'errno' : None, 'strerror' : None}), | 
| Andrew Svetlov | f7a17b4 | 2012-12-25 16:47:37 +0200 | [diff] [blame] | 263 |             (OSError, ('foo', 'bar'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 264 |                 {'args' : ('foo', 'bar'), 'filename' : None, | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 265 |                  'errno' : 'foo', 'strerror' : 'bar'}), | 
| Andrew Svetlov | f7a17b4 | 2012-12-25 16:47:37 +0200 | [diff] [blame] | 266 |             (OSError, ('foo', 'bar', 'baz'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 267 |                 {'args' : ('foo', 'bar'), 'filename' : 'baz', | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 268 |                  'errno' : 'foo', 'strerror' : 'bar'}), | 
| Larry Hastings | b082731 | 2014-02-09 22:05:19 -0800 | [diff] [blame] | 269 |             (OSError, ('foo', 'bar', 'baz', None, 'quux'), | 
 | 270 |                 {'args' : ('foo', 'bar'), 'filename' : 'baz', 'filename2': 'quux'}), | 
| Andrew Svetlov | 3438fa4 | 2012-12-17 23:35:18 +0200 | [diff] [blame] | 271 |             (OSError, ('errnoStr', 'strErrorStr', 'filenameStr'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 272 |                 {'args' : ('errnoStr', 'strErrorStr'), | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 273 |                  'strerror' : 'strErrorStr', 'errno' : 'errnoStr', | 
 | 274 |                  'filename' : 'filenameStr'}), | 
| Andrew Svetlov | 3438fa4 | 2012-12-17 23:35:18 +0200 | [diff] [blame] | 275 |             (OSError, (1, 'strErrorStr', 'filenameStr'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 276 |                 {'args' : (1, 'strErrorStr'), 'errno' : 1, | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 277 |                  'strerror' : 'strErrorStr', 'filename' : 'filenameStr'}), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 278 |             (SyntaxError, (), {'msg' : None, 'text' : None, | 
| Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 279 |                 'filename' : None, 'lineno' : None, 'offset' : None, | 
 | 280 |                 'print_file_and_line' : None}), | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 281 |             (SyntaxError, ('msgStr',), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 282 |                 {'args' : ('msgStr',), 'text' : None, | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 283 |                  'print_file_and_line' : None, 'msg' : 'msgStr', | 
 | 284 |                  'filename' : None, 'lineno' : None, 'offset' : None}), | 
 | 285 |             (SyntaxError, ('msgStr', ('filenameStr', 'linenoStr', 'offsetStr', | 
 | 286 |                            'textStr')), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 287 |                 {'offset' : 'offsetStr', 'text' : 'textStr', | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 288 |                  '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 Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 294 |                 {'text' : None, | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 295 |                  '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 Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 299 |             (UnicodeError, (), {'args' : (),}), | 
| Walter Dörwald | eceb0fb | 2007-05-24 17:49:56 +0000 | [diff] [blame] | 300 |             (UnicodeEncodeError, ('ascii', 'a', 0, 1, | 
 | 301 |                                   'ordinal not in range'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 302 |                 {'args' : ('ascii', 'a', 0, 1, | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 303 |                                            'ordinal not in range'), | 
| Guido van Rossum | ef87d6e | 2007-05-02 19:09:54 +0000 | [diff] [blame] | 304 |                  'encoding' : 'ascii', 'object' : 'a', | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 305 |                  'start' : 0, 'reason' : 'ordinal not in range'}), | 
| Guido van Rossum | 254348e | 2007-11-21 19:29:53 +0000 | [diff] [blame] | 306 |             (UnicodeDecodeError, ('ascii', bytearray(b'\xff'), 0, 1, | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 307 |                                   'ordinal not in range'), | 
| Guido van Rossum | 254348e | 2007-11-21 19:29:53 +0000 | [diff] [blame] | 308 |                 {'args' : ('ascii', bytearray(b'\xff'), 0, 1, | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 309 |                                            'ordinal not in range'), | 
 | 310 |                  'encoding' : 'ascii', 'object' : b'\xff', | 
 | 311 |                  'start' : 0, 'reason' : 'ordinal not in range'}), | 
| Walter Dörwald | eceb0fb | 2007-05-24 17:49:56 +0000 | [diff] [blame] | 312 |             (UnicodeDecodeError, ('ascii', b'\xff', 0, 1, | 
 | 313 |                                   'ordinal not in range'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 314 |                 {'args' : ('ascii', b'\xff', 0, 1, | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 315 |                                            'ordinal not in range'), | 
| Guido van Rossum | b8142c3 | 2007-05-08 17:49:10 +0000 | [diff] [blame] | 316 |                  'encoding' : 'ascii', 'object' : b'\xff', | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 317 |                  'start' : 0, 'reason' : 'ordinal not in range'}), | 
| Walter Dörwald | eceb0fb | 2007-05-24 17:49:56 +0000 | [diff] [blame] | 318 |             (UnicodeTranslateError, ("\u3042", 0, 1, "ouch"), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 319 |                 {'args' : ('\u3042', 0, 1, 'ouch'), | 
| Guido van Rossum | ef87d6e | 2007-05-02 19:09:54 +0000 | [diff] [blame] | 320 |                  'object' : '\u3042', 'reason' : 'ouch', | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 321 |                  'start' : 0, 'end' : 1}), | 
| Richard Oudkerk | 5562d9d | 2012-07-28 17:45:28 +0100 | [diff] [blame] | 322 |             (NaiveException, ('foo',), | 
 | 323 |                 {'args': ('foo',), 'x': 'foo'}), | 
 | 324 |             (SlottedNaiveException, ('foo',), | 
 | 325 |                 {'args': ('foo',), 'x': 'foo'}), | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 326 |         ] | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 327 |         try: | 
| Antoine Pitrou | 6b4883d | 2011-10-12 02:54:14 +0200 | [diff] [blame] | 328 |             # More tests are in test_WindowsError | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 329 |             exceptionList.append( | 
 | 330 |                 (WindowsError, (1, 'strErrorStr', 'filenameStr'), | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 331 |                     {'args' : (1, 'strErrorStr'), | 
| Antoine Pitrou | 6b4883d | 2011-10-12 02:54:14 +0200 | [diff] [blame] | 332 |                      'strerror' : 'strErrorStr', 'winerror' : None, | 
 | 333 |                      'errno' : 1, 'filename' : 'filenameStr'}) | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 334 |             ) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 335 |         except NameError: | 
 | 336 |             pass | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 337 |  | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 338 |         for exc, args, expected in exceptionList: | 
 | 339 |             try: | 
 | 340 |                 e = exc(*args) | 
 | 341 |             except: | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 342 |                 print("\nexc=%r, args=%r" % (exc, args), file=sys.stderr) | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 343 |                 raise | 
 | 344 |             else: | 
 | 345 |                 # Verify module name | 
| Richard Oudkerk | 5562d9d | 2012-07-28 17:45:28 +0100 | [diff] [blame] | 346 |                 if not type(e).__name__.endswith('NaiveException'): | 
 | 347 |                     self.assertEqual(type(e).__module__, 'builtins') | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 348 |                 # Verify no ref leaks in Exc_str() | 
 | 349 |                 s = str(e) | 
 | 350 |                 for checkArgName in expected: | 
 | 351 |                     value = getattr(e, checkArgName) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 352 |                     self.assertEqual(repr(value), | 
 | 353 |                                      repr(expected[checkArgName]), | 
 | 354 |                                      '%r.%s == %r, expected %r' % ( | 
 | 355 |                                      e, checkArgName, | 
 | 356 |                                      value, expected[checkArgName])) | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 357 |  | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 358 |                 # test for pickling support | 
| Guido van Rossum | 99603b0 | 2007-07-20 00:22:32 +0000 | [diff] [blame] | 359 |                 for p in [pickle]: | 
| Guido van Rossum | ebe3e16 | 2007-05-17 18:20:34 +0000 | [diff] [blame] | 360 |                     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 Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 366 |                             self.assertEqual(got, want, | 
 | 367 |                                              'pickled "%r", attribute "%s' % | 
 | 368 |                                              (e, checkArgName)) | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 369 |  | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 370 |     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 Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 377 |         self.assertIsInstance(e, BaseException) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 378 |         self.assertEqual(e.__traceback__, tb) | 
 | 379 |  | 
 | 380 |         e = IndexError(5).with_traceback(tb) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 381 |         self.assertIsInstance(e, IndexError) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 382 |         self.assertEqual(e.__traceback__, tb) | 
 | 383 |  | 
 | 384 |         class MyException(Exception): | 
 | 385 |             pass | 
 | 386 |  | 
 | 387 |         e = MyException().with_traceback(tb) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 388 |         self.assertIsInstance(e, MyException) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 389 |         self.assertEqual(e.__traceback__, tb) | 
 | 390 |  | 
 | 391 |     def testInvalidTraceback(self): | 
 | 392 |         try: | 
 | 393 |             Exception().__traceback__ = 5 | 
 | 394 |         except TypeError as e: | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 395 |             self.assertIn("__traceback__ must be a traceback", str(e)) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 396 |         else: | 
 | 397 |             self.fail("No exception raised") | 
 | 398 |  | 
| Georg Brandl | ab6f2f6 | 2009-03-31 04:16:10 +0000 | [diff] [blame] | 399 |     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 Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 405 |     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 Coghlan | ab7bf21 | 2012-02-26 17:49:52 +1000 | [diff] [blame] | 418 |         self.assertIsNone(e.__context__) | 
| Benjamin Peterson | d5a1c44 | 2012-05-14 22:09:31 -0700 | [diff] [blame] | 419 |         self.assertIsNone(e.__cause__) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 420 |  | 
 | 421 |         e = TypeError() | 
| Nick Coghlan | ab7bf21 | 2012-02-26 17:49:52 +1000 | [diff] [blame] | 422 |         self.assertIsNone(e.__context__) | 
| Benjamin Peterson | d5a1c44 | 2012-05-14 22:09:31 -0700 | [diff] [blame] | 423 |         self.assertIsNone(e.__cause__) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 424 |  | 
| Andrew Svetlov | 3438fa4 | 2012-12-17 23:35:18 +0200 | [diff] [blame] | 425 |         class MyException(OSError): | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 426 |             pass | 
 | 427 |  | 
 | 428 |         e = MyException() | 
| Nick Coghlan | ab7bf21 | 2012-02-26 17:49:52 +1000 | [diff] [blame] | 429 |         self.assertIsNone(e.__context__) | 
| Benjamin Peterson | d5a1c44 | 2012-05-14 22:09:31 -0700 | [diff] [blame] | 430 |         self.assertIsNone(e.__cause__) | 
| Nick Coghlan | ab7bf21 | 2012-02-26 17:49:52 +1000 | [diff] [blame] | 431 |  | 
 | 432 |     def testChainingDescriptors(self): | 
 | 433 |         try: | 
 | 434 |             raise Exception() | 
 | 435 |         except Exception as exc: | 
 | 436 |             e = exc | 
 | 437 |  | 
 | 438 |         self.assertIsNone(e.__context__) | 
| Benjamin Peterson | d5a1c44 | 2012-05-14 22:09:31 -0700 | [diff] [blame] | 439 |         self.assertIsNone(e.__cause__) | 
 | 440 |         self.assertFalse(e.__suppress_context__) | 
| Nick Coghlan | ab7bf21 | 2012-02-26 17:49:52 +1000 | [diff] [blame] | 441 |  | 
 | 442 |         e.__context__ = NameError() | 
 | 443 |         e.__cause__ = None | 
 | 444 |         self.assertIsInstance(e.__context__, NameError) | 
 | 445 |         self.assertIsNone(e.__cause__) | 
| Benjamin Peterson | d5a1c44 | 2012-05-14 22:09:31 -0700 | [diff] [blame] | 446 |         self.assertTrue(e.__suppress_context__) | 
 | 447 |         e.__suppress_context__ = False | 
 | 448 |         self.assertFalse(e.__suppress_context__) | 
| Collin Winter | 828f04a | 2007-08-31 00:04:24 +0000 | [diff] [blame] | 449 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 450 |     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 Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 461 |         self.assertEqual(x.fancy_arg, 42) | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 462 |  | 
| Brett Cannon | 31f5929 | 2011-02-21 19:29:56 +0000 | [diff] [blame] | 463 |     @no_tracing | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 464 |     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 Melotti | 2f5a78c | 2009-12-24 22:54:06 +0000 | [diff] [blame] | 476 |     def test_str(self): | 
 | 477 |         # Make sure both instances and classes have a str representation. | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 478 |         self.assertTrue(str(Exception)) | 
 | 479 |         self.assertTrue(str(Exception('a'))) | 
| Ezio Melotti | 2f5a78c | 2009-12-24 22:54:06 +0000 | [diff] [blame] | 480 |         self.assertTrue(str(Exception('a', 'b'))) | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 481 |  | 
| Barry Warsaw | 8d109cb | 2008-05-08 04:26:35 +0000 | [diff] [blame] | 482 |     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 Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 485 |         try: | 
 | 486 |             raise Exception() | 
 | 487 |         except Exception as e: | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 488 |             self.assertTrue(e) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 489 |             del e | 
| Ezio Melotti | b58e0bd | 2010-01-23 15:40:09 +0000 | [diff] [blame] | 490 |         self.assertNotIn('e', locals()) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 491 |  | 
| Barry Warsaw | 8d109cb | 2008-05-08 04:26:35 +0000 | [diff] [blame] | 492 |     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 Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 507 |         # Qualified "except" with "as" | 
| Barry Warsaw | 8d109cb | 2008-05-08 04:26:35 +0000 | [diff] [blame] | 508 |         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 Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 516 |         self.assertTrue(obj is None, "%s" % obj) | 
| Barry Warsaw | 8d109cb | 2008-05-08 04:26:35 +0000 | [diff] [blame] | 517 |  | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 518 |         # 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 Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 527 |         self.assertTrue(obj is None, "%s" % obj) | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 528 |  | 
 | 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 Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 538 |         self.assertTrue(obj is None, "%s" % obj) | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 539 |  | 
 | 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 Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 550 |         self.assertTrue(obj is None, "%s" % obj) | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 551 |  | 
 | 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 Rossum | b4fb6e4 | 2008-06-14 20:20:24 +0000 | [diff] [blame] | 560 |         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 Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 566 |             obj = None | 
 | 567 |             obj = wr() | 
| Philip Jenvey | b37ac8e | 2012-11-14 14:37:24 -0800 | [diff] [blame] | 568 |             # guarantee no ref cycles on CPython (don't gc_collect) | 
 | 569 |             if check_impl_detail(cpython=False): | 
 | 570 |                 gc_collect() | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 571 |             self.assertTrue(obj is None, "%s" % obj) | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 572 |  | 
 | 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 Jenvey | b37ac8e | 2012-11-14 14:37:24 -0800 | [diff] [blame] | 587 |         if check_impl_detail(cpython=False): | 
 | 588 |             gc_collect() | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 589 |         obj = wr() | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 590 |         self.assertTrue(obj is None, "%s" % obj) | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 591 |  | 
 | 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 Jenvey | b37ac8e | 2012-11-14 14:37:24 -0800 | [diff] [blame] | 603 |         if check_impl_detail(cpython=False): | 
 | 604 |             gc_collect() | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 605 |         obj = wr() | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 606 |         self.assertTrue(obj is None, "%s" % obj) | 
| Benjamin Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 607 |  | 
| Amaury Forgeot d'Arc | ba117ef | 2010-09-10 21:39:53 +0000 | [diff] [blame] | 608 |     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 Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 619 |     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 Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 630 |         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 Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 635 |  | 
 | 636 |         # Same test, but inside an exception handler | 
 | 637 |         try: | 
 | 638 |             raise TypeError("foo") | 
 | 639 |         except TypeError: | 
 | 640 |             g = yield_raise() | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 641 |             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 Peterson | eec3d71 | 2008-06-11 15:59:43 +0000 | [diff] [blame] | 646 |             del g | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 647 |             self.assertEqual(sys.exc_info()[0], TypeError) | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 648 |  | 
| Benjamin Peterson | 83195c3 | 2011-07-03 13:44:00 -0500 | [diff] [blame] | 649 |     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 Pitrou | c4c19b3 | 2015-03-18 22:22:46 +0100 | [diff] [blame] | 664 |     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 Peterson | ac91341 | 2011-07-03 16:25:11 -0500 | [diff] [blame] | 710 |     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 Peterson | ae5f2f4 | 2010-03-07 17:10:51 +0000 | [diff] [blame] | 722 |     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 Pitrou | a370fcf | 2011-08-20 14:15:03 +0200 | [diff] [blame] | 736 |     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 Peterson | 27d6367 | 2008-06-15 20:09:12 +0000 | [diff] [blame] | 798 |     def test_3114(self): | 
 | 799 |         # Bug #3114: in its destructor, MyObject retrieves a pointer to | 
 | 800 |         # obsolete and/or deallocated objects. | 
| Benjamin Peterson | 979f311 | 2008-06-15 00:05:44 +0000 | [diff] [blame] | 801 |         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 Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 810 |         self.assertEqual(e, (None, None, None)) | 
| Benjamin Peterson | 979f311 | 2008-06-15 00:05:44 +0000 | [diff] [blame] | 811 |  | 
| Benjamin Peterson | 24dfb05 | 2014-04-02 12:05:35 -0400 | [diff] [blame] | 812 |     def test_unicode_change_attributes(self): | 
| Eric Smith | 0facd77 | 2010-02-24 15:42:29 +0000 | [diff] [blame] | 813 |         # 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 Peterson | 6e7740c | 2008-08-20 23:23:34 +0000 | [diff] [blame] | 848 |  | 
| Benjamin Peterson | 9b09ba1 | 2014-04-02 12:15:06 -0400 | [diff] [blame] | 849 |     def test_unicode_errors_no_object(self): | 
 | 850 |         # See issue #21134. | 
| Benjamin Peterson | e331121 | 2014-04-02 15:51:38 -0400 | [diff] [blame] | 851 |         klasses = UnicodeEncodeError, UnicodeDecodeError, UnicodeTranslateError | 
| Benjamin Peterson | 9b09ba1 | 2014-04-02 12:15:06 -0400 | [diff] [blame] | 852 |         for klass in klasses: | 
 | 853 |             self.assertEqual(str(klass.__new__(klass)), "") | 
 | 854 |  | 
| Brett Cannon | 31f5929 | 2011-02-21 19:29:56 +0000 | [diff] [blame] | 855 |     @no_tracing | 
| Benjamin Peterson | 69c88f7 | 2008-07-31 01:47:08 +0000 | [diff] [blame] | 856 |     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 Pitrou | ec569b7 | 2008-08-26 22:40:48 +0000 | [diff] [blame] | 873 |                 self.fail("Should have raised KeyError") | 
| Benjamin Peterson | 69c88f7 | 2008-07-31 01:47:08 +0000 | [diff] [blame] | 874 |             else: | 
| Antoine Pitrou | ec569b7 | 2008-08-26 22:40:48 +0000 | [diff] [blame] | 875 |                 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 Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 883 |         self.assertTrue(isinstance(v, RuntimeError), type(v)) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 884 |         self.assertIn("maximum recursion depth exceeded", str(v)) | 
| Benjamin Peterson | 69c88f7 | 2008-07-31 01:47:08 +0000 | [diff] [blame] | 885 |  | 
| Amaury Forgeot d'Arc | e19cadb | 2008-07-31 22:56:02 +0000 | [diff] [blame] | 886 |  | 
| Serhiy Storchaka | 5cfc79d | 2014-02-07 10:06:39 +0200 | [diff] [blame] | 887 |     @cpython_only | 
| Amaury Forgeot d'Arc | e19cadb | 2008-07-31 22:56:02 +0000 | [diff] [blame] | 888 |     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 Peterson | 0067bd6 | 2008-08-16 16:11:03 +0000 | [diff] [blame] | 892 |         from _testcapi import raise_memoryerror | 
| Amaury Forgeot d'Arc | e19cadb | 2008-07-31 22:56:02 +0000 | [diff] [blame] | 893 |         def raiseMemError(): | 
 | 894 |             try: | 
| Benjamin Peterson | 0067bd6 | 2008-08-16 16:11:03 +0000 | [diff] [blame] | 895 |                 raise_memoryerror() | 
| Amaury Forgeot d'Arc | e19cadb | 2008-07-31 22:56:02 +0000 | [diff] [blame] | 896 |             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 Peterson | 17e0bbc | 2010-06-28 15:39:55 +0000 | [diff] [blame] | 906 |     @cpython_only | 
| Georg Brandl | 1e28a27 | 2009-12-28 08:41:01 +0000 | [diff] [blame] | 907 |     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 Storchaka | 5cfc79d | 2014-02-07 10:06:39 +0200 | [diff] [blame] | 946 |     @cpython_only | 
| Antoine Pitrou | 07e20ef | 2010-10-28 22:56:58 +0000 | [diff] [blame] | 947 |     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 Cannon | 31f5929 | 2011-02-21 19:29:56 +0000 | [diff] [blame] | 968 |     @no_tracing | 
| Antoine Pitrou | 07e20ef | 2010-10-28 22:56:58 +0000 | [diff] [blame] | 969 |     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 Brandl | 1e28a27 | 2009-12-28 08:41:01 +0000 | [diff] [blame] | 987 |  | 
| Antoine Pitrou | a762285 | 2011-09-01 21:37:43 +0200 | [diff] [blame] | 988 |     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 Cannon | 79ec55e | 2012-04-12 20:24:54 -0400 | [diff] [blame] | 995 | class 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 Cannon | 07c6e71 | 2012-08-24 13:05:09 -0400 | [diff] [blame] | 1015 |     def test_non_str_argument(self): | 
 | 1016 |         # Issue #15778 | 
| Nadeem Vawda | 6d70870 | 2012-10-14 01:42:32 +0200 | [diff] [blame] | 1017 |         with check_warnings(('', BytesWarning), quiet=True): | 
 | 1018 |             arg = b'abc' | 
 | 1019 |             exc = ImportError(arg) | 
 | 1020 |             self.assertEqual(str(arg), str(exc)) | 
| Brett Cannon | 79ec55e | 2012-04-12 20:24:54 -0400 | [diff] [blame] | 1021 |  | 
 | 1022 |  | 
| Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 1023 | if __name__ == '__main__': | 
| Guido van Rossum | b8142c3 | 2007-05-08 17:49:10 +0000 | [diff] [blame] | 1024 |     unittest.main() |