ansible-later/env_27/lib/python2.7/site-packages/testfixtures/tests/test_shouldwarn.py

151 lines
4.8 KiB
Python
Raw Normal View History

2019-04-11 13:56:20 +00:00
from unittest import TestCase
import warnings
from testfixtures import (
ShouldWarn, compare, ShouldRaise, ShouldNotWarn,
Comparison as C
)
from testfixtures.compat import PY3, PY_36_PLUS, PY_37_PLUS
from testfixtures.shouldraise import ShouldAssert
if PY3:
warn_module = 'builtins'
else:
warn_module = 'exceptions'
if PY_37_PLUS:
comma = ''
else:
comma = ','
class ShouldWarnTests(TestCase):
def test_warn_expected(self):
with warnings.catch_warnings(record=True) as backstop:
with ShouldWarn(UserWarning('foo')):
warnings.warn('foo')
compare(len(backstop), expected=0)
def test_warn_not_expected(self):
with ShouldAssert(
"sequence not as expected:\n\n"
"same:\n[]\n\n"
"expected:\n[]\n\n"
"actual:\n[UserWarning('foo'"+comma+")]"
):
with warnings.catch_warnings(record=True) as backstop:
with ShouldNotWarn():
warnings.warn('foo')
compare(len(backstop), expected=0)
def test_no_warn_expected(self):
with ShouldNotWarn():
pass
def test_no_warn_not_expected(self):
with ShouldAssert(
"sequence not as expected:\n\n"
"same:\n[]\n\n"
"expected:\n[<C:"+warn_module+".UserWarning>args: ('foo',)</C>]"
"\n\nactual:\n[]"
):
with ShouldWarn(UserWarning('foo')):
pass
def test_filters_removed(self):
with warnings.catch_warnings():
warnings.simplefilter("ignore")
with ShouldWarn(UserWarning("foo")):
warnings.warn('foo')
def test_multiple_warnings(self):
with ShouldRaise(AssertionError) as s:
with ShouldWarn(UserWarning('foo')):
warnings.warn('foo')
warnings.warn('bar')
content = str(s.raised)
self.assertTrue('foo' in content)
self.assertTrue('bar' in content)
def test_minimal_ok(self):
with ShouldWarn(UserWarning):
warnings.warn('foo')
def test_minimal_bad(self):
with ShouldAssert(
"sequence not as expected:\n\n"
"same:\n[]\n\n"
"expected:\n"
"[<C(failed):"+warn_module+".DeprecationWarning>wrong type</C>]\n\n"
"actual:\n[UserWarning('foo'"+comma+")]"
):
with ShouldWarn(DeprecationWarning):
warnings.warn('foo')
def test_maximal_ok(self):
with ShouldWarn(DeprecationWarning('foo')):
warnings.warn_explicit(
'foo', DeprecationWarning, 'bar.py', 42, 'bar_module'
)
def test_maximal_bad(self):
with ShouldAssert(
"sequence not as expected:\n\n"
"same:\n[]\n\n"
"expected:\n[\n"
"<C(failed):"+warn_module+".DeprecationWarning>\n"
"attributes differ:\n"
"'args': ('bar',) (Comparison) != ('foo',) (actual)\n</C>]\n\n"
"actual:\n[DeprecationWarning('foo'"+comma+")]"
):
with ShouldWarn(DeprecationWarning('bar')):
warnings.warn_explicit(
'foo', DeprecationWarning, 'bar.py', 42, 'bar_module'
)
def test_maximal_explore(self):
with ShouldWarn() as recorded:
warnings.warn_explicit(
'foo', DeprecationWarning, 'bar.py', 42, 'bar_module'
)
compare(len(recorded), expected=1)
expected_attrs = dict(
_category_name='DeprecationWarning',
category=DeprecationWarning,
file=None,
filename='bar.py',
line=None,
lineno=42,
message=C(DeprecationWarning('foo')),
)
if PY_36_PLUS:
expected_attrs['source'] = None
compare(expected=C(warnings.WarningMessage, **expected_attrs),
actual=recorded[0])
def test_filter_present(self):
with ShouldWarn(DeprecationWarning,
message="This function is deprecated."):
warnings.warn("This utility is deprecated.", DeprecationWarning)
warnings.warn("This function is deprecated.", DeprecationWarning)
def test_filter_missing(self):
if PY3:
type_repr = 'builtins.DeprecationWarning'
else:
type_repr = 'exceptions.DeprecationWarning'
with ShouldAssert(
"sequence not as expected:\n\n"
"same:\n[]\n\n"
"expected:\n[<C:{}>]\n\n"
"actual:\n[]".format(type_repr)
):
with ShouldWarn(DeprecationWarning,
message="This function is deprecated."):
warnings.warn("This utility is deprecated.", DeprecationWarning)