2010-11-06 18:38:53 +08:00
|
|
|
"""
|
2011-05-27 02:15:21 +08:00
|
|
|
support for presenting detailed information in failing assertions.
|
2010-11-06 18:38:53 +08:00
|
|
|
"""
|
2009-08-27 23:26:02 +08:00
|
|
|
import py
|
2009-08-29 01:16:15 +08:00
|
|
|
import sys
|
2010-11-13 18:10:45 +08:00
|
|
|
from _pytest.monkeypatch import monkeypatch
|
2011-06-29 10:13:12 +08:00
|
|
|
from _pytest.assertion import util
|
2009-08-27 23:26:02 +08:00
|
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
|
|
group = parser.getgroup("debugconfig")
|
2011-07-05 23:29:53 +08:00
|
|
|
group.addoption('--assert', action="store", dest="assertmode",
|
|
|
|
choices=("rewrite", "reinterp", "plain",),
|
|
|
|
default="rewrite", metavar="MODE",
|
2011-05-27 12:15:33 +08:00
|
|
|
help="""control assertion debugging tools.
|
2011-07-05 23:29:53 +08:00
|
|
|
'plain' performs no assertion debugging.
|
|
|
|
'reinterp' reinterprets assert statements after they failed to provide assertion expression information.
|
|
|
|
'rewrite' (the default) rewrites assert statements in test modules on import
|
|
|
|
to provide assert expression information. """)
|
2011-05-27 12:15:33 +08:00
|
|
|
group.addoption('--no-assert', action="store_true", default=False,
|
2011-07-05 23:29:53 +08:00
|
|
|
dest="noassert", help="DEPRECATED equivalent to --assert=plain")
|
2013-10-02 20:32:40 +08:00
|
|
|
group.addoption('--nomagic', '--no-magic', action="store_true",
|
2013-08-01 22:49:26 +08:00
|
|
|
default=False, help="DEPRECATED equivalent to --assert=plain")
|
2011-05-27 03:34:27 +08:00
|
|
|
|
2011-05-27 05:08:25 +08:00
|
|
|
class AssertionState:
|
|
|
|
"""State for the assertion plugin."""
|
|
|
|
|
|
|
|
def __init__(self, config, mode):
|
|
|
|
self.mode = mode
|
|
|
|
self.trace = config.trace.root.get("assertion")
|
2009-08-27 23:26:02 +08:00
|
|
|
|
|
|
|
def pytest_configure(config):
|
2011-05-27 03:34:27 +08:00
|
|
|
mode = config.getvalue("assertmode")
|
|
|
|
if config.getvalue("noassert") or config.getvalue("nomagic"):
|
2011-07-05 23:29:53 +08:00
|
|
|
mode = "plain"
|
2011-07-05 21:21:08 +08:00
|
|
|
if mode == "rewrite":
|
|
|
|
try:
|
2013-10-12 21:39:22 +08:00
|
|
|
import ast # noqa
|
2011-07-05 21:21:08 +08:00
|
|
|
except ImportError:
|
|
|
|
mode = "reinterp"
|
2011-09-21 12:45:40 +08:00
|
|
|
else:
|
2013-03-22 01:19:01 +08:00
|
|
|
# Both Jython and CPython 2.6.0 have AST bugs that make the
|
|
|
|
# assertion rewriting hook malfunction.
|
|
|
|
if (sys.platform.startswith('java') or
|
|
|
|
sys.version_info[:3] == (2, 6, 0)):
|
2011-09-21 12:45:40 +08:00
|
|
|
mode = "reinterp"
|
2011-07-05 23:29:53 +08:00
|
|
|
if mode != "plain":
|
2011-06-29 10:13:12 +08:00
|
|
|
_load_modules(mode)
|
2011-05-27 03:34:27 +08:00
|
|
|
m = monkeypatch()
|
|
|
|
config._cleanup.append(m.undo)
|
2011-05-26 06:54:02 +08:00
|
|
|
m.setattr(py.builtin.builtins, 'AssertionError',
|
2013-10-12 21:39:22 +08:00
|
|
|
reinterpret.AssertionError) # noqa
|
2011-06-29 10:13:12 +08:00
|
|
|
hook = None
|
2011-06-29 22:44:04 +08:00
|
|
|
if mode == "rewrite":
|
2013-10-12 21:39:22 +08:00
|
|
|
hook = rewrite.AssertionRewritingHook() # noqa
|
2012-05-06 05:31:05 +08:00
|
|
|
sys.meta_path.insert(0, hook)
|
2011-06-29 10:13:12 +08:00
|
|
|
warn_about_missing_assertion(mode)
|
2011-05-27 05:18:18 +08:00
|
|
|
config._assertstate = AssertionState(config, mode)
|
2011-06-29 10:13:12 +08:00
|
|
|
config._assertstate.hook = hook
|
2011-05-27 05:18:18 +08:00
|
|
|
config._assertstate.trace("configured with mode set to %r" % (mode,))
|
2011-05-20 10:52:10 +08:00
|
|
|
|
2011-06-29 10:13:12 +08:00
|
|
|
def pytest_unconfigure(config):
|
|
|
|
hook = config._assertstate.hook
|
|
|
|
if hook is not None:
|
|
|
|
sys.meta_path.remove(hook)
|
2011-05-25 06:48:56 +08:00
|
|
|
|
2011-07-06 00:01:31 +08:00
|
|
|
def pytest_collection(session):
|
|
|
|
# this hook is only called when test modules are collected
|
|
|
|
# so for example not in the master process of pytest-xdist
|
|
|
|
# (which does not collect test modules)
|
2011-06-29 10:13:12 +08:00
|
|
|
hook = session.config._assertstate.hook
|
|
|
|
if hook is not None:
|
|
|
|
hook.set_session(session)
|
2011-05-20 10:52:10 +08:00
|
|
|
|
2011-10-16 18:51:15 +08:00
|
|
|
def pytest_runtest_setup(item):
|
|
|
|
def callbinrepr(op, left, right):
|
|
|
|
hook_result = item.ihook.pytest_assertrepr_compare(
|
|
|
|
config=item.config, op=op, left=left, right=right)
|
2012-06-27 23:26:55 +08:00
|
|
|
|
2011-10-16 18:51:15 +08:00
|
|
|
for new_expl in hook_result:
|
|
|
|
if new_expl:
|
2013-11-22 20:28:59 +08:00
|
|
|
# Don't include pageloads of data unless we are very
|
|
|
|
# verbose (-vv)
|
2014-01-29 08:39:04 +08:00
|
|
|
if (sum(len(p) for p in new_expl[1:]) > 80*8
|
2013-11-22 20:28:59 +08:00
|
|
|
and item.config.option.verbose < 2):
|
|
|
|
new_expl[1:] = [py.builtin._totext(
|
2014-01-29 08:39:04 +08:00
|
|
|
'Detailed information truncated, use "-vv" to show')]
|
2013-11-22 20:28:59 +08:00
|
|
|
res = py.builtin._totext('\n~').join(new_expl)
|
2011-10-16 18:51:15 +08:00
|
|
|
if item.config.getvalue("assertmode") == "rewrite":
|
|
|
|
# The result will be fed back a python % formatting
|
|
|
|
# operation, which will fail if there are extraneous
|
|
|
|
# '%'s in the string. Escape them here.
|
|
|
|
res = res.replace("%", "%%")
|
|
|
|
return res
|
|
|
|
util._reprcompare = callbinrepr
|
|
|
|
|
|
|
|
def pytest_runtest_teardown(item):
|
|
|
|
util._reprcompare = None
|
|
|
|
|
2011-06-29 10:13:12 +08:00
|
|
|
def pytest_sessionfinish(session):
|
|
|
|
hook = session.config._assertstate.hook
|
|
|
|
if hook is not None:
|
|
|
|
hook.session = None
|
2009-08-27 23:26:02 +08:00
|
|
|
|
2011-06-29 10:13:12 +08:00
|
|
|
def _load_modules(mode):
|
|
|
|
"""Lazily import assertion related code."""
|
|
|
|
global rewrite, reinterpret
|
2013-10-12 21:39:22 +08:00
|
|
|
from _pytest.assertion import reinterpret # noqa
|
2011-06-29 22:44:04 +08:00
|
|
|
if mode == "rewrite":
|
2013-10-12 21:39:22 +08:00
|
|
|
from _pytest.assertion import rewrite # noqa
|
2011-06-29 10:13:12 +08:00
|
|
|
|
|
|
|
def warn_about_missing_assertion(mode):
|
2009-08-27 23:26:02 +08:00
|
|
|
try:
|
|
|
|
assert False
|
|
|
|
except AssertionError:
|
|
|
|
pass
|
|
|
|
else:
|
2011-06-29 22:44:04 +08:00
|
|
|
if mode == "rewrite":
|
2011-06-29 10:13:12 +08:00
|
|
|
specifically = ("assertions which are not in test modules "
|
|
|
|
"will be ignored")
|
|
|
|
else:
|
|
|
|
specifically = "failing tests may report as passing"
|
|
|
|
|
|
|
|
sys.stderr.write("WARNING: " + specifically +
|
2011-07-05 23:29:53 +08:00
|
|
|
" because assert statements are not executed "
|
|
|
|
"by the underlying Python interpreter "
|
2011-06-29 10:13:12 +08:00
|
|
|
"(are you using python -O?)\n")
|
2010-09-07 02:35:17 +08:00
|
|
|
|
2011-05-27 01:01:34 +08:00
|
|
|
pytest_assertrepr_compare = util.assertrepr_compare
|