# spaceconfig = {"usemodules": ["_warnings"]} import pytest import warnings import _warnings import io import os import sys def test_defaults(): assert _warnings._onceregistry == {} assert _warnings._defaultaction == 'default' expected = [('default', None, DeprecationWarning, '__main__', 0), ('ignore', None, DeprecationWarning, None, 0), ('ignore', None, PendingDeprecationWarning, None, 0), ('ignore', None, ImportWarning, None, 0), ('ignore', None, ResourceWarning, None, 0)] try: import pkg_resources except: pass if len(_warnings.filters) == 6: expected.append(('ignore', None, pkg_resources.PEP440Warning, None, 0)) assert _warnings.filters == expected def test_warn(): with warnings.catch_warnings(record=True) as log: _warnings.warn("some message", DeprecationWarning) _warnings.warn("some message", Warning) _warnings.warn(("some message",1), Warning) def test_use_builtin__warnings(): """Check that the stdlib warnings.py module manages to import our _warnings module. If something is missing, it doesn't, and silently continues. Then things don't reliably work: either the functionality of the pure Python version is subtly different, or more likely we get confusion because of a half-imported _warnings. """ assert not hasattr(warnings, '_filters_version') def test_lineno(): with warnings.catch_warnings(record=True) as w: _warnings.warn("some message", Warning) lineno = sys._getframe().f_lineno - 1 # the line above assert w[-1].lineno == lineno def test_warn_explicit(): with warnings.catch_warnings(record=True) as log: _warnings.warn_explicit("some message", DeprecationWarning, "", 1, module_globals=globals()) _warnings.warn_explicit("some message", Warning, "", 1, module_globals=globals()) def test_with_source(): source = [] with warnings.catch_warnings(record=True) as w: _warnings.warn("some message", Warning, source=source) assert w[0].source is source def test_default_action(): warnings.defaultaction = 'ignore' warnings.resetwarnings() with warnings.catch_warnings(record=True) as w: _warnings.warn_explicit("message", UserWarning, "", 44, registry={}) assert len(w) == 0 warnings.defaultaction = 'default' def test_ignore(): warnings.resetwarnings() with warnings.catch_warnings(record=True) as w: __warningregistry__ = {} warnings.filterwarnings("ignore", category=UserWarning) _warnings.warn_explicit("message", UserWarning, "", 44, registry=__warningregistry__) assert len(w) == 0 assert list(__warningregistry__) == ['version'] def test_show_source_line(): # Something is wrong with pytest > 4.0.0 (which is the version run for -D # pypy tests: it cannot redirect sys.stderr if getattr(pytest, "__version__", "untranslated")[0] >= '4': pytest.skip("fails on this version of pytest") def inner(message, stacklevel=1): warnings.warn(message, stacklevel=stacklevel) # With showarning() missing, make sure that output is okay. saved = warnings.showwarning try: del warnings.showwarning stderr = sys.stderr try: sys.stderr = io.StringIO() inner('test message') result = sys.stderr.getvalue() finally: sys.stderr = stderr assert result.count('\n') == 2 assert ' warnings.warn(message, ' in result finally: warnings.showwarning = saved def test_filename_none(): with warnings.catch_warnings(record=True) as log: globals()['__file__'] = 'test.pyc' _warnings.warn('test', UserWarning) globals()['__file__'] = None _warnings.warn('test', UserWarning) assert len(log) == 2 def test_warn_unicode(): # Something is wrong with pytest > 4.0.0 (which is the version run for -D # pypy tests: it cannot redirect sys.stderr if getattr(pytest, "__version__", "untranslated")[0] >= '4': pytest.skip("fails on this version of pytest") old = sys.stderr, warnings.showwarning with warnings.catch_warnings(record=True) as log: try: class Grab: def write(self, u): self.data.append(u) sys.stderr = Grab() sys.stderr.data = data = [] if sys.version_info > (3, 0, 0): # Copy from lib-python/3/warnings.py def orig_showwarning(message, category, filename, lineno, file=None, line=None): msg = warnings.WarningMessage(message, category, filename, lineno, file, line) warnings._showwarnmsg_impl(msg) warnings.showwarning = orig_showwarning _unicode = str else: warnings.showwarning = warnings._show_warning _unicode = unicode # ^^^ disables any catch_warnings() issued by the test runner _warnings.warn_explicit("9238exbexn8", Warning, "", 1421, module_globals=globals()) assert data # the warning was not swallowed assert isinstance(''.join(data), str) _warnings.warn_explicit(u"\u1234\u5678", UserWarning, "", 831, module_globals=globals()) assert isinstance(''.join(data), _unicode) assert ''.join(data).endswith( u':831: UserWarning: \u1234\u5678\n') finally: sys.stderr, warnings.showwarning = old assert len(log) == 5 def test_bad_category(): with raises(TypeError): _warnings.warn("text", 123) class Foo: pass with raises(TypeError): _warnings.warn("text", Foo) def test_surrogate_in_filename(): with warnings.catch_warnings(record=True) as log: for filename in ("nonascii\xe9\u20ac", "surrogate\udc80"): try: os.fsencode(filename) except UnicodeEncodeError: continue _warnings.warn_explicit("text", UserWarning, filename, 1) assert len(log) == 2 def test_issue31285(): def get_bad_loader(splitlines_ret_val): class BadLoader: def get_source(self, fullname): class BadSource(str): def splitlines(self): return splitlines_ret_val return BadSource('spam') return BadLoader() # does not raise: with warnings.catch_warnings(record=True) as log: _warnings.warn_explicit( 'eggs', UserWarning, 'bar', 1, module_globals={'__loader__': get_bad_loader(42), '__name__': 'foobar'}) assert len(log) == 1 def test_once_is_not_broken(): def f(): warnings.warn("deprecated", DeprecationWarning, 2) with warnings.catch_warnings(record=True) as w: warnings.simplefilter("once") assert len(w) == 0 f() assert len(w) == 1 f() assert len(w) == 1 f() assert len(w) == 1 def test_filename_from_code(): f = eval("lambda: warnings.warn('foo')") with warnings.catch_warnings(record=True) as w: assert len(w) == 0 f() print(w[0].__dict__) assert len(w) == 1 assert w[0].filename == ""