blob: 85b5cad239d5317e370ca383f24f25d372108382 [file] [log] [blame]
Christian Heimes33fe8092008-04-13 13:53:33 +00001from contextlib import contextmanager
Christian Heimes5fb7c2a2007-12-24 08:52:31 +00002import linecache
Raymond Hettingerdc9dcf12003-07-13 06:15:11 +00003import os
Christian Heimes5fb7c2a2007-12-24 08:52:31 +00004from io import StringIO
Guido van Rossum61e21b52007-08-20 19:06:03 +00005import sys
Raymond Hettingerd6f6e502003-07-13 08:37:40 +00006import unittest
Victor Stinner148051a2010-05-20 21:00:34 +00007import tempfile
8import subprocess
Benjamin Petersonee8712c2008-05-20 21:35:26 +00009from test import support
Jeremy Hylton85014662003-07-11 15:37:59 +000010
Guido van Rossum805365e2007-05-07 22:24:25 +000011from test import warning_tests
Jeremy Hylton85014662003-07-11 15:37:59 +000012
Christian Heimes33fe8092008-04-13 13:53:33 +000013import warnings as original_warnings
Jeremy Hylton85014662003-07-11 15:37:59 +000014
Nick Coghlan47384702009-04-22 16:13:36 +000015py_warnings = support.import_fresh_module('warnings', blocked=['_warnings'])
16c_warnings = support.import_fresh_module('warnings', fresh=['_warnings'])
Christian Heimes33fe8092008-04-13 13:53:33 +000017
18@contextmanager
19def warnings_state(module):
20 """Use a specific warnings implementation in warning_tests."""
21 global __warningregistry__
22 for to_clear in (sys, warning_tests):
23 try:
24 to_clear.__warningregistry__.clear()
25 except AttributeError:
26 pass
27 try:
28 __warningregistry__.clear()
29 except NameError:
30 pass
31 original_warnings = warning_tests.warnings
Florent Xicluna9b0e9182010-03-28 11:42:38 +000032 original_filters = module.filters
Christian Heimes33fe8092008-04-13 13:53:33 +000033 try:
Florent Xicluna9b0e9182010-03-28 11:42:38 +000034 module.filters = original_filters[:]
35 module.simplefilter("once")
Christian Heimes33fe8092008-04-13 13:53:33 +000036 warning_tests.warnings = module
37 yield
38 finally:
39 warning_tests.warnings = original_warnings
Florent Xicluna9b0e9182010-03-28 11:42:38 +000040 module.filters = original_filters
Christian Heimes33fe8092008-04-13 13:53:33 +000041
42
43class BaseTest(unittest.TestCase):
44
45 """Basic bookkeeping required for testing."""
46
47 def setUp(self):
48 # The __warningregistry__ needs to be in a pristine state for tests
49 # to work properly.
50 if '__warningregistry__' in globals():
51 del globals()['__warningregistry__']
52 if hasattr(warning_tests, '__warningregistry__'):
53 del warning_tests.__warningregistry__
54 if hasattr(sys, '__warningregistry__'):
55 del sys.__warningregistry__
56 # The 'warnings' module must be explicitly set so that the proper
57 # interaction between _warnings and 'warnings' can be controlled.
58 sys.modules['warnings'] = self.module
59 super(BaseTest, self).setUp()
60
61 def tearDown(self):
62 sys.modules['warnings'] = original_warnings
63 super(BaseTest, self).tearDown()
64
65
66class FilterTests(object):
67
68 """Testing the filtering functionality."""
69
70 def test_error(self):
Brett Cannon1cd02472008-09-09 01:52:27 +000071 with original_warnings.catch_warnings(module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +000072 self.module.resetwarnings()
73 self.module.filterwarnings("error", category=UserWarning)
74 self.assertRaises(UserWarning, self.module.warn,
75 "FilterTests.test_error")
76
77 def test_ignore(self):
Brett Cannon1cd02472008-09-09 01:52:27 +000078 with original_warnings.catch_warnings(record=True,
79 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +000080 self.module.resetwarnings()
81 self.module.filterwarnings("ignore", category=UserWarning)
82 self.module.warn("FilterTests.test_ignore", UserWarning)
Brett Cannonec92e182008-09-02 02:46:59 +000083 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +000084
85 def test_always(self):
Brett Cannon1cd02472008-09-09 01:52:27 +000086 with original_warnings.catch_warnings(record=True,
87 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +000088 self.module.resetwarnings()
89 self.module.filterwarnings("always", category=UserWarning)
90 message = "FilterTests.test_always"
91 self.module.warn(message, UserWarning)
Georg Brandlab91fde2009-08-13 08:51:18 +000092 self.assertTrue(message, w[-1].message)
Christian Heimes33fe8092008-04-13 13:53:33 +000093 self.module.warn(message, UserWarning)
Georg Brandlab91fde2009-08-13 08:51:18 +000094 self.assertTrue(w[-1].message, message)
Christian Heimes33fe8092008-04-13 13:53:33 +000095
96 def test_default(self):
Brett Cannon1cd02472008-09-09 01:52:27 +000097 with original_warnings.catch_warnings(record=True,
98 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +000099 self.module.resetwarnings()
100 self.module.filterwarnings("default", category=UserWarning)
101 message = UserWarning("FilterTests.test_default")
102 for x in range(2):
103 self.module.warn(message, UserWarning)
104 if x == 0:
Brett Cannon1cd02472008-09-09 01:52:27 +0000105 self.assertEquals(w[-1].message, message)
106 del w[:]
Christian Heimes33fe8092008-04-13 13:53:33 +0000107 elif x == 1:
Brett Cannon1cd02472008-09-09 01:52:27 +0000108 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000109 else:
110 raise ValueError("loop variant unhandled")
111
112 def test_module(self):
Brett Cannon1cd02472008-09-09 01:52:27 +0000113 with original_warnings.catch_warnings(record=True,
114 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000115 self.module.resetwarnings()
116 self.module.filterwarnings("module", category=UserWarning)
117 message = UserWarning("FilterTests.test_module")
118 self.module.warn(message, UserWarning)
Brett Cannon1cd02472008-09-09 01:52:27 +0000119 self.assertEquals(w[-1].message, message)
120 del w[:]
Christian Heimes33fe8092008-04-13 13:53:33 +0000121 self.module.warn(message, UserWarning)
Brett Cannon1cd02472008-09-09 01:52:27 +0000122 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000123
124 def test_once(self):
Brett Cannon1cd02472008-09-09 01:52:27 +0000125 with original_warnings.catch_warnings(record=True,
126 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000127 self.module.resetwarnings()
128 self.module.filterwarnings("once", category=UserWarning)
129 message = UserWarning("FilterTests.test_once")
130 self.module.warn_explicit(message, UserWarning, "test_warnings.py",
131 42)
Brett Cannon1cd02472008-09-09 01:52:27 +0000132 self.assertEquals(w[-1].message, message)
133 del w[:]
Christian Heimes33fe8092008-04-13 13:53:33 +0000134 self.module.warn_explicit(message, UserWarning, "test_warnings.py",
135 13)
Brett Cannonec92e182008-09-02 02:46:59 +0000136 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000137 self.module.warn_explicit(message, UserWarning, "test_warnings2.py",
138 42)
Brett Cannonec92e182008-09-02 02:46:59 +0000139 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000140
141 def test_inheritance(self):
Brett Cannon1cd02472008-09-09 01:52:27 +0000142 with original_warnings.catch_warnings(module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000143 self.module.resetwarnings()
144 self.module.filterwarnings("error", category=Warning)
145 self.assertRaises(UserWarning, self.module.warn,
146 "FilterTests.test_inheritance", UserWarning)
147
148 def test_ordering(self):
Brett Cannon1cd02472008-09-09 01:52:27 +0000149 with original_warnings.catch_warnings(record=True,
150 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000151 self.module.resetwarnings()
152 self.module.filterwarnings("ignore", category=UserWarning)
153 self.module.filterwarnings("error", category=UserWarning,
154 append=True)
Brett Cannon1cd02472008-09-09 01:52:27 +0000155 del w[:]
Christian Heimes33fe8092008-04-13 13:53:33 +0000156 try:
157 self.module.warn("FilterTests.test_ordering", UserWarning)
158 except UserWarning:
159 self.fail("order handling for actions failed")
Brett Cannonec92e182008-09-02 02:46:59 +0000160 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000161
162 def test_filterwarnings(self):
163 # Test filterwarnings().
164 # Implicitly also tests resetwarnings().
Brett Cannon1cd02472008-09-09 01:52:27 +0000165 with original_warnings.catch_warnings(record=True,
166 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000167 self.module.filterwarnings("error", "", Warning, "", 0)
168 self.assertRaises(UserWarning, self.module.warn, 'convert to error')
169
170 self.module.resetwarnings()
171 text = 'handle normally'
172 self.module.warn(text)
Brett Cannon1cd02472008-09-09 01:52:27 +0000173 self.assertEqual(str(w[-1].message), text)
Georg Brandlab91fde2009-08-13 08:51:18 +0000174 self.assertTrue(w[-1].category is UserWarning)
Christian Heimes33fe8092008-04-13 13:53:33 +0000175
176 self.module.filterwarnings("ignore", "", Warning, "", 0)
177 text = 'filtered out'
178 self.module.warn(text)
Brett Cannon1cd02472008-09-09 01:52:27 +0000179 self.assertNotEqual(str(w[-1].message), text)
Christian Heimes33fe8092008-04-13 13:53:33 +0000180
181 self.module.resetwarnings()
182 self.module.filterwarnings("error", "hex*", Warning, "", 0)
183 self.assertRaises(UserWarning, self.module.warn, 'hex/oct')
184 text = 'nonmatching text'
185 self.module.warn(text)
Brett Cannon1cd02472008-09-09 01:52:27 +0000186 self.assertEqual(str(w[-1].message), text)
Georg Brandlab91fde2009-08-13 08:51:18 +0000187 self.assertTrue(w[-1].category is UserWarning)
Christian Heimes33fe8092008-04-13 13:53:33 +0000188
189class CFilterTests(BaseTest, FilterTests):
190 module = c_warnings
191
192class PyFilterTests(BaseTest, FilterTests):
193 module = py_warnings
194
195
196class WarnTests(unittest.TestCase):
197
198 """Test warnings.warn() and warnings.warn_explicit()."""
199
200 def test_message(self):
Brett Cannon1cd02472008-09-09 01:52:27 +0000201 with original_warnings.catch_warnings(record=True,
202 module=self.module) as w:
Florent Xicluna9b0e9182010-03-28 11:42:38 +0000203 self.module.simplefilter("once")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000204 for i in range(4):
Christian Heimes33fe8092008-04-13 13:53:33 +0000205 text = 'multi %d' %i # Different text on each call.
206 self.module.warn(text)
Brett Cannon1cd02472008-09-09 01:52:27 +0000207 self.assertEqual(str(w[-1].message), text)
Georg Brandlab91fde2009-08-13 08:51:18 +0000208 self.assertTrue(w[-1].category is UserWarning)
Raymond Hettingerd6f6e502003-07-13 08:37:40 +0000209
Brett Cannon54bd41d2008-09-02 04:01:42 +0000210 # Issue 3639
211 def test_warn_nonstandard_types(self):
212 # warn() should handle non-standard types without issue.
213 for ob in (Warning, None, 42):
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000214 with original_warnings.catch_warnings(record=True,
215 module=self.module) as w:
Florent Xicluna9b0e9182010-03-28 11:42:38 +0000216 self.module.simplefilter("once")
Brett Cannon54bd41d2008-09-02 04:01:42 +0000217 self.module.warn(ob)
218 # Don't directly compare objects since
219 # ``Warning() != Warning()``.
Brett Cannon1cd02472008-09-09 01:52:27 +0000220 self.assertEquals(str(w[-1].message), str(UserWarning(ob)))
Brett Cannon54bd41d2008-09-02 04:01:42 +0000221
Guido van Rossumd8faa362007-04-27 19:54:29 +0000222 def test_filename(self):
Christian Heimes33fe8092008-04-13 13:53:33 +0000223 with warnings_state(self.module):
Brett Cannon1cd02472008-09-09 01:52:27 +0000224 with original_warnings.catch_warnings(record=True,
225 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000226 warning_tests.inner("spam1")
Brett Cannon1cd02472008-09-09 01:52:27 +0000227 self.assertEqual(os.path.basename(w[-1].filename),
228 "warning_tests.py")
Christian Heimes33fe8092008-04-13 13:53:33 +0000229 warning_tests.outer("spam2")
Brett Cannon1cd02472008-09-09 01:52:27 +0000230 self.assertEqual(os.path.basename(w[-1].filename),
231 "warning_tests.py")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000232
233 def test_stacklevel(self):
234 # Test stacklevel argument
235 # make sure all messages are different, so the warning won't be skipped
Christian Heimes33fe8092008-04-13 13:53:33 +0000236 with warnings_state(self.module):
Brett Cannon1cd02472008-09-09 01:52:27 +0000237 with original_warnings.catch_warnings(record=True,
238 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000239 warning_tests.inner("spam3", stacklevel=1)
Brett Cannon1cd02472008-09-09 01:52:27 +0000240 self.assertEqual(os.path.basename(w[-1].filename),
241 "warning_tests.py")
Christian Heimes33fe8092008-04-13 13:53:33 +0000242 warning_tests.outer("spam4", stacklevel=1)
Brett Cannon1cd02472008-09-09 01:52:27 +0000243 self.assertEqual(os.path.basename(w[-1].filename),
244 "warning_tests.py")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000245
Christian Heimes33fe8092008-04-13 13:53:33 +0000246 warning_tests.inner("spam5", stacklevel=2)
Brett Cannon1cd02472008-09-09 01:52:27 +0000247 self.assertEqual(os.path.basename(w[-1].filename),
248 "test_warnings.py")
Christian Heimes33fe8092008-04-13 13:53:33 +0000249 warning_tests.outer("spam6", stacklevel=2)
Brett Cannon1cd02472008-09-09 01:52:27 +0000250 self.assertEqual(os.path.basename(w[-1].filename),
251 "warning_tests.py")
Christian Heimes5d8da202008-05-06 13:58:24 +0000252 warning_tests.outer("spam6.5", stacklevel=3)
Brett Cannon1cd02472008-09-09 01:52:27 +0000253 self.assertEqual(os.path.basename(w[-1].filename),
254 "test_warnings.py")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000255
Christian Heimes33fe8092008-04-13 13:53:33 +0000256 warning_tests.inner("spam7", stacklevel=9999)
Brett Cannon1cd02472008-09-09 01:52:27 +0000257 self.assertEqual(os.path.basename(w[-1].filename),
258 "sys")
Christian Heimes33fe8092008-04-13 13:53:33 +0000259
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000260 def test_missing_filename_not_main(self):
261 # If __file__ is not specified and __main__ is not the module name,
262 # then __file__ should be set to the module name.
263 filename = warning_tests.__file__
264 try:
265 del warning_tests.__file__
266 with warnings_state(self.module):
Brett Cannon1cd02472008-09-09 01:52:27 +0000267 with original_warnings.catch_warnings(record=True,
268 module=self.module) as w:
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000269 warning_tests.inner("spam8", stacklevel=1)
Brett Cannon1cd02472008-09-09 01:52:27 +0000270 self.assertEqual(w[-1].filename, warning_tests.__name__)
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000271 finally:
272 warning_tests.__file__ = filename
273
274 def test_missing_filename_main_with_argv(self):
275 # If __file__ is not specified and the caller is __main__ and sys.argv
276 # exists, then use sys.argv[0] as the file.
277 if not hasattr(sys, 'argv'):
278 return
279 filename = warning_tests.__file__
280 module_name = warning_tests.__name__
281 try:
282 del warning_tests.__file__
283 warning_tests.__name__ = '__main__'
284 with warnings_state(self.module):
Brett Cannon1cd02472008-09-09 01:52:27 +0000285 with original_warnings.catch_warnings(record=True,
286 module=self.module) as w:
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000287 warning_tests.inner('spam9', stacklevel=1)
Brett Cannon1cd02472008-09-09 01:52:27 +0000288 self.assertEqual(w[-1].filename, sys.argv[0])
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000289 finally:
290 warning_tests.__file__ = filename
291 warning_tests.__name__ = module_name
292
293 def test_missing_filename_main_without_argv(self):
294 # If __file__ is not specified, the caller is __main__, and sys.argv
295 # is not set, then '__main__' is the file name.
296 filename = warning_tests.__file__
297 module_name = warning_tests.__name__
298 argv = sys.argv
299 try:
300 del warning_tests.__file__
301 warning_tests.__name__ = '__main__'
302 del sys.argv
303 with warnings_state(self.module):
Brett Cannon1cd02472008-09-09 01:52:27 +0000304 with original_warnings.catch_warnings(record=True,
305 module=self.module) as w:
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000306 warning_tests.inner('spam10', stacklevel=1)
Brett Cannon1cd02472008-09-09 01:52:27 +0000307 self.assertEqual(w[-1].filename, '__main__')
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000308 finally:
309 warning_tests.__file__ = filename
310 warning_tests.__name__ = module_name
311 sys.argv = argv
312
Christian Heimesdaaf8ee2008-05-04 23:58:41 +0000313 def test_missing_filename_main_with_argv_empty_string(self):
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000314 # If __file__ is not specified, the caller is __main__, and sys.argv[0]
315 # is the empty string, then '__main__ is the file name.
316 # Tests issue 2743.
317 file_name = warning_tests.__file__
318 module_name = warning_tests.__name__
319 argv = sys.argv
320 try:
321 del warning_tests.__file__
322 warning_tests.__name__ = '__main__'
323 sys.argv = ['']
324 with warnings_state(self.module):
Brett Cannon1cd02472008-09-09 01:52:27 +0000325 with original_warnings.catch_warnings(record=True,
326 module=self.module) as w:
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000327 warning_tests.inner('spam11', stacklevel=1)
Brett Cannon1cd02472008-09-09 01:52:27 +0000328 self.assertEqual(w[-1].filename, '__main__')
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000329 finally:
330 warning_tests.__file__ = file_name
331 warning_tests.__name__ = module_name
332 sys.argv = argv
333
Brett Cannondb734912008-06-27 00:52:15 +0000334 def test_warn_explicit_type_errors(self):
335 # warn_explicit() shoud error out gracefully if it is given objects
336 # of the wrong types.
337 # lineno is expected to be an integer.
338 self.assertRaises(TypeError, self.module.warn_explicit,
339 None, UserWarning, None, None)
340 # Either 'message' needs to be an instance of Warning or 'category'
341 # needs to be a subclass.
342 self.assertRaises(TypeError, self.module.warn_explicit,
343 None, None, None, 1)
344 # 'registry' must be a dict or None.
345 self.assertRaises((TypeError, AttributeError),
346 self.module.warn_explicit,
347 None, Warning, None, 1, registry=42)
348
Hirokazu Yamamoto680ac292009-07-17 07:02:18 +0000349 def test_bad_str(self):
350 # issue 6415
351 # Warnings instance with a bad format string for __str__ should not
352 # trigger a bus error.
353 class BadStrWarning(Warning):
354 """Warning with a bad format string for __str__."""
355 def __str__(self):
356 return ("A bad formatted string %(err)" %
357 {"err" : "there is no %(err)s"})
358
359 with self.assertRaises(ValueError):
360 self.module.warn(BadStrWarning())
361
362
Christian Heimes33fe8092008-04-13 13:53:33 +0000363class CWarnTests(BaseTest, WarnTests):
364 module = c_warnings
365
Nick Coghlanfce769e2009-04-11 14:30:59 +0000366 # As an early adopter, we sanity check the
367 # test.support.import_fresh_module utility function
368 def test_accelerated(self):
369 self.assertFalse(original_warnings is self.module)
370 self.assertFalse(hasattr(self.module.warn, '__code__'))
371
Christian Heimes33fe8092008-04-13 13:53:33 +0000372class PyWarnTests(BaseTest, WarnTests):
373 module = py_warnings
374
Nick Coghlanfce769e2009-04-11 14:30:59 +0000375 # As an early adopter, we sanity check the
376 # test.support.import_fresh_module utility function
377 def test_pure_python(self):
378 self.assertFalse(original_warnings is self.module)
379 self.assertTrue(hasattr(self.module.warn, '__code__'))
380
Christian Heimes33fe8092008-04-13 13:53:33 +0000381
382class WCmdLineTests(unittest.TestCase):
383
384 def test_improper_input(self):
385 # Uses the private _setoption() function to test the parsing
386 # of command-line warning arguments
Brett Cannon1cd02472008-09-09 01:52:27 +0000387 with original_warnings.catch_warnings(module=self.module):
Christian Heimes33fe8092008-04-13 13:53:33 +0000388 self.assertRaises(self.module._OptionError,
389 self.module._setoption, '1:2:3:4:5:6')
390 self.assertRaises(self.module._OptionError,
391 self.module._setoption, 'bogus::Warning')
392 self.assertRaises(self.module._OptionError,
393 self.module._setoption, 'ignore:2::4:-5')
394 self.module._setoption('error::Warning::0')
395 self.assertRaises(UserWarning, self.module.warn, 'convert to error')
396
397class CWCmdLineTests(BaseTest, WCmdLineTests):
398 module = c_warnings
399
400class PyWCmdLineTests(BaseTest, WCmdLineTests):
401 module = py_warnings
402
403
404class _WarningsTests(BaseTest):
405
406 """Tests specific to the _warnings module."""
407
408 module = c_warnings
409
410 def test_filter(self):
411 # Everything should function even if 'filters' is not in warnings.
Brett Cannon1cd02472008-09-09 01:52:27 +0000412 with original_warnings.catch_warnings(module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000413 self.module.filterwarnings("error", "", Warning, "", 0)
414 self.assertRaises(UserWarning, self.module.warn,
415 'convert to error')
416 del self.module.filters
417 self.assertRaises(UserWarning, self.module.warn,
418 'convert to error')
419
420 def test_onceregistry(self):
421 # Replacing or removing the onceregistry should be okay.
422 global __warningregistry__
423 message = UserWarning('onceregistry test')
424 try:
425 original_registry = self.module.onceregistry
426 __warningregistry__ = {}
Brett Cannon1cd02472008-09-09 01:52:27 +0000427 with original_warnings.catch_warnings(record=True,
428 module=self.module) as w:
Christian Heimes33fe8092008-04-13 13:53:33 +0000429 self.module.resetwarnings()
430 self.module.filterwarnings("once", category=UserWarning)
431 self.module.warn_explicit(message, UserWarning, "file", 42)
Georg Brandlab91fde2009-08-13 08:51:18 +0000432 self.assertEqual(w[-1].message, message)
Brett Cannon1cd02472008-09-09 01:52:27 +0000433 del w[:]
Christian Heimes33fe8092008-04-13 13:53:33 +0000434 self.module.warn_explicit(message, UserWarning, "file", 42)
Brett Cannonec92e182008-09-02 02:46:59 +0000435 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000436 # Test the resetting of onceregistry.
437 self.module.onceregistry = {}
438 __warningregistry__ = {}
439 self.module.warn('onceregistry test')
Georg Brandlab91fde2009-08-13 08:51:18 +0000440 self.assertEqual(w[-1].message.args, message.args)
Christian Heimes33fe8092008-04-13 13:53:33 +0000441 # Removal of onceregistry is okay.
Brett Cannon1cd02472008-09-09 01:52:27 +0000442 del w[:]
Christian Heimes33fe8092008-04-13 13:53:33 +0000443 del self.module.onceregistry
444 __warningregistry__ = {}
445 self.module.warn_explicit(message, UserWarning, "file", 42)
Brett Cannonec92e182008-09-02 02:46:59 +0000446 self.assertEquals(len(w), 0)
Christian Heimes33fe8092008-04-13 13:53:33 +0000447 finally:
448 self.module.onceregistry = original_registry
449
Brett Cannon0759dd62009-04-01 18:13:07 +0000450 def test_default_action(self):
451 # Replacing or removing defaultaction should be okay.
452 message = UserWarning("defaultaction test")
453 original = self.module.defaultaction
454 try:
455 with original_warnings.catch_warnings(record=True,
456 module=self.module) as w:
457 self.module.resetwarnings()
458 registry = {}
459 self.module.warn_explicit(message, UserWarning, "<test>", 42,
460 registry=registry)
461 self.assertEqual(w[-1].message, message)
462 self.assertEqual(len(w), 1)
463 self.assertEqual(len(registry), 1)
464 del w[:]
465 # Test removal.
466 del self.module.defaultaction
467 __warningregistry__ = {}
468 registry = {}
469 self.module.warn_explicit(message, UserWarning, "<test>", 43,
470 registry=registry)
471 self.assertEqual(w[-1].message, message)
472 self.assertEqual(len(w), 1)
473 self.assertEqual(len(registry), 1)
474 del w[:]
475 # Test setting.
476 self.module.defaultaction = "ignore"
477 __warningregistry__ = {}
478 registry = {}
479 self.module.warn_explicit(message, UserWarning, "<test>", 44,
480 registry=registry)
481 self.assertEqual(len(w), 0)
482 finally:
483 self.module.defaultaction = original
484
Christian Heimes33fe8092008-04-13 13:53:33 +0000485 def test_showwarning_missing(self):
486 # Test that showwarning() missing is okay.
487 text = 'del showwarning test'
Brett Cannon1cd02472008-09-09 01:52:27 +0000488 with original_warnings.catch_warnings(module=self.module):
Christian Heimes33fe8092008-04-13 13:53:33 +0000489 self.module.filterwarnings("always", category=UserWarning)
490 del self.module.showwarning
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000491 with support.captured_output('stderr') as stream:
Christian Heimes33fe8092008-04-13 13:53:33 +0000492 self.module.warn(text)
493 result = stream.getvalue()
Georg Brandlab91fde2009-08-13 08:51:18 +0000494 self.assertTrue(text in result)
Christian Heimes33fe8092008-04-13 13:53:33 +0000495
Christian Heimes8dc226f2008-05-06 23:45:46 +0000496 def test_showwarning_not_callable(self):
Brett Cannonfcc05272009-04-01 20:27:29 +0000497 with original_warnings.catch_warnings(module=self.module):
498 self.module.filterwarnings("always", category=UserWarning)
499 old_showwarning = self.module.showwarning
500 self.module.showwarning = 23
501 try:
502 self.assertRaises(TypeError, self.module.warn, "Warning!")
503 finally:
504 self.module.showwarning = old_showwarning
Christian Heimes8dc226f2008-05-06 23:45:46 +0000505
Christian Heimes33fe8092008-04-13 13:53:33 +0000506 def test_show_warning_output(self):
507 # With showarning() missing, make sure that output is okay.
508 text = 'test show_warning'
Brett Cannon1cd02472008-09-09 01:52:27 +0000509 with original_warnings.catch_warnings(module=self.module):
Christian Heimes33fe8092008-04-13 13:53:33 +0000510 self.module.filterwarnings("always", category=UserWarning)
511 del self.module.showwarning
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000512 with support.captured_output('stderr') as stream:
Christian Heimes33fe8092008-04-13 13:53:33 +0000513 warning_tests.inner(text)
514 result = stream.getvalue()
Georg Brandlab91fde2009-08-13 08:51:18 +0000515 self.assertEqual(result.count('\n'), 2,
Christian Heimes33fe8092008-04-13 13:53:33 +0000516 "Too many newlines in %r" % result)
517 first_line, second_line = result.split('\n', 1)
518 expected_file = os.path.splitext(warning_tests.__file__)[0] + '.py'
Neal Norwitz32dde222008-04-15 06:43:13 +0000519 first_line_parts = first_line.rsplit(':', 3)
520 path, line, warning_class, message = first_line_parts
Christian Heimes33fe8092008-04-13 13:53:33 +0000521 line = int(line)
Georg Brandlab91fde2009-08-13 08:51:18 +0000522 self.assertEqual(expected_file, path)
523 self.assertEqual(warning_class, ' ' + UserWarning.__name__)
524 self.assertEqual(message, ' ' + text)
Christian Heimes33fe8092008-04-13 13:53:33 +0000525 expected_line = ' ' + linecache.getline(path, line).strip() + '\n'
526 assert expected_line
Georg Brandlab91fde2009-08-13 08:51:18 +0000527 self.assertEqual(second_line, expected_line)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000528
529
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000530class WarningsDisplayTests(unittest.TestCase):
531
Christian Heimes33fe8092008-04-13 13:53:33 +0000532 """Test the displaying of warnings and the ability to overload functions
533 related to displaying warnings."""
534
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000535 def test_formatwarning(self):
536 message = "msg"
537 category = Warning
538 file_name = os.path.splitext(warning_tests.__file__)[0] + '.py'
539 line_num = 3
540 file_line = linecache.getline(file_name, line_num).strip()
Christian Heimes33fe8092008-04-13 13:53:33 +0000541 format = "%s:%s: %s: %s\n %s\n"
542 expect = format % (file_name, line_num, category.__name__, message,
543 file_line)
Georg Brandlab91fde2009-08-13 08:51:18 +0000544 self.assertEqual(expect, self.module.formatwarning(message,
Christian Heimes33fe8092008-04-13 13:53:33 +0000545 category, file_name, line_num))
546 # Test the 'line' argument.
547 file_line += " for the win!"
548 expect = format % (file_name, line_num, category.__name__, message,
549 file_line)
Georg Brandlab91fde2009-08-13 08:51:18 +0000550 self.assertEqual(expect, self.module.formatwarning(message,
Christian Heimes33fe8092008-04-13 13:53:33 +0000551 category, file_name, line_num, file_line))
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000552
553 def test_showwarning(self):
554 file_name = os.path.splitext(warning_tests.__file__)[0] + '.py'
555 line_num = 3
556 expected_file_line = linecache.getline(file_name, line_num).strip()
557 message = 'msg'
558 category = Warning
559 file_object = StringIO()
Christian Heimes33fe8092008-04-13 13:53:33 +0000560 expect = self.module.formatwarning(message, category, file_name,
561 line_num)
562 self.module.showwarning(message, category, file_name, line_num,
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000563 file_object)
Georg Brandlab91fde2009-08-13 08:51:18 +0000564 self.assertEqual(file_object.getvalue(), expect)
Christian Heimes33fe8092008-04-13 13:53:33 +0000565 # Test 'line' argument.
566 expected_file_line += "for the win!"
567 expect = self.module.formatwarning(message, category, file_name,
568 line_num, expected_file_line)
569 file_object = StringIO()
570 self.module.showwarning(message, category, file_name, line_num,
571 file_object, expected_file_line)
Georg Brandlab91fde2009-08-13 08:51:18 +0000572 self.assertEqual(expect, file_object.getvalue())
Christian Heimes33fe8092008-04-13 13:53:33 +0000573
574class CWarningsDisplayTests(BaseTest, WarningsDisplayTests):
575 module = c_warnings
576
577class PyWarningsDisplayTests(BaseTest, WarningsDisplayTests):
578 module = py_warnings
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000579
Brett Cannon1cd02472008-09-09 01:52:27 +0000580
Brett Cannonec92e182008-09-02 02:46:59 +0000581class CatchWarningTests(BaseTest):
Nick Coghlanb1304932008-07-13 12:25:08 +0000582
Brett Cannonec92e182008-09-02 02:46:59 +0000583 """Test catch_warnings()."""
584
585 def test_catch_warnings_restore(self):
Nick Coghlanb1304932008-07-13 12:25:08 +0000586 wmod = self.module
587 orig_filters = wmod.filters
588 orig_showwarning = wmod.showwarning
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000589 # Ensure both showwarning and filters are restored when recording
590 with wmod.catch_warnings(module=wmod, record=True):
Nick Coghlanb1304932008-07-13 12:25:08 +0000591 wmod.filters = wmod.showwarning = object()
Georg Brandlab91fde2009-08-13 08:51:18 +0000592 self.assertTrue(wmod.filters is orig_filters)
593 self.assertTrue(wmod.showwarning is orig_showwarning)
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000594 # Same test, but with recording disabled
595 with wmod.catch_warnings(module=wmod, record=False):
Nick Coghlanb1304932008-07-13 12:25:08 +0000596 wmod.filters = wmod.showwarning = object()
Georg Brandlab91fde2009-08-13 08:51:18 +0000597 self.assertTrue(wmod.filters is orig_filters)
598 self.assertTrue(wmod.showwarning is orig_showwarning)
Nick Coghlanb1304932008-07-13 12:25:08 +0000599
Brett Cannonec92e182008-09-02 02:46:59 +0000600 def test_catch_warnings_recording(self):
Nick Coghlanb1304932008-07-13 12:25:08 +0000601 wmod = self.module
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000602 # Ensure warnings are recorded when requested
603 with wmod.catch_warnings(module=wmod, record=True) as w:
Brett Cannonec92e182008-09-02 02:46:59 +0000604 self.assertEqual(w, [])
Georg Brandlab91fde2009-08-13 08:51:18 +0000605 self.assertTrue(type(w) is list)
Nick Coghlanb1304932008-07-13 12:25:08 +0000606 wmod.simplefilter("always")
607 wmod.warn("foo")
Brett Cannon1cd02472008-09-09 01:52:27 +0000608 self.assertEqual(str(w[-1].message), "foo")
Nick Coghlanb1304932008-07-13 12:25:08 +0000609 wmod.warn("bar")
Brett Cannon1cd02472008-09-09 01:52:27 +0000610 self.assertEqual(str(w[-1].message), "bar")
Brett Cannonec92e182008-09-02 02:46:59 +0000611 self.assertEqual(str(w[0].message), "foo")
612 self.assertEqual(str(w[1].message), "bar")
Brett Cannon1cd02472008-09-09 01:52:27 +0000613 del w[:]
Brett Cannonec92e182008-09-02 02:46:59 +0000614 self.assertEqual(w, [])
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000615 # Ensure warnings are not recorded when not requested
Nick Coghlanb1304932008-07-13 12:25:08 +0000616 orig_showwarning = wmod.showwarning
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000617 with wmod.catch_warnings(module=wmod, record=False) as w:
Georg Brandlab91fde2009-08-13 08:51:18 +0000618 self.assertTrue(w is None)
619 self.assertTrue(wmod.showwarning is orig_showwarning)
Nick Coghlanb1304932008-07-13 12:25:08 +0000620
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000621 def test_catch_warnings_reentry_guard(self):
622 wmod = self.module
623 # Ensure catch_warnings is protected against incorrect usage
624 x = wmod.catch_warnings(module=wmod, record=True)
625 self.assertRaises(RuntimeError, x.__exit__)
626 with x:
627 self.assertRaises(RuntimeError, x.__enter__)
628 # Same test, but with recording disabled
629 x = wmod.catch_warnings(module=wmod, record=False)
630 self.assertRaises(RuntimeError, x.__exit__)
631 with x:
632 self.assertRaises(RuntimeError, x.__enter__)
633
634 def test_catch_warnings_defaults(self):
635 wmod = self.module
636 orig_filters = wmod.filters
637 orig_showwarning = wmod.showwarning
638 # Ensure default behaviour is not to record warnings
639 with wmod.catch_warnings(module=wmod) as w:
Georg Brandlab91fde2009-08-13 08:51:18 +0000640 self.assertTrue(w is None)
641 self.assertTrue(wmod.showwarning is orig_showwarning)
642 self.assertTrue(wmod.filters is not orig_filters)
643 self.assertTrue(wmod.filters is orig_filters)
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000644 if wmod is sys.modules['warnings']:
645 # Ensure the default module is this one
646 with wmod.catch_warnings() as w:
Georg Brandlab91fde2009-08-13 08:51:18 +0000647 self.assertTrue(w is None)
648 self.assertTrue(wmod.showwarning is orig_showwarning)
649 self.assertTrue(wmod.filters is not orig_filters)
650 self.assertTrue(wmod.filters is orig_filters)
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000651
652 def test_check_warnings(self):
653 # Explicit tests for the test.support convenience wrapper
654 wmod = self.module
655 if wmod is sys.modules['warnings']:
656 with support.check_warnings() as w:
657 self.assertEqual(w.warnings, [])
658 wmod.simplefilter("always")
659 wmod.warn("foo")
660 self.assertEqual(str(w.message), "foo")
661 wmod.warn("bar")
662 self.assertEqual(str(w.message), "bar")
663 self.assertEqual(str(w.warnings[0].message), "foo")
664 self.assertEqual(str(w.warnings[1].message), "bar")
665 w.reset()
666 self.assertEqual(w.warnings, [])
667
Brett Cannonec92e182008-09-02 02:46:59 +0000668class CCatchWarningTests(CatchWarningTests):
Nick Coghlanb1304932008-07-13 12:25:08 +0000669 module = c_warnings
670
Brett Cannonec92e182008-09-02 02:46:59 +0000671class PyCatchWarningTests(CatchWarningTests):
Nick Coghlanb1304932008-07-13 12:25:08 +0000672 module = py_warnings
673
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000674
Victor Stinner148051a2010-05-20 21:00:34 +0000675class BootstrapTest(unittest.TestCase):
676 def test_issue_8766(self):
677 # "import encodings" emits a warning whereas the warnings is not loaded
678 # or not completly loaded (warnings imports indirectly encodings by
679 # importing linecache) yet
Victor Stinnerd6563542010-05-20 21:42:00 +0000680 cwd = tempfile.mkdtemp()
Victor Stinner148051a2010-05-20 21:00:34 +0000681 try:
Victor Stinnerd6563542010-05-20 21:42:00 +0000682 encodings = os.path.join(cwd, 'encodings')
683 os.mkdir(encodings)
Victor Stinner148051a2010-05-20 21:00:34 +0000684 try:
Victor Stinner148051a2010-05-20 21:00:34 +0000685 env = os.environ.copy()
686 env['PYTHONPATH'] = cwd
687
688 # encodings loaded by initfsencoding()
689 retcode = subprocess.call([sys.executable, '-c', 'pass'], env=env)
690 self.assertEqual(retcode, 0)
691
692 # Use -W to load warnings module at startup
693 retcode = subprocess.call(
694 [sys.executable, '-c', 'pass', '-W', 'always'],
695 env=env)
696 self.assertEqual(retcode, 0)
697 finally:
Victor Stinnerd6563542010-05-20 21:42:00 +0000698 os.rmdir(encodings)
Victor Stinner148051a2010-05-20 21:00:34 +0000699 finally:
Victor Stinnerd6563542010-05-20 21:42:00 +0000700 os.rmdir(cwd)
Victor Stinner148051a2010-05-20 21:00:34 +0000701
Christian Heimes33fe8092008-04-13 13:53:33 +0000702def test_main():
Christian Heimesdae2a892008-04-19 00:55:37 +0000703 py_warnings.onceregistry.clear()
704 c_warnings.onceregistry.clear()
Victor Stinner148051a2010-05-20 21:00:34 +0000705 support.run_unittest(
706 CFilterTests, PyFilterTests,
707 CWarnTests, PyWarnTests,
708 CWCmdLineTests, PyWCmdLineTests,
709 _WarningsTests,
710 CWarningsDisplayTests, PyWarningsDisplayTests,
711 CCatchWarningTests, PyCatchWarningTests,
712 BootstrapTest,
713 )
Christian Heimes33fe8092008-04-13 13:53:33 +0000714
Raymond Hettingerd6f6e502003-07-13 08:37:40 +0000715
716if __name__ == "__main__":
Christian Heimes33fe8092008-04-13 13:53:33 +0000717 test_main()