/usr/lib/python3/dist-packages/testfixtures/shouldraise.py is in python3-testfixtures 4.14.3-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | # Copyright (c) 2008-2014 Simplistix Ltd
# See license.txt for license details.
from functools import wraps
from testfixtures import Comparison
param_docs = """
:param exception: This can be one of the following:
* `None`, indicating that an exception must be
raised, but the type is unimportant.
* An exception class, indicating that the type
of the exception is important but not the
parameters it is created with.
* An exception instance, indicating that an
exception exactly matching the one supplied
should be raised.
:param unless: Can be passed a boolean that, when ``True`` indicates that
no exception is expected. This is useful when checking
that exceptions are only raised on certain versions of
Python.
"""
class ShouldRaise(object):
__doc__ = """
This context manager is used to assert that an exception is raised
within the context it is managing.
""" + param_docs
#: The exception captured by the context manager.
#: Can be used to inspect specific attributes of the exception.
raised = None
def __init__(self, exception=None, unless=False):
self.exception = exception
self.expected = not unless
def __enter__(self):
return self
def __exit__(self, type, actual, traceback):
__tracebackhide__ = True
# bug in python :-(
if type is not None and not isinstance(actual, type):
# fixed in 2.7 onwards!
actual = type(actual) # pragma: no cover
self.raised = actual
if self.expected:
if self.exception:
comparison = Comparison(self.exception)
if comparison != actual:
repr_actual = repr(actual)
repr_expected = repr(self.exception)
message = '%s raised, %s expected' % (
repr_actual, repr_expected
)
if repr_actual == repr_expected:
extra = [', attributes differ:']
extra.extend(str(comparison).split('\n')[2:-1])
message += '\n'.join(extra)
raise AssertionError(message)
elif not actual:
raise AssertionError('No exception raised!')
elif actual:
raise AssertionError('%r raised, no exception expected' % actual)
return True
class should_raise:
__doc__ = """
A decorator to assert that the decorated function will raised
an exception. An exception class or exception instance may be
passed to check more specifically exactly what exception will be
raised.
""" + param_docs
def __init__(self, exception=None, unless=None):
self.exception = exception
self.unless = unless
def __call__(self, target):
@wraps(target)
def _should_raise_wrapper(*args, **kw):
with ShouldRaise(self.exception, self.unless):
target(*args, **kw)
return _should_raise_wrapper
|