2010-11-06 18:38:53 +08:00
|
|
|
""" Python test discovery, setup and run of test functions. """
|
2008-08-16 23:26:59 +08:00
|
|
|
import py
|
2009-09-01 22:10:21 +08:00
|
|
|
import inspect
|
2010-07-04 23:06:50 +08:00
|
|
|
import sys
|
2010-10-07 17:59:00 +08:00
|
|
|
import pytest
|
2013-10-12 21:39:22 +08:00
|
|
|
from _pytest.mark import MarkDecorator
|
2012-08-01 15:23:39 +08:00
|
|
|
from py._code.code import TerminalRepr
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2010-11-13 18:10:45 +08:00
|
|
|
import _pytest
|
|
|
|
cutdir = py.path.local(_pytest.__file__).dirpath()
|
2010-10-10 19:48:49 +08:00
|
|
|
|
2013-07-16 21:43:20 +08:00
|
|
|
NoneType = type(None)
|
2013-11-21 20:15:32 +08:00
|
|
|
NOTSET = object()
|
2013-07-16 21:43:20 +08:00
|
|
|
|
2012-10-19 16:07:11 +08:00
|
|
|
callable = py.builtin.callable
|
|
|
|
|
2013-11-20 06:22:27 +08:00
|
|
|
def getfslineno(obj):
|
|
|
|
# xxx let decorators etc specify a sane ordering
|
|
|
|
while hasattr(obj, "__wrapped__"):
|
|
|
|
obj = obj.__wrapped__
|
|
|
|
if hasattr(obj, 'place_as'):
|
|
|
|
obj = obj.place_as
|
|
|
|
fslineno = py.code.getfslineno(obj)
|
|
|
|
assert isinstance(fslineno[1], int), obj
|
|
|
|
return fslineno
|
|
|
|
|
2012-10-20 15:59:20 +08:00
|
|
|
def getimfunc(func):
|
|
|
|
try:
|
|
|
|
return func.__func__
|
|
|
|
except AttributeError:
|
|
|
|
try:
|
|
|
|
return func.im_func
|
|
|
|
except AttributeError:
|
|
|
|
return func
|
|
|
|
|
|
|
|
|
2012-10-05 16:21:35 +08:00
|
|
|
class FixtureFunctionMarker:
|
2013-09-26 18:57:21 +08:00
|
|
|
def __init__(self, scope, params, autouse=False, yieldctx=False):
|
2012-08-02 18:41:46 +08:00
|
|
|
self.scope = scope
|
|
|
|
self.params = params
|
2012-10-12 20:52:36 +08:00
|
|
|
self.autouse = autouse
|
2013-09-26 18:57:21 +08:00
|
|
|
self.yieldctx = yieldctx
|
2012-10-05 16:21:35 +08:00
|
|
|
|
2012-08-02 18:41:46 +08:00
|
|
|
def __call__(self, function):
|
2012-10-08 17:22:31 +08:00
|
|
|
if inspect.isclass(function):
|
2013-09-26 18:57:21 +08:00
|
|
|
raise ValueError(
|
|
|
|
"class fixtures not supported (may be in the future)")
|
2012-10-05 16:21:35 +08:00
|
|
|
function._pytestfixturefunction = self
|
2012-08-02 18:41:46 +08:00
|
|
|
return function
|
|
|
|
|
|
|
|
|
2013-09-30 19:42:39 +08:00
|
|
|
def fixture(scope="function", params=None, autouse=False):
|
2012-10-07 19:06:17 +08:00
|
|
|
""" (return a) decorator to mark a fixture factory function.
|
|
|
|
|
2012-10-08 14:34:21 +08:00
|
|
|
This decorator can be used (with or or without parameters) to define
|
2012-10-07 19:06:17 +08:00
|
|
|
a fixture function. The name of the fixture function can later be
|
|
|
|
referenced to cause its invocation ahead of running tests: test
|
|
|
|
modules or classes can use the pytest.mark.usefixtures(fixturename)
|
2012-10-08 14:34:21 +08:00
|
|
|
marker. Test functions can directly use fixture names as input
|
2012-10-07 03:03:55 +08:00
|
|
|
arguments in which case the fixture instance returned from the fixture
|
|
|
|
function will be injected.
|
2012-10-05 16:21:35 +08:00
|
|
|
|
|
|
|
:arg scope: the scope for which this fixture is shared, one of
|
2012-10-07 19:06:17 +08:00
|
|
|
"function" (default), "class", "module", "session".
|
|
|
|
|
2012-08-02 18:41:46 +08:00
|
|
|
:arg params: an optional list of parameters which will cause multiple
|
2012-10-07 19:06:17 +08:00
|
|
|
invocations of the fixture function and all of the tests
|
|
|
|
using it.
|
2012-10-07 03:03:55 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
:arg autouse: if True, the fixture func is activated for all tests that
|
2012-10-07 03:03:55 +08:00
|
|
|
can see it. If False (the default) then an explicit
|
|
|
|
reference is needed to activate the fixture.
|
2013-09-26 18:57:21 +08:00
|
|
|
|
2012-08-02 18:41:46 +08:00
|
|
|
"""
|
2012-10-19 16:07:11 +08:00
|
|
|
if callable(scope) and params is None and autouse == False:
|
2012-10-07 03:03:55 +08:00
|
|
|
# direct decoration
|
2013-09-26 18:57:21 +08:00
|
|
|
return FixtureFunctionMarker(
|
2013-09-30 19:42:39 +08:00
|
|
|
"function", params, autouse)(scope)
|
2012-10-07 03:03:55 +08:00
|
|
|
else:
|
2013-09-30 19:42:39 +08:00
|
|
|
return FixtureFunctionMarker(scope, params, autouse)
|
|
|
|
|
|
|
|
def yield_fixture(scope="function", params=None, autouse=False):
|
|
|
|
""" (return a) decorator to mark a yield-fixture factory function
|
|
|
|
(EXPERIMENTAL).
|
|
|
|
|
|
|
|
This takes the same arguments as :py:func:`pytest.fixture` but
|
|
|
|
expects a fixture function to use a ``yield`` instead of a ``return``
|
|
|
|
statement to provide a fixture. See
|
|
|
|
http://pytest.org/en/latest/yieldfixture.html for more info.
|
|
|
|
"""
|
|
|
|
if callable(scope) and params is None and autouse == False:
|
|
|
|
# direct decoration
|
|
|
|
return FixtureFunctionMarker(
|
|
|
|
"function", params, autouse, yieldctx=True)(scope)
|
|
|
|
else:
|
|
|
|
return FixtureFunctionMarker(scope, params, autouse, yieldctx=True)
|
2012-08-02 18:41:46 +08:00
|
|
|
|
2012-10-05 16:21:35 +08:00
|
|
|
defaultfuncargprefixmarker = fixture()
|
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
def pyobj_property(name):
|
|
|
|
def get(self):
|
|
|
|
node = self.getparent(getattr(pytest, name))
|
|
|
|
if node is not None:
|
|
|
|
return node.obj
|
|
|
|
doc = "python %s object this node was collected from (can be None)." % (
|
|
|
|
name.lower(),)
|
|
|
|
return property(get, None, None, doc)
|
|
|
|
|
2012-06-25 23:35:33 +08:00
|
|
|
|
2010-09-26 00:23:26 +08:00
|
|
|
def pytest_addoption(parser):
|
2010-11-07 05:17:33 +08:00
|
|
|
group = parser.getgroup("general")
|
2012-10-17 18:57:05 +08:00
|
|
|
group.addoption('--fixtures', '--funcargs',
|
2012-10-05 20:24:44 +08:00
|
|
|
action="store_true", dest="showfixtures", default=False,
|
|
|
|
help="show available fixtures, sorted by plugin appearance")
|
2012-10-16 22:13:12 +08:00
|
|
|
parser.addini("usefixtures", type="args", default=[],
|
2012-10-06 01:20:40 +08:00
|
|
|
help="list of default fixtures to be used with this project")
|
2010-11-25 05:01:04 +08:00
|
|
|
parser.addini("python_files", type="args",
|
|
|
|
default=('test_*.py', '*_test.py'),
|
|
|
|
help="glob-style file patterns for Python test module discovery")
|
|
|
|
parser.addini("python_classes", type="args", default=("Test",),
|
|
|
|
help="prefixes for Python test class discovery")
|
|
|
|
parser.addini("python_functions", type="args", default=("test",),
|
|
|
|
help="prefixes for Python test function and method discovery")
|
2010-09-26 00:23:26 +08:00
|
|
|
|
|
|
|
def pytest_cmdline_main(config):
|
2012-10-05 20:24:44 +08:00
|
|
|
if config.option.showfixtures:
|
|
|
|
showfixtures(config)
|
2010-09-26 00:23:26 +08:00
|
|
|
return 0
|
|
|
|
|
2011-11-17 19:09:21 +08:00
|
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
|
|
try:
|
2012-10-06 17:34:06 +08:00
|
|
|
markers = metafunc.function.parametrize
|
2011-11-17 19:09:21 +08:00
|
|
|
except AttributeError:
|
|
|
|
return
|
2012-10-06 17:34:06 +08:00
|
|
|
for marker in markers:
|
|
|
|
metafunc.parametrize(*marker.args, **marker.kwargs)
|
2011-11-17 19:09:21 +08:00
|
|
|
|
|
|
|
def pytest_configure(config):
|
|
|
|
config.addinivalue_line("markers",
|
|
|
|
"parametrize(argnames, argvalues): call a test function multiple "
|
|
|
|
"times passing in multiple different argument value sets. Example: "
|
2011-11-19 02:32:11 +08:00
|
|
|
"@parametrize('arg1', [1,2]) would lead to two calls of the decorated "
|
2011-11-17 19:09:21 +08:00
|
|
|
"test function, one with arg1=1 and another with arg1=2."
|
2012-10-20 19:56:53 +08:00
|
|
|
" see http://pytest.org/latest/parametrize.html for more info and "
|
|
|
|
"examples."
|
2011-11-17 19:09:21 +08:00
|
|
|
)
|
2012-10-19 16:17:13 +08:00
|
|
|
config.addinivalue_line("markers",
|
|
|
|
"usefixtures(fixturename1, fixturename2, ...): mark tests as needing "
|
2012-10-20 19:56:53 +08:00
|
|
|
"all of the specified fixtures. see http://pytest.org/latest/fixture.html#usefixtures "
|
2012-10-19 16:17:13 +08:00
|
|
|
)
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2012-08-01 15:23:39 +08:00
|
|
|
def pytest_sessionstart(session):
|
2012-10-05 20:24:44 +08:00
|
|
|
session._fixturemanager = FixtureManager(session)
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2010-11-22 06:17:59 +08:00
|
|
|
@pytest.mark.trylast
|
|
|
|
def pytest_namespace():
|
2010-11-06 06:37:31 +08:00
|
|
|
raises.Exception = pytest.fail.Exception
|
|
|
|
return {
|
2012-10-05 16:21:35 +08:00
|
|
|
'fixture': fixture,
|
2013-09-30 19:42:39 +08:00
|
|
|
'yield_fixture': yield_fixture,
|
2010-11-06 06:37:31 +08:00
|
|
|
'raises' : raises,
|
|
|
|
'collect': {
|
2010-10-07 17:59:00 +08:00
|
|
|
'Module': Module, 'Class': Class, 'Instance': Instance,
|
|
|
|
'Function': Function, 'Generator': Generator,
|
2012-10-05 20:24:44 +08:00
|
|
|
'_fillfuncargs': fillfixtures}
|
2010-11-06 06:37:31 +08:00
|
|
|
}
|
2010-09-26 00:23:26 +08:00
|
|
|
|
2013-10-21 19:33:36 +08:00
|
|
|
@fixture(scope="session")
|
2012-10-05 20:24:44 +08:00
|
|
|
def pytestconfig(request):
|
2010-09-26 00:23:26 +08:00
|
|
|
""" the pytest config object with access to command line opts."""
|
|
|
|
return request.config
|
|
|
|
|
2012-06-25 23:35:33 +08:00
|
|
|
|
2012-07-19 01:49:14 +08:00
|
|
|
def pytest_pyfunc_call(__multicall__, pyfuncitem):
|
|
|
|
if not __multicall__.execute():
|
|
|
|
testfunction = pyfuncitem.obj
|
|
|
|
if pyfuncitem._isyieldedfunction():
|
|
|
|
testfunction(*pyfuncitem._args)
|
|
|
|
else:
|
|
|
|
funcargs = pyfuncitem.funcargs
|
2012-10-05 16:21:35 +08:00
|
|
|
testargs = {}
|
2012-10-16 20:19:38 +08:00
|
|
|
for arg in pyfuncitem._fixtureinfo.argnames:
|
2012-10-05 16:21:35 +08:00
|
|
|
testargs[arg] = funcargs[arg]
|
|
|
|
testfunction(**testargs)
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2010-09-26 00:23:26 +08:00
|
|
|
def pytest_collect_file(path, parent):
|
|
|
|
ext = path.ext
|
2010-11-25 05:01:04 +08:00
|
|
|
if ext == ".py":
|
|
|
|
if not parent.session.isinitpath(path):
|
|
|
|
for pat in parent.config.getini('python_files'):
|
|
|
|
if path.fnmatch(pat):
|
|
|
|
break
|
|
|
|
else:
|
|
|
|
return
|
2012-10-22 16:17:50 +08:00
|
|
|
ihook = parent.session.gethookproxy(path)
|
|
|
|
return ihook.pytest_pycollect_makemodule(path=path, parent=parent)
|
2010-09-26 00:23:26 +08:00
|
|
|
|
|
|
|
def pytest_pycollect_makemodule(path, parent):
|
2010-10-14 00:41:53 +08:00
|
|
|
return Module(path, parent)
|
2010-09-26 00:23:26 +08:00
|
|
|
|
|
|
|
def pytest_pycollect_makeitem(__multicall__, collector, name, obj):
|
|
|
|
res = __multicall__.execute()
|
|
|
|
if res is not None:
|
|
|
|
return res
|
2011-03-16 23:36:18 +08:00
|
|
|
if inspect.isclass(obj):
|
2010-11-17 19:21:24 +08:00
|
|
|
#if hasattr(collector.obj, 'unittest'):
|
|
|
|
# return # we assume it's a mixin class for a TestCase derived one
|
2011-03-16 23:36:18 +08:00
|
|
|
if collector.classnamefilter(name):
|
2013-02-15 17:18:00 +08:00
|
|
|
Class = collector._getcustomclass("Class")
|
|
|
|
return Class(name, parent=collector)
|
2013-04-29 16:31:51 +08:00
|
|
|
elif collector.funcnamefilter(name) and hasattr(obj, '__call__') and \
|
|
|
|
getfixturemarker(obj) is None:
|
2010-09-26 00:23:26 +08:00
|
|
|
if is_generator(obj):
|
2010-10-14 00:41:53 +08:00
|
|
|
return Generator(name, parent=collector)
|
2010-09-26 00:23:26 +08:00
|
|
|
else:
|
2012-10-06 17:34:06 +08:00
|
|
|
return list(collector._genfunctions(name, obj))
|
2010-09-26 00:23:26 +08:00
|
|
|
|
|
|
|
def is_generator(func):
|
|
|
|
try:
|
|
|
|
return py.code.getrawcode(func).co_flags & 32 # generator function
|
|
|
|
except AttributeError: # builtin functions have no bytecode
|
|
|
|
# assume them to not be generators
|
|
|
|
return False
|
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
class PyobjContext(object):
|
|
|
|
module = pyobj_property("Module")
|
|
|
|
cls = pyobj_property("Class")
|
|
|
|
instance = pyobj_property("Instance")
|
2012-06-25 23:35:33 +08:00
|
|
|
|
2012-07-19 01:49:14 +08:00
|
|
|
class PyobjMixin(PyobjContext):
|
2010-07-27 03:15:15 +08:00
|
|
|
def obj():
|
2008-08-16 23:26:59 +08:00
|
|
|
def fget(self):
|
2010-07-27 03:15:15 +08:00
|
|
|
try:
|
|
|
|
return self._obj
|
|
|
|
except AttributeError:
|
|
|
|
self._obj = obj = self._getobj()
|
|
|
|
return obj
|
|
|
|
def fset(self, value):
|
|
|
|
self._obj = value
|
|
|
|
return property(fget, fset, None, "underlying python object")
|
2008-08-16 23:26:59 +08:00
|
|
|
obj = obj()
|
|
|
|
|
|
|
|
def _getobj(self):
|
|
|
|
return getattr(self.parent.obj, self.name)
|
|
|
|
|
2009-02-27 18:18:27 +08:00
|
|
|
def getmodpath(self, stopatmodule=True, includemodule=False):
|
2008-08-16 23:26:59 +08:00
|
|
|
""" return python path relative to the containing module. """
|
|
|
|
chain = self.listchain()
|
|
|
|
chain.reverse()
|
|
|
|
parts = []
|
|
|
|
for node in chain:
|
|
|
|
if isinstance(node, Instance):
|
|
|
|
continue
|
2010-07-27 03:15:15 +08:00
|
|
|
name = node.name
|
2008-09-02 22:31:42 +08:00
|
|
|
if isinstance(node, Module):
|
|
|
|
assert name.endswith(".py")
|
|
|
|
name = name[:-3]
|
2009-02-27 18:18:27 +08:00
|
|
|
if stopatmodule:
|
|
|
|
if includemodule:
|
|
|
|
parts.append(name)
|
|
|
|
break
|
2008-09-02 22:31:42 +08:00
|
|
|
parts.append(name)
|
2008-08-16 23:26:59 +08:00
|
|
|
parts.reverse()
|
|
|
|
s = ".".join(parts)
|
|
|
|
return s.replace(".[", "[")
|
|
|
|
|
2009-05-12 19:39:09 +08:00
|
|
|
def _getfslineno(self):
|
2012-07-20 20:16:28 +08:00
|
|
|
return getfslineno(self.obj)
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2009-05-12 23:02:22 +08:00
|
|
|
def reportinfo(self):
|
2010-11-07 06:45:48 +08:00
|
|
|
# XXX caching?
|
2010-11-05 06:21:23 +08:00
|
|
|
obj = self.obj
|
|
|
|
if hasattr(obj, 'compat_co_firstlineno'):
|
|
|
|
# nose compatibility
|
|
|
|
fspath = sys.modules[obj.__module__].__file__
|
|
|
|
if fspath.endswith(".pyc"):
|
|
|
|
fspath = fspath[:-1]
|
|
|
|
lineno = obj.compat_co_firstlineno
|
|
|
|
else:
|
2012-07-20 20:16:28 +08:00
|
|
|
fspath, lineno = getfslineno(obj)
|
2013-12-03 18:23:22 +08:00
|
|
|
modpath = self.getmodpath()
|
2011-12-14 18:56:51 +08:00
|
|
|
assert isinstance(lineno, int)
|
2010-07-27 03:15:15 +08:00
|
|
|
return fspath, lineno, modpath
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
class PyCollector(PyobjMixin, pytest.Collector):
|
2010-07-27 03:15:15 +08:00
|
|
|
|
|
|
|
def funcnamefilter(self, name):
|
2010-11-25 05:01:04 +08:00
|
|
|
for prefix in self.config.getini("python_functions"):
|
|
|
|
if name.startswith(prefix):
|
|
|
|
return True
|
2011-01-18 19:47:31 +08:00
|
|
|
|
2010-07-27 03:15:15 +08:00
|
|
|
def classnamefilter(self, name):
|
2010-11-25 05:01:04 +08:00
|
|
|
for prefix in self.config.getini("python_classes"):
|
|
|
|
if name.startswith(prefix):
|
|
|
|
return True
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2008-09-02 16:58:14 +08:00
|
|
|
def collect(self):
|
2008-08-16 23:26:59 +08:00
|
|
|
# NB. we avoid random getattrs and peek in the __dict__ instead
|
2010-10-26 05:08:56 +08:00
|
|
|
# (XXX originally introduced from a PyPy need, still true?)
|
2008-08-16 23:26:59 +08:00
|
|
|
dicts = [getattr(self.obj, '__dict__', {})]
|
2009-09-01 22:10:21 +08:00
|
|
|
for basecls in inspect.getmro(self.obj.__class__):
|
2008-08-16 23:26:59 +08:00
|
|
|
dicts.append(basecls.__dict__)
|
|
|
|
seen = {}
|
2010-01-16 00:38:09 +08:00
|
|
|
l = []
|
2008-08-16 23:26:59 +08:00
|
|
|
for dic in dicts:
|
|
|
|
for name, obj in dic.items():
|
|
|
|
if name in seen:
|
|
|
|
continue
|
|
|
|
seen[name] = True
|
2012-11-28 00:58:08 +08:00
|
|
|
res = self.makeitem(name, obj)
|
|
|
|
if res is None:
|
|
|
|
continue
|
|
|
|
if not isinstance(res, list):
|
|
|
|
res = [res]
|
|
|
|
l.extend(res)
|
2010-01-16 00:38:09 +08:00
|
|
|
l.sort(key=lambda item: item.reportinfo()[:2])
|
|
|
|
return l
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2009-02-27 18:18:27 +08:00
|
|
|
def makeitem(self, name, obj):
|
2012-10-22 22:12:22 +08:00
|
|
|
#assert self.ihook.fspath == self.fspath, self
|
2009-12-30 09:36:58 +08:00
|
|
|
return self.ihook.pytest_pycollect_makeitem(
|
2009-05-08 05:12:17 +08:00
|
|
|
collector=self, name=name, obj=obj)
|
2009-05-12 01:23:57 +08:00
|
|
|
|
|
|
|
def _genfunctions(self, name, funcobj):
|
2009-05-20 00:13:33 +08:00
|
|
|
module = self.getparent(Module).obj
|
|
|
|
clscol = self.getparent(Class)
|
2009-05-12 01:23:57 +08:00
|
|
|
cls = clscol and clscol.obj or None
|
2011-12-29 00:47:08 +08:00
|
|
|
transfer_markers(funcobj, cls, module)
|
2012-10-17 17:20:45 +08:00
|
|
|
fm = self.session._fixturemanager
|
|
|
|
fixtureinfo = fm.getfixtureinfo(self, funcobj, cls)
|
2012-10-16 19:47:59 +08:00
|
|
|
metafunc = Metafunc(funcobj, fixtureinfo, self.config,
|
|
|
|
cls=cls, module=module)
|
2009-12-30 09:36:58 +08:00
|
|
|
gentesthook = self.config.hook.pytest_generate_tests
|
2010-11-25 05:22:52 +08:00
|
|
|
extra = [module]
|
|
|
|
if cls is not None:
|
|
|
|
extra.append(cls())
|
|
|
|
plugins = self.getplugins() + extra
|
2009-12-30 21:05:41 +08:00
|
|
|
gentesthook.pcall(plugins, metafunc=metafunc)
|
2011-03-05 21:16:27 +08:00
|
|
|
Function = self._getcustomclass("Function")
|
2012-07-19 15:20:14 +08:00
|
|
|
if not metafunc._calls:
|
2012-10-06 17:34:06 +08:00
|
|
|
yield Function(name, parent=self)
|
|
|
|
else:
|
|
|
|
for callspec in metafunc._calls:
|
|
|
|
subname = "%s[%s]" %(name, callspec.id)
|
|
|
|
yield Function(name=subname, parent=self,
|
|
|
|
callspec=callspec, callobj=funcobj,
|
|
|
|
keywords={callspec.id:True})
|
2010-07-27 03:15:15 +08:00
|
|
|
|
2012-10-16 19:47:59 +08:00
|
|
|
|
|
|
|
class FuncFixtureInfo:
|
2012-10-16 20:19:38 +08:00
|
|
|
def __init__(self, argnames, names_closure, name2fixturedefs):
|
|
|
|
self.argnames = argnames
|
2012-10-16 19:47:59 +08:00
|
|
|
self.names_closure = names_closure
|
2012-10-16 19:59:12 +08:00
|
|
|
self.name2fixturedefs = name2fixturedefs
|
2012-10-16 19:48:00 +08:00
|
|
|
|
2011-12-29 00:47:08 +08:00
|
|
|
def transfer_markers(funcobj, cls, mod):
|
2011-12-28 23:47:19 +08:00
|
|
|
# XXX this should rather be code in the mark plugin or the mark
|
|
|
|
# plugin should merge with the python plugin.
|
2011-12-29 00:47:08 +08:00
|
|
|
for holder in (cls, mod):
|
2011-12-28 23:47:19 +08:00
|
|
|
try:
|
|
|
|
pytestmark = holder.pytestmark
|
|
|
|
except AttributeError:
|
|
|
|
continue
|
|
|
|
if isinstance(pytestmark, list):
|
|
|
|
for mark in pytestmark:
|
2011-12-29 00:47:08 +08:00
|
|
|
mark(funcobj)
|
2011-12-28 23:47:19 +08:00
|
|
|
else:
|
2011-12-29 00:47:08 +08:00
|
|
|
pytestmark(funcobj)
|
2011-03-05 21:16:27 +08:00
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
class Module(pytest.File, PyCollector):
|
2012-06-25 23:35:33 +08:00
|
|
|
""" Collector for test classes and functions. """
|
2008-08-16 23:26:59 +08:00
|
|
|
def _getobj(self):
|
2012-09-24 17:36:24 +08:00
|
|
|
return self._memoizedcall('_obj', self._importtestmodule)
|
2009-02-27 18:18:27 +08:00
|
|
|
|
2012-07-20 20:16:28 +08:00
|
|
|
def collect(self):
|
2012-10-16 22:13:12 +08:00
|
|
|
self.session._fixturemanager.parsefactories(self)
|
2012-07-20 20:16:28 +08:00
|
|
|
return super(Module, self).collect()
|
|
|
|
|
2009-02-27 18:18:27 +08:00
|
|
|
def _importtestmodule(self):
|
2010-07-27 03:15:15 +08:00
|
|
|
# we assume we are only called once per module
|
2010-07-04 23:06:50 +08:00
|
|
|
try:
|
2011-06-29 10:13:12 +08:00
|
|
|
mod = self.fspath.pyimport(ensuresyspath=True)
|
2010-07-04 23:06:50 +08:00
|
|
|
except SyntaxError:
|
|
|
|
excinfo = py.code.ExceptionInfo()
|
|
|
|
raise self.CollectError(excinfo.getrepr(style="short"))
|
|
|
|
except self.fspath.ImportMismatchError:
|
|
|
|
e = sys.exc_info()[1]
|
|
|
|
raise self.CollectError(
|
|
|
|
"import file mismatch:\n"
|
2010-07-27 03:15:15 +08:00
|
|
|
"imported module %r has this __file__ attribute:\n"
|
2010-07-04 23:06:50 +08:00
|
|
|
" %s\n"
|
|
|
|
"which is not the same as the test file we want to collect:\n"
|
|
|
|
" %s\n"
|
2011-12-20 20:20:59 +08:00
|
|
|
"HINT: remove __pycache__ / .pyc files and/or use a "
|
|
|
|
"unique basename for your test file modules"
|
2010-07-04 23:06:50 +08:00
|
|
|
% e.args
|
|
|
|
)
|
2009-02-27 18:18:27 +08:00
|
|
|
#print "imported test module", mod
|
2009-04-09 22:03:09 +08:00
|
|
|
self.config.pluginmanager.consider_module(mod)
|
2009-02-27 18:18:27 +08:00
|
|
|
return mod
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2010-07-27 03:15:15 +08:00
|
|
|
def setup(self):
|
2013-06-20 22:43:42 +08:00
|
|
|
setup_module = xunitsetup(self.obj, "setUpModule")
|
|
|
|
if setup_module is None:
|
|
|
|
setup_module = xunitsetup(self.obj, "setup_module")
|
2012-08-13 19:37:14 +08:00
|
|
|
if setup_module is not None:
|
2009-10-23 22:17:06 +08:00
|
|
|
#XXX: nose compat hack, move to nose plugin
|
2010-10-07 17:59:00 +08:00
|
|
|
# if it takes a positional arg, its probably a pytest style one
|
2009-10-23 22:17:06 +08:00
|
|
|
# so we pass the current module object
|
2012-08-13 19:37:14 +08:00
|
|
|
if inspect.getargspec(setup_module)[0]:
|
|
|
|
setup_module(self.obj)
|
2009-10-23 22:17:06 +08:00
|
|
|
else:
|
2012-08-13 19:37:14 +08:00
|
|
|
setup_module()
|
2013-08-02 15:52:40 +08:00
|
|
|
fin = getattr(self.obj, 'tearDownModule', None)
|
|
|
|
if fin is None:
|
|
|
|
fin = getattr(self.obj, 'teardown_module', None)
|
|
|
|
if fin is not None:
|
2009-10-23 22:17:06 +08:00
|
|
|
#XXX: nose compat hack, move to nose plugin
|
|
|
|
# if it takes a positional arg, its probably a py.test style one
|
|
|
|
# so we pass the current module object
|
2013-08-02 15:52:40 +08:00
|
|
|
if inspect.getargspec(fin)[0]:
|
|
|
|
finalizer = lambda: fin(self.obj)
|
2009-10-23 22:17:06 +08:00
|
|
|
else:
|
2013-08-02 15:52:40 +08:00
|
|
|
finalizer = fin
|
|
|
|
self.addfinalizer(finalizer)
|
|
|
|
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
class Class(PyCollector):
|
2012-06-25 23:35:33 +08:00
|
|
|
""" Collector for test methods. """
|
2008-09-02 16:58:14 +08:00
|
|
|
def collect(self):
|
2013-02-15 17:18:00 +08:00
|
|
|
if hasinit(self.obj):
|
|
|
|
pytest.skip("class %s.%s with __init__ won't get collected" % (
|
|
|
|
self.obj.__module__,
|
|
|
|
self.obj.__name__,
|
|
|
|
))
|
2011-03-05 21:16:27 +08:00
|
|
|
return [self._getcustomclass("Instance")(name="()", parent=self)]
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2010-07-27 03:15:15 +08:00
|
|
|
def setup(self):
|
2012-08-13 19:37:14 +08:00
|
|
|
setup_class = xunitsetup(self.obj, 'setup_class')
|
2010-07-27 03:15:15 +08:00
|
|
|
if setup_class is not None:
|
|
|
|
setup_class = getattr(setup_class, 'im_func', setup_class)
|
2012-05-07 05:03:16 +08:00
|
|
|
setup_class = getattr(setup_class, '__func__', setup_class)
|
2010-07-27 03:15:15 +08:00
|
|
|
setup_class(self.obj)
|
|
|
|
|
2013-08-02 15:52:40 +08:00
|
|
|
fin_class = getattr(self.obj, 'teardown_class', None)
|
|
|
|
if fin_class is not None:
|
|
|
|
fin_class = getattr(fin_class, 'im_func', fin_class)
|
|
|
|
fin_class = getattr(fin_class, '__func__', fin_class)
|
|
|
|
self.addfinalizer(lambda: fin_class(self.obj))
|
2010-07-27 03:15:15 +08:00
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
class Instance(PyCollector):
|
2010-07-27 03:15:15 +08:00
|
|
|
def _getobj(self):
|
2012-07-20 20:16:28 +08:00
|
|
|
obj = self.parent.obj()
|
|
|
|
return obj
|
|
|
|
|
|
|
|
def collect(self):
|
2012-10-16 22:13:12 +08:00
|
|
|
self.session._fixturemanager.parsefactories(self)
|
2012-07-20 20:16:28 +08:00
|
|
|
return super(Instance, self).collect()
|
2010-09-26 00:23:26 +08:00
|
|
|
|
2010-07-27 03:15:15 +08:00
|
|
|
def newinstance(self):
|
2008-08-16 23:26:59 +08:00
|
|
|
self.obj = self._getobj()
|
|
|
|
return self.obj
|
|
|
|
|
|
|
|
class FunctionMixin(PyobjMixin):
|
|
|
|
""" mixin for the code common to Function and Generator.
|
|
|
|
"""
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2010-07-27 03:15:15 +08:00
|
|
|
def setup(self):
|
2008-08-16 23:26:59 +08:00
|
|
|
""" perform setup for this test function. """
|
2011-01-14 20:30:36 +08:00
|
|
|
if hasattr(self, '_preservedparent'):
|
|
|
|
obj = self._preservedparent
|
|
|
|
elif isinstance(self.parent, Instance):
|
2008-08-16 23:26:59 +08:00
|
|
|
obj = self.parent.newinstance()
|
|
|
|
self.obj = self._getobj()
|
|
|
|
else:
|
2010-07-27 03:15:15 +08:00
|
|
|
obj = self.parent.obj
|
2011-03-03 01:03:43 +08:00
|
|
|
if inspect.ismethod(self.obj):
|
2013-08-02 15:52:40 +08:00
|
|
|
setup_name = 'setup_method'
|
|
|
|
teardown_name = 'teardown_method'
|
2011-03-03 01:03:43 +08:00
|
|
|
else:
|
2013-08-02 15:52:40 +08:00
|
|
|
setup_name = 'setup_function'
|
|
|
|
teardown_name = 'teardown_function'
|
|
|
|
setup_func_or_method = xunitsetup(obj, setup_name)
|
2010-07-27 03:15:15 +08:00
|
|
|
if setup_func_or_method is not None:
|
|
|
|
setup_func_or_method(self.obj)
|
2013-08-02 15:52:40 +08:00
|
|
|
fin = getattr(obj, teardown_name, None)
|
|
|
|
if fin is not None:
|
|
|
|
self.addfinalizer(lambda: fin(self.obj))
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2010-11-06 06:37:31 +08:00
|
|
|
def _prunetraceback(self, excinfo):
|
2010-07-27 03:15:15 +08:00
|
|
|
if hasattr(self, '_obj') and not self.config.option.fulltrace:
|
|
|
|
code = py.code.Code(self.obj)
|
|
|
|
path, firstlineno = code.path, code.firstlineno
|
2010-11-06 06:37:31 +08:00
|
|
|
traceback = excinfo.traceback
|
2008-08-16 23:26:59 +08:00
|
|
|
ntraceback = traceback.cut(path=path, firstlineno=firstlineno)
|
|
|
|
if ntraceback == traceback:
|
|
|
|
ntraceback = ntraceback.cut(path=path)
|
2009-03-01 19:24:52 +08:00
|
|
|
if ntraceback == traceback:
|
2010-10-10 19:48:49 +08:00
|
|
|
ntraceback = ntraceback.cut(excludepath=cutdir)
|
2010-11-06 06:37:31 +08:00
|
|
|
excinfo.traceback = ntraceback.filter()
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2010-05-22 22:18:24 +08:00
|
|
|
def _repr_failure_py(self, excinfo, style="long"):
|
2010-11-24 23:43:55 +08:00
|
|
|
if excinfo.errisinstance(pytest.fail.Exception):
|
|
|
|
if not excinfo.value.pytrace:
|
|
|
|
return str(excinfo.value)
|
2010-07-27 03:15:15 +08:00
|
|
|
return super(FunctionMixin, self)._repr_failure_py(excinfo,
|
2010-05-22 22:18:24 +08:00
|
|
|
style=style)
|
2010-02-04 23:01:02 +08:00
|
|
|
|
2009-07-26 00:09:01 +08:00
|
|
|
def repr_failure(self, excinfo, outerr=None):
|
|
|
|
assert outerr is None, "XXX outerr usage is deprecated"
|
2010-07-27 03:15:15 +08:00
|
|
|
return self._repr_failure_py(excinfo,
|
2010-11-01 06:28:31 +08:00
|
|
|
style=self.config.option.tbstyle)
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2011-11-19 00:01:29 +08:00
|
|
|
|
2012-07-16 16:46:44 +08:00
|
|
|
class Generator(FunctionMixin, PyCollector):
|
2008-09-02 16:58:14 +08:00
|
|
|
def collect(self):
|
2010-07-27 03:15:15 +08:00
|
|
|
# test generators are seen as collectors but they also
|
|
|
|
# invoke setup/teardown on popular request
|
2009-05-07 21:44:56 +08:00
|
|
|
# (induced by the common "test_*" naming shared with normal tests)
|
2011-05-27 06:08:56 +08:00
|
|
|
self.session._setupstate.prepare(self)
|
2011-01-14 20:30:36 +08:00
|
|
|
# see FunctionMixin.setup and test_setupstate_is_preserved_134
|
|
|
|
self._preservedparent = self.parent.obj
|
2008-09-02 16:58:14 +08:00
|
|
|
l = []
|
2009-03-19 22:34:33 +08:00
|
|
|
seen = {}
|
2010-07-27 03:15:15 +08:00
|
|
|
for i, x in enumerate(self.obj()):
|
2008-11-26 00:10:16 +08:00
|
|
|
name, call, args = self.getcallargs(x)
|
2012-10-19 16:07:11 +08:00
|
|
|
if not callable(call):
|
2008-08-16 23:26:59 +08:00
|
|
|
raise TypeError("%r yielded non callable test %r" %(self.obj, call,))
|
2008-11-26 00:10:16 +08:00
|
|
|
if name is None:
|
|
|
|
name = "[%d]" % i
|
|
|
|
else:
|
|
|
|
name = "['%s']" % name
|
2009-03-19 22:34:33 +08:00
|
|
|
if name in seen:
|
|
|
|
raise ValueError("%r generated tests with non-unique name %r" %(self, name))
|
|
|
|
seen[name] = True
|
2011-01-18 19:47:31 +08:00
|
|
|
l.append(self.Function(name, self, args=args, callobj=call))
|
2008-09-02 16:58:14 +08:00
|
|
|
return l
|
2010-07-27 03:15:15 +08:00
|
|
|
|
2008-08-16 23:26:59 +08:00
|
|
|
def getcallargs(self, obj):
|
2008-11-26 00:10:16 +08:00
|
|
|
if not isinstance(obj, (tuple, list)):
|
|
|
|
obj = (obj,)
|
|
|
|
# explict naming
|
2009-08-29 22:40:03 +08:00
|
|
|
if isinstance(obj[0], py.builtin._basestring):
|
2008-11-26 00:10:16 +08:00
|
|
|
name = obj[0]
|
|
|
|
obj = obj[1:]
|
2008-08-16 23:26:59 +08:00
|
|
|
else:
|
2008-11-26 00:10:16 +08:00
|
|
|
name = None
|
|
|
|
call, args = obj[0], obj[1:]
|
2010-07-27 03:15:15 +08:00
|
|
|
return name, call, args
|
|
|
|
|
2008-08-16 23:26:59 +08:00
|
|
|
|
2009-08-07 00:15:21 +08:00
|
|
|
def hasinit(obj):
|
|
|
|
init = getattr(obj, '__init__', None)
|
|
|
|
if init:
|
2010-05-09 18:27:04 +08:00
|
|
|
if init != object.__init__:
|
2009-08-07 00:15:21 +08:00
|
|
|
return True
|
2010-09-26 00:23:26 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
def fillfixtures(function):
|
2012-10-01 15:23:39 +08:00
|
|
|
""" fill missing funcargs for a test function. """
|
2012-11-20 05:17:59 +08:00
|
|
|
if 1 or getattr(function, "_args", None) is None: # not a yielded function
|
2012-07-19 01:49:14 +08:00
|
|
|
try:
|
|
|
|
request = function._request
|
|
|
|
except AttributeError:
|
2012-10-17 17:20:45 +08:00
|
|
|
# XXX this special code path is only expected to execute
|
|
|
|
# with the oejskit plugin. It uses classes with funcargs
|
|
|
|
# and we thus have to work a bit to allow this.
|
|
|
|
fm = function.session._fixturemanager
|
|
|
|
fi = fm.getfixtureinfo(function.parent, function.obj, None)
|
2012-10-16 19:48:00 +08:00
|
|
|
function._fixtureinfo = fi
|
2012-10-05 20:24:44 +08:00
|
|
|
request = function._request = FixtureRequest(function)
|
2012-10-17 17:20:45 +08:00
|
|
|
request._fillfixtures()
|
|
|
|
# prune out funcargs for jstests
|
|
|
|
newfuncargs = {}
|
|
|
|
for name in fi.argnames:
|
|
|
|
newfuncargs[name] = function.funcargs[name]
|
|
|
|
function.funcargs = newfuncargs
|
|
|
|
else:
|
|
|
|
request._fillfixtures()
|
|
|
|
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2010-09-26 00:23:26 +08:00
|
|
|
_notexists = object()
|
2011-11-17 19:09:21 +08:00
|
|
|
|
|
|
|
class CallSpec2(object):
|
|
|
|
def __init__(self, metafunc):
|
|
|
|
self.metafunc = metafunc
|
|
|
|
self.funcargs = {}
|
|
|
|
self._idlist = []
|
|
|
|
self.params = {}
|
|
|
|
self._globalid = _notexists
|
|
|
|
self._globalid_args = set()
|
|
|
|
self._globalparam = _notexists
|
2012-07-30 16:46:03 +08:00
|
|
|
self._arg2scopenum = {} # used for sorting parametrized resources
|
2013-05-17 16:46:36 +08:00
|
|
|
self.keywords = {}
|
2011-11-17 19:09:21 +08:00
|
|
|
|
|
|
|
def copy(self, metafunc):
|
|
|
|
cs = CallSpec2(self.metafunc)
|
|
|
|
cs.funcargs.update(self.funcargs)
|
|
|
|
cs.params.update(self.params)
|
2013-05-17 16:46:36 +08:00
|
|
|
cs.keywords.update(self.keywords)
|
2012-07-30 16:46:03 +08:00
|
|
|
cs._arg2scopenum.update(self._arg2scopenum)
|
2011-11-17 19:09:21 +08:00
|
|
|
cs._idlist = list(self._idlist)
|
|
|
|
cs._globalid = self._globalid
|
|
|
|
cs._globalid_args = self._globalid_args
|
|
|
|
cs._globalparam = self._globalparam
|
|
|
|
return cs
|
|
|
|
|
|
|
|
def _checkargnotcontained(self, arg):
|
|
|
|
if arg in self.params or arg in self.funcargs:
|
|
|
|
raise ValueError("duplicate %r" %(arg,))
|
|
|
|
|
|
|
|
def getparam(self, name):
|
|
|
|
try:
|
|
|
|
return self.params[name]
|
|
|
|
except KeyError:
|
|
|
|
if self._globalparam is _notexists:
|
|
|
|
raise ValueError(name)
|
|
|
|
return self._globalparam
|
|
|
|
|
|
|
|
@property
|
|
|
|
def id(self):
|
2011-12-28 23:47:19 +08:00
|
|
|
return "-".join(map(str, filter(None, self._idlist)))
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2013-05-17 16:46:36 +08:00
|
|
|
def setmulti(self, valtype, argnames, valset, id, keywords, scopenum=0):
|
2011-11-17 19:09:21 +08:00
|
|
|
for arg,val in zip(argnames, valset):
|
|
|
|
self._checkargnotcontained(arg)
|
|
|
|
getattr(self, valtype)[arg] = val
|
2012-09-21 15:39:54 +08:00
|
|
|
# we want self.params to be always set because of
|
|
|
|
# parametrize_sorted() which groups tests by params/scope
|
|
|
|
if valtype == "funcargs":
|
|
|
|
self.params[arg] = id
|
2012-07-30 16:46:03 +08:00
|
|
|
self._arg2scopenum[arg] = scopenum
|
2012-10-28 21:52:43 +08:00
|
|
|
if val is _notexists:
|
2012-10-06 17:34:06 +08:00
|
|
|
self._emptyparamspecified = True
|
2011-11-17 19:09:21 +08:00
|
|
|
self._idlist.append(id)
|
2013-05-17 16:46:36 +08:00
|
|
|
self.keywords.update(keywords)
|
2011-11-17 19:09:21 +08:00
|
|
|
|
|
|
|
def setall(self, funcargs, id, param):
|
|
|
|
for x in funcargs:
|
|
|
|
self._checkargnotcontained(x)
|
|
|
|
self.funcargs.update(funcargs)
|
|
|
|
if id is not _notexists:
|
|
|
|
self._idlist.append(id)
|
2010-09-26 00:23:26 +08:00
|
|
|
if param is not _notexists:
|
2011-11-17 19:09:21 +08:00
|
|
|
assert self._globalparam is _notexists
|
|
|
|
self._globalparam = param
|
|
|
|
|
2010-09-26 00:23:26 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class FuncargnamesCompatAttr:
|
|
|
|
""" helper class so that Metafunc, Function and FixtureRequest
|
|
|
|
don't need to each define the "funcargnames" compatibility attribute.
|
|
|
|
"""
|
|
|
|
@property
|
|
|
|
def funcargnames(self):
|
|
|
|
""" alias attribute for ``fixturenames`` for pre-2.3 compatibility"""
|
|
|
|
return self.fixturenames
|
|
|
|
|
|
|
|
class Metafunc(FuncargnamesCompatAttr):
|
2012-10-16 19:47:59 +08:00
|
|
|
def __init__(self, function, fixtureinfo, config, cls=None, module=None):
|
|
|
|
self.config = config
|
2010-09-26 00:23:26 +08:00
|
|
|
self.module = module
|
|
|
|
self.function = function
|
2012-10-16 19:47:59 +08:00
|
|
|
self.fixturenames = fixtureinfo.names_closure
|
2012-10-16 19:59:12 +08:00
|
|
|
self._arg2fixturedefs = fixtureinfo.name2fixturedefs
|
2010-09-26 00:23:26 +08:00
|
|
|
self.cls = cls
|
|
|
|
self.module = module
|
|
|
|
self._calls = []
|
|
|
|
self._ids = py.builtin.set()
|
|
|
|
|
2012-07-30 16:46:03 +08:00
|
|
|
def parametrize(self, argnames, argvalues, indirect=False, ids=None,
|
2012-10-07 03:01:13 +08:00
|
|
|
scope=None):
|
2011-12-05 18:10:48 +08:00
|
|
|
""" Add new invocations to the underlying test function using the list
|
|
|
|
of argvalues for the given argnames. Parametrization is performed
|
2011-11-20 07:45:05 +08:00
|
|
|
during the collection phase. If you need to setup expensive resources
|
2012-10-09 20:35:17 +08:00
|
|
|
see about setting indirect=True to do it rather at test setup time.
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2013-05-28 16:32:54 +08:00
|
|
|
:arg argnames: a comma-separated string denoting one or more argument
|
|
|
|
names, or a list/tuple of argument strings.
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2013-05-22 21:24:58 +08:00
|
|
|
:arg argvalues: The list of argvalues determines how often a
|
|
|
|
test is invoked with different argument values. If only one
|
|
|
|
argname was specified argvalues is a list of simple values. If N
|
|
|
|
argnames were specified, argvalues must be a list of N-tuples,
|
|
|
|
where each tuple-element specifies a value for its respective
|
|
|
|
argname.
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2012-10-09 20:35:17 +08:00
|
|
|
:arg indirect: if True each argvalue corresponding to an argname will
|
|
|
|
be passed as request.param to its respective argname fixture
|
|
|
|
function so that it can perform more expensive setups during the
|
|
|
|
setup phase of a test rather than at collection time.
|
2011-11-17 19:09:21 +08:00
|
|
|
|
2011-12-05 18:10:48 +08:00
|
|
|
:arg ids: list of string ids each corresponding to the argvalues so
|
|
|
|
that they are part of the test id. If no ids are provided they will
|
|
|
|
be generated automatically from the argvalues.
|
2012-10-07 03:01:13 +08:00
|
|
|
|
2012-10-09 20:35:17 +08:00
|
|
|
:arg scope: if specified it denotes the scope of the parameters.
|
|
|
|
The scope is used for grouping tests by parameter instances.
|
|
|
|
It will also override any fixture-function defined scope, allowing
|
|
|
|
to set a dynamic scope using test context or configuration.
|
2011-11-17 19:09:21 +08:00
|
|
|
"""
|
2013-05-22 21:24:58 +08:00
|
|
|
|
2013-11-02 12:10:13 +08:00
|
|
|
# individual parametrized argument sets can be wrapped in a series
|
|
|
|
# of markers in which case we unwrap the values and apply the mark
|
2013-05-22 21:24:58 +08:00
|
|
|
# at Function init
|
2013-05-17 16:46:36 +08:00
|
|
|
newkeywords = {}
|
2013-05-22 21:24:58 +08:00
|
|
|
unwrapped_argvalues = []
|
2013-05-17 16:46:36 +08:00
|
|
|
for i, argval in enumerate(argvalues):
|
2013-11-02 12:10:13 +08:00
|
|
|
while isinstance(argval, MarkDecorator):
|
2013-05-22 21:24:58 +08:00
|
|
|
newmark = MarkDecorator(argval.markname,
|
|
|
|
argval.args[:-1], argval.kwargs)
|
2013-11-02 12:10:13 +08:00
|
|
|
newmarks = newkeywords.setdefault(i, {})
|
|
|
|
newmarks[newmark.markname] = newmark
|
2013-05-22 21:24:58 +08:00
|
|
|
argval = argval.args[-1]
|
|
|
|
unwrapped_argvalues.append(argval)
|
|
|
|
argvalues = unwrapped_argvalues
|
2013-05-17 16:46:36 +08:00
|
|
|
|
2013-05-20 10:52:20 +08:00
|
|
|
if not isinstance(argnames, (tuple, list)):
|
2013-05-28 16:32:54 +08:00
|
|
|
argnames = [x.strip() for x in argnames.split(",") if x.strip()]
|
2013-10-02 14:08:26 +08:00
|
|
|
if len(argnames) == 1:
|
|
|
|
argvalues = [(val,) for val in argvalues]
|
2013-05-20 10:52:20 +08:00
|
|
|
if not argvalues:
|
|
|
|
argvalues = [(_notexists,) * len(argnames)]
|
2013-05-17 16:46:36 +08:00
|
|
|
|
2012-07-30 18:39:45 +08:00
|
|
|
if scope is None:
|
2012-10-07 03:01:13 +08:00
|
|
|
scope = "subfunction"
|
2012-07-30 16:46:03 +08:00
|
|
|
scopenum = scopes.index(scope)
|
2012-02-03 23:54:00 +08:00
|
|
|
if not indirect:
|
|
|
|
#XXX should we also check for the opposite case?
|
|
|
|
for arg in argnames:
|
2012-10-05 20:24:44 +08:00
|
|
|
if arg not in self.fixturenames:
|
2012-10-06 17:34:06 +08:00
|
|
|
raise ValueError("%r uses no fixture %r" %(
|
|
|
|
self.function, arg))
|
2011-11-17 19:09:21 +08:00
|
|
|
valtype = indirect and "params" or "funcargs"
|
2013-05-29 10:59:47 +08:00
|
|
|
if ids and len(ids) != len(argvalues):
|
|
|
|
raise ValueError('%d tests specified with %d ids' %(
|
|
|
|
len(argvalues), len(ids)))
|
2011-11-17 19:09:21 +08:00
|
|
|
if not ids:
|
2012-10-19 16:07:13 +08:00
|
|
|
ids = idmaker(argnames, argvalues)
|
2011-11-17 19:09:21 +08:00
|
|
|
newcalls = []
|
|
|
|
for callspec in self._calls or [CallSpec2(self)]:
|
|
|
|
for i, valset in enumerate(argvalues):
|
|
|
|
assert len(valset) == len(argnames)
|
|
|
|
newcallspec = callspec.copy(self)
|
2012-07-30 16:46:03 +08:00
|
|
|
newcallspec.setmulti(valtype, argnames, valset, ids[i],
|
2013-05-22 21:24:58 +08:00
|
|
|
newkeywords.get(i, {}), scopenum)
|
2011-11-17 19:09:21 +08:00
|
|
|
newcalls.append(newcallspec)
|
|
|
|
self._calls = newcalls
|
|
|
|
|
2010-09-26 00:23:26 +08:00
|
|
|
def addcall(self, funcargs=None, id=_notexists, param=_notexists):
|
2011-12-05 18:10:48 +08:00
|
|
|
""" (deprecated, use parametrize) Add a new call to the underlying
|
|
|
|
test function during the collection phase of a test run. Note that
|
|
|
|
request.addcall() is called during the test collection phase prior and
|
|
|
|
independently to actual test execution. You should only use addcall()
|
|
|
|
if you need to specify multiple arguments of a test function.
|
2011-01-18 19:47:31 +08:00
|
|
|
|
2010-10-11 05:45:45 +08:00
|
|
|
:arg funcargs: argument keyword dictionary used when invoking
|
|
|
|
the test function.
|
2011-01-18 19:47:31 +08:00
|
|
|
|
2010-11-05 06:21:23 +08:00
|
|
|
:arg id: used for reporting and identification purposes. If you
|
2011-11-17 19:09:21 +08:00
|
|
|
don't supply an `id` an automatic unique id will be generated.
|
2010-10-11 05:45:45 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
:arg param: a parameter which will be exposed to a later fixture function
|
2011-11-17 19:09:21 +08:00
|
|
|
invocation through the ``request.param`` attribute.
|
2010-10-11 05:45:45 +08:00
|
|
|
"""
|
2010-09-26 00:23:26 +08:00
|
|
|
assert funcargs is None or isinstance(funcargs, dict)
|
2011-03-05 19:11:35 +08:00
|
|
|
if funcargs is not None:
|
|
|
|
for name in funcargs:
|
2012-10-05 20:24:44 +08:00
|
|
|
if name not in self.fixturenames:
|
2011-03-05 19:11:35 +08:00
|
|
|
pytest.fail("funcarg %r not used in this function." % name)
|
2011-11-17 19:09:21 +08:00
|
|
|
else:
|
|
|
|
funcargs = {}
|
2010-09-26 00:23:26 +08:00
|
|
|
if id is None:
|
|
|
|
raise ValueError("id=None not allowed")
|
|
|
|
if id is _notexists:
|
|
|
|
id = len(self._calls)
|
|
|
|
id = str(id)
|
|
|
|
if id in self._ids:
|
|
|
|
raise ValueError("duplicate id %r" % id)
|
|
|
|
self._ids.add(id)
|
2011-11-17 19:09:21 +08:00
|
|
|
|
|
|
|
cs = CallSpec2(self)
|
|
|
|
cs.setall(funcargs, id, param)
|
|
|
|
self._calls.append(cs)
|
|
|
|
|
2012-10-19 16:07:13 +08:00
|
|
|
def idmaker(argnames, argvalues):
|
|
|
|
idlist = []
|
|
|
|
for valindex, valset in enumerate(argvalues):
|
|
|
|
this_id = []
|
|
|
|
for nameindex, val in enumerate(valset):
|
2013-05-29 10:59:47 +08:00
|
|
|
if not isinstance(val, (float, int, str, bool, NoneType)):
|
2013-02-13 06:30:34 +08:00
|
|
|
this_id.append(str(argnames[nameindex])+str(valindex))
|
2012-10-19 16:07:13 +08:00
|
|
|
else:
|
|
|
|
this_id.append(str(val))
|
|
|
|
idlist.append("-".join(this_id))
|
|
|
|
return idlist
|
2010-09-26 00:23:26 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
def showfixtures(config):
|
2011-05-27 08:09:42 +08:00
|
|
|
from _pytest.main import wrap_session
|
2012-10-05 20:24:44 +08:00
|
|
|
return wrap_session(config, _showfixtures_main)
|
2011-05-27 08:09:42 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
def _showfixtures_main(config, session):
|
2010-11-07 17:19:58 +08:00
|
|
|
session.perform_collect()
|
2012-10-09 22:49:04 +08:00
|
|
|
curdir = py.path.local()
|
2010-11-07 17:19:58 +08:00
|
|
|
if session.items:
|
2012-10-09 22:49:04 +08:00
|
|
|
nodeid = session.items[0].nodeid
|
2010-11-06 16:58:04 +08:00
|
|
|
else:
|
2012-10-09 22:49:04 +08:00
|
|
|
part = session._initialparts[0]
|
|
|
|
nodeid = "::".join(map(str, [curdir.bestrelpath(part[0])] + part[1:]))
|
2012-10-17 19:42:40 +08:00
|
|
|
nodeid.replace(session.fspath.sep, "/")
|
2012-10-09 22:49:04 +08:00
|
|
|
|
2010-09-26 00:23:26 +08:00
|
|
|
tw = py.io.TerminalWriter()
|
|
|
|
verbose = config.getvalue("verbose")
|
2012-10-09 22:49:04 +08:00
|
|
|
|
|
|
|
fm = session._fixturemanager
|
|
|
|
|
|
|
|
available = []
|
2012-10-16 22:13:12 +08:00
|
|
|
for argname in fm._arg2fixturedefs:
|
2012-10-16 19:59:12 +08:00
|
|
|
fixturedefs = fm.getfixturedefs(argname, nodeid)
|
|
|
|
assert fixturedefs is not None
|
|
|
|
if not fixturedefs:
|
2012-10-09 22:49:04 +08:00
|
|
|
continue
|
2012-10-16 19:59:12 +08:00
|
|
|
fixturedef = fixturedefs[-1]
|
2012-10-09 22:49:04 +08:00
|
|
|
loc = getlocation(fixturedef.func, curdir)
|
|
|
|
available.append((len(fixturedef.baseid),
|
2012-10-17 18:57:05 +08:00
|
|
|
fixturedef.func.__module__,
|
2012-10-09 22:49:04 +08:00
|
|
|
curdir.bestrelpath(loc),
|
|
|
|
fixturedef.argname, fixturedef))
|
|
|
|
|
|
|
|
available.sort()
|
2012-10-17 18:57:05 +08:00
|
|
|
currentmodule = None
|
|
|
|
for baseid, module, bestrel, argname, fixturedef in available:
|
|
|
|
if currentmodule != module:
|
|
|
|
if not module.startswith("_pytest."):
|
|
|
|
tw.line()
|
|
|
|
tw.sep("-", "fixtures defined from %s" %(module,))
|
|
|
|
currentmodule = module
|
|
|
|
if verbose <= 0 and argname[0] == "_":
|
|
|
|
continue
|
2012-10-09 22:49:04 +08:00
|
|
|
if verbose > 0:
|
2012-10-17 19:12:33 +08:00
|
|
|
funcargspec = "%s -- %s" %(argname, bestrel,)
|
2012-10-09 22:49:04 +08:00
|
|
|
else:
|
2012-10-17 18:57:05 +08:00
|
|
|
funcargspec = argname
|
2012-10-09 22:49:04 +08:00
|
|
|
tw.line(funcargspec, green=True)
|
|
|
|
loc = getlocation(fixturedef.func, curdir)
|
|
|
|
doc = fixturedef.func.__doc__ or ""
|
|
|
|
if doc:
|
|
|
|
for line in doc.split("\n"):
|
|
|
|
tw.line(" " + line.strip())
|
|
|
|
else:
|
|
|
|
tw.line(" %s: no docstring available" %(loc,),
|
|
|
|
red=True)
|
2010-09-26 00:23:26 +08:00
|
|
|
|
|
|
|
def getlocation(function, curdir):
|
|
|
|
import inspect
|
|
|
|
fn = py.path.local(inspect.getfile(function))
|
|
|
|
lineno = py.builtin._getcode(function).co_firstlineno
|
|
|
|
if fn.relto(curdir):
|
|
|
|
fn = fn.relto(curdir)
|
|
|
|
return "%s:%d" %(fn, lineno+1)
|
2010-11-06 06:37:31 +08:00
|
|
|
|
|
|
|
# builtin pytest.raises helper
|
|
|
|
|
|
|
|
def raises(ExpectedException, *args, **kwargs):
|
2010-11-25 19:11:10 +08:00
|
|
|
""" assert that a code block/function call raises @ExpectedException
|
|
|
|
and raise a failure exception otherwise.
|
2011-01-18 19:47:31 +08:00
|
|
|
|
2013-08-01 17:19:47 +08:00
|
|
|
This helper produces a ``py.code.ExceptionInfo()`` object.
|
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
If using Python 2.5 or above, you may use this function as a
|
|
|
|
context manager::
|
2010-11-06 06:37:31 +08:00
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
>>> with raises(ZeroDivisionError):
|
|
|
|
... 1/0
|
2010-11-18 21:56:16 +08:00
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
Or you can specify a callable by passing a to-be-called lambda::
|
2010-11-18 21:56:16 +08:00
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
>>> raises(ZeroDivisionError, lambda: 1/0)
|
|
|
|
<ExceptionInfo ...>
|
2010-11-18 21:56:16 +08:00
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
or you can specify an arbitrary callable with arguments::
|
2010-11-18 21:56:16 +08:00
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
>>> def f(x): return 1/x
|
|
|
|
...
|
|
|
|
>>> raises(ZeroDivisionError, f, 0)
|
|
|
|
<ExceptionInfo ...>
|
|
|
|
>>> raises(ZeroDivisionError, f, x=0)
|
|
|
|
<ExceptionInfo ...>
|
2010-11-18 21:56:16 +08:00
|
|
|
|
2011-12-05 18:10:48 +08:00
|
|
|
A third possibility is to use a string to be executed::
|
2011-01-18 19:47:31 +08:00
|
|
|
|
2010-11-25 19:11:10 +08:00
|
|
|
>>> raises(ZeroDivisionError, "f(0)")
|
|
|
|
<ExceptionInfo ...>
|
2010-11-06 06:37:31 +08:00
|
|
|
"""
|
|
|
|
__tracebackhide__ = True
|
2012-08-19 19:45:26 +08:00
|
|
|
if ExpectedException is AssertionError:
|
|
|
|
# we want to catch a AssertionError
|
|
|
|
# replace our subclass with the builtin one
|
|
|
|
# see https://bitbucket.org/hpk42/pytest/issue/176/pytestraises
|
2012-08-29 04:35:06 +08:00
|
|
|
from _pytest.assertion.util import BuiltinAssertionError as ExpectedException
|
2010-11-06 06:37:31 +08:00
|
|
|
|
|
|
|
if not args:
|
|
|
|
return RaisesContext(ExpectedException)
|
|
|
|
elif isinstance(args[0], str):
|
|
|
|
code, = args
|
|
|
|
assert isinstance(code, str)
|
|
|
|
frame = sys._getframe(1)
|
|
|
|
loc = frame.f_locals.copy()
|
|
|
|
loc.update(kwargs)
|
|
|
|
#print "raises frame scope: %r" % frame.f_locals
|
|
|
|
try:
|
|
|
|
code = py.code.Source(code).compile()
|
|
|
|
py.builtin.exec_(code, frame.f_globals, loc)
|
|
|
|
# XXX didn'T mean f_globals == f_locals something special?
|
|
|
|
# this is destroyed here ...
|
|
|
|
except ExpectedException:
|
|
|
|
return py.code.ExceptionInfo()
|
|
|
|
else:
|
|
|
|
func = args[0]
|
|
|
|
try:
|
|
|
|
func(*args[1:], **kwargs)
|
|
|
|
except ExpectedException:
|
|
|
|
return py.code.ExceptionInfo()
|
|
|
|
pytest.fail("DID NOT RAISE")
|
|
|
|
|
|
|
|
class RaisesContext(object):
|
|
|
|
def __init__(self, ExpectedException):
|
|
|
|
self.ExpectedException = ExpectedException
|
|
|
|
self.excinfo = None
|
|
|
|
|
|
|
|
def __enter__(self):
|
|
|
|
self.excinfo = object.__new__(py.code.ExceptionInfo)
|
|
|
|
return self.excinfo
|
|
|
|
|
|
|
|
def __exit__(self, *tp):
|
|
|
|
__tracebackhide__ = True
|
|
|
|
if tp[0] is None:
|
|
|
|
pytest.fail("DID NOT RAISE")
|
|
|
|
self.excinfo.__init__(tp)
|
|
|
|
return issubclass(self.excinfo.type, self.ExpectedException)
|
2011-01-18 19:47:31 +08:00
|
|
|
|
2012-06-25 23:35:33 +08:00
|
|
|
#
|
|
|
|
# the basic py.test Function item
|
|
|
|
#
|
2013-11-21 20:15:32 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class Function(FunctionMixin, pytest.Item, FuncargnamesCompatAttr):
|
2012-06-25 23:35:33 +08:00
|
|
|
""" a Function Item is responsible for setting up and executing a
|
|
|
|
Python test function.
|
|
|
|
"""
|
|
|
|
_genid = None
|
2012-10-16 19:47:59 +08:00
|
|
|
def __init__(self, name, parent, args=None, config=None,
|
2013-11-21 20:15:32 +08:00
|
|
|
callspec=None, callobj=NOTSET, keywords=None, session=None):
|
2012-06-25 23:35:33 +08:00
|
|
|
super(Function, self).__init__(name, parent, config=config,
|
|
|
|
session=session)
|
|
|
|
self._args = args
|
2013-11-21 20:15:32 +08:00
|
|
|
if callobj is not NOTSET:
|
2012-10-16 19:48:00 +08:00
|
|
|
self.obj = callobj
|
|
|
|
|
|
|
|
for name, val in (py.builtin._getfuncdict(self.obj) or {}).items():
|
2012-10-18 19:52:32 +08:00
|
|
|
self.keywords[name] = val
|
2013-05-17 16:46:36 +08:00
|
|
|
if callspec:
|
|
|
|
for name, val in callspec.keywords.items():
|
|
|
|
self.keywords[name] = val
|
2012-10-16 19:48:00 +08:00
|
|
|
if keywords:
|
|
|
|
for name, val in keywords.items():
|
2012-10-18 19:52:32 +08:00
|
|
|
self.keywords[name] = val
|
2012-10-16 19:48:00 +08:00
|
|
|
|
2012-10-17 17:20:45 +08:00
|
|
|
fm = self.session._fixturemanager
|
2012-11-20 05:17:59 +08:00
|
|
|
isyield = self._isyieldedfunction()
|
|
|
|
self._fixtureinfo = fi = fm.getfixtureinfo(self.parent, self.obj,
|
|
|
|
self.cls,
|
|
|
|
funcargs=not isyield)
|
2012-10-16 19:48:00 +08:00
|
|
|
self.fixturenames = fi.names_closure
|
2013-04-22 16:35:48 +08:00
|
|
|
if callspec is not None:
|
|
|
|
self.callspec = callspec
|
|
|
|
self._initrequest()
|
|
|
|
|
|
|
|
def _initrequest(self):
|
|
|
|
if self._isyieldedfunction():
|
|
|
|
assert not hasattr(self, "callspec"), (
|
2012-06-25 23:35:33 +08:00
|
|
|
"yielded functions (deprecated) cannot have funcargs")
|
2012-11-20 05:17:59 +08:00
|
|
|
self.funcargs = {}
|
2012-06-25 23:35:33 +08:00
|
|
|
else:
|
2013-04-22 16:35:48 +08:00
|
|
|
if hasattr(self, "callspec"):
|
|
|
|
callspec = self.callspec
|
|
|
|
self.funcargs = callspec.funcargs.copy()
|
2012-06-25 23:35:33 +08:00
|
|
|
self._genid = callspec.id
|
|
|
|
if hasattr(callspec, "param"):
|
|
|
|
self.param = callspec.param
|
2012-07-19 01:49:14 +08:00
|
|
|
else:
|
|
|
|
self.funcargs = {}
|
2013-04-22 16:35:48 +08:00
|
|
|
self._request = FixtureRequest(self)
|
2012-10-05 16:21:35 +08:00
|
|
|
|
2012-06-25 23:35:33 +08:00
|
|
|
@property
|
|
|
|
def function(self):
|
|
|
|
"underlying python 'function' object"
|
|
|
|
return getattr(self.obj, 'im_func', self.obj)
|
|
|
|
|
|
|
|
def _getobj(self):
|
|
|
|
name = self.name
|
|
|
|
i = name.find("[") # parametrization
|
|
|
|
if i != -1:
|
|
|
|
name = name[:i]
|
|
|
|
return getattr(self.parent.obj, name)
|
|
|
|
|
|
|
|
@property
|
|
|
|
def _pyfuncitem(self):
|
|
|
|
"(compatonly) for code expecting pytest-2.2 style request objects"
|
|
|
|
return self
|
|
|
|
|
|
|
|
def _isyieldedfunction(self):
|
|
|
|
return getattr(self, "_args", None) is not None
|
|
|
|
|
|
|
|
def runtest(self):
|
|
|
|
""" execute the underlying test function. """
|
|
|
|
self.ihook.pytest_pyfunc_call(pyfuncitem=self)
|
|
|
|
|
|
|
|
def setup(self):
|
2012-10-06 17:34:06 +08:00
|
|
|
# check if parametrization happend with an empty list
|
|
|
|
try:
|
2013-10-12 21:39:22 +08:00
|
|
|
self.callspec._emptyparamspecified
|
2012-10-06 17:34:06 +08:00
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
fs, lineno = self._getfslineno()
|
|
|
|
pytest.skip("got empty parameter set, function %s at %s:%d" %(
|
|
|
|
self.function.__name__, fs, lineno))
|
2012-06-25 23:35:33 +08:00
|
|
|
super(Function, self).setup()
|
2012-10-05 20:24:44 +08:00
|
|
|
fillfixtures(self)
|
2012-06-25 23:35:33 +08:00
|
|
|
|
|
|
|
def __eq__(self, other):
|
|
|
|
try:
|
|
|
|
return (self.name == other.name and
|
|
|
|
self._args == other._args and
|
|
|
|
self.parent == other.parent and
|
|
|
|
self.obj == other.obj and
|
|
|
|
getattr(self, '_genid', None) ==
|
|
|
|
getattr(other, '_genid', None)
|
|
|
|
)
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
return False
|
|
|
|
|
|
|
|
def __ne__(self, other):
|
|
|
|
return not self == other
|
|
|
|
|
|
|
|
def __hash__(self):
|
|
|
|
return hash((self.parent, self.name))
|
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
scope2props = dict(session=())
|
|
|
|
scope2props["module"] = ("fspath", "module")
|
|
|
|
scope2props["class"] = scope2props["module"] + ("cls",)
|
|
|
|
scope2props["instance"] = scope2props["class"] + ("instance", )
|
|
|
|
scope2props["function"] = scope2props["instance"] + ("function", "keywords")
|
|
|
|
|
|
|
|
def scopeproperty(name=None, doc=None):
|
|
|
|
def decoratescope(func):
|
|
|
|
scopename = name or func.__name__
|
|
|
|
def provide(self):
|
|
|
|
if func.__name__ in scope2props[self.scope]:
|
|
|
|
return func(self)
|
|
|
|
raise AttributeError("%s not available in %s-scoped context" % (
|
|
|
|
scopename, self.scope))
|
|
|
|
return property(provide, None, None, func.__doc__)
|
|
|
|
return decoratescope
|
|
|
|
|
2012-06-25 23:35:33 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class FixtureRequest(FuncargnamesCompatAttr):
|
2012-10-09 20:35:17 +08:00
|
|
|
""" A request for a fixture from a test or fixture function.
|
2012-09-17 22:36:10 +08:00
|
|
|
|
2012-10-09 20:35:17 +08:00
|
|
|
A request object gives access to the requesting test context
|
|
|
|
and has an optional ``param`` attribute in case
|
|
|
|
the fixture is parametrized indirectly.
|
2012-07-19 01:49:14 +08:00
|
|
|
"""
|
2012-06-25 23:35:33 +08:00
|
|
|
|
|
|
|
def __init__(self, pyfuncitem):
|
|
|
|
self._pyfuncitem = pyfuncitem
|
2012-07-19 01:49:14 +08:00
|
|
|
if hasattr(pyfuncitem, '_requestparam'):
|
|
|
|
self.param = pyfuncitem._requestparam
|
2012-10-09 20:35:17 +08:00
|
|
|
#: fixture for which this request is being performed
|
|
|
|
self.fixturename = None
|
2012-09-17 22:36:10 +08:00
|
|
|
#: Scope string, one of "function", "cls", "module", "session"
|
|
|
|
self.scope = "function"
|
2012-07-19 01:49:14 +08:00
|
|
|
self._funcargs = self._pyfuncitem.funcargs.copy()
|
2013-11-21 19:21:52 +08:00
|
|
|
fixtureinfo = pyfuncitem._fixtureinfo
|
|
|
|
self._arg2fixturedefs = fixtureinfo.name2fixturedefs
|
2012-10-16 19:59:12 +08:00
|
|
|
self._arg2index = {}
|
2013-11-21 19:21:52 +08:00
|
|
|
self.fixturenames = fixtureinfo.names_closure
|
2012-10-05 20:24:44 +08:00
|
|
|
self._fixturemanager = pyfuncitem.session._fixturemanager
|
2012-07-19 15:20:14 +08:00
|
|
|
|
2012-09-17 23:32:23 +08:00
|
|
|
@property
|
2012-09-18 02:43:37 +08:00
|
|
|
def node(self):
|
2012-10-07 19:06:17 +08:00
|
|
|
""" underlying collection node (depends on current request scope)"""
|
2012-09-18 02:43:37 +08:00
|
|
|
return self._getscopeitem(self.scope)
|
2012-09-17 23:32:23 +08:00
|
|
|
|
2012-10-16 19:59:12 +08:00
|
|
|
def _getnextfixturedef(self, argname):
|
|
|
|
fixturedefs = self._arg2fixturedefs.get(argname, None)
|
|
|
|
if fixturedefs is None:
|
|
|
|
# we arrive here because of a getfuncargvalue(argname) usage which
|
|
|
|
# was naturally not knowable at parsing/collection time
|
|
|
|
fixturedefs = self._fixturemanager.getfixturedefs(
|
2013-11-21 19:21:52 +08:00
|
|
|
argname, self._pyfuncitem.parent.nodeid)
|
2012-10-16 19:59:12 +08:00
|
|
|
self._arg2fixturedefs[argname] = fixturedefs
|
|
|
|
# fixturedefs is immutable so we maintain a decreasing index
|
|
|
|
index = self._arg2index.get(argname, 0) - 1
|
|
|
|
if fixturedefs is None or (-index > len(fixturedefs)):
|
2012-10-05 20:24:45 +08:00
|
|
|
raise FixtureLookupError(argname, self)
|
2012-10-16 19:59:12 +08:00
|
|
|
self._arg2index[argname] = index
|
|
|
|
return fixturedefs[index]
|
2012-06-25 23:35:33 +08:00
|
|
|
|
2012-07-19 01:49:14 +08:00
|
|
|
@property
|
2012-09-17 22:36:10 +08:00
|
|
|
def config(self):
|
|
|
|
""" the pytest config object associated with this request. """
|
|
|
|
return self._pyfuncitem.config
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2012-08-01 19:57:09 +08:00
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
@scopeproperty()
|
|
|
|
def function(self):
|
|
|
|
""" test function object if the request has a per-function scope. """
|
|
|
|
return self._pyfuncitem.obj
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
@scopeproperty("class")
|
2012-07-19 01:49:14 +08:00
|
|
|
def cls(self):
|
|
|
|
""" class (can be None) where the test function was collected. """
|
|
|
|
clscol = self._pyfuncitem.getparent(pytest.Class)
|
|
|
|
if clscol:
|
|
|
|
return clscol.obj
|
2012-09-17 22:36:10 +08:00
|
|
|
|
2012-09-18 16:54:12 +08:00
|
|
|
@property
|
2012-07-19 01:49:14 +08:00
|
|
|
def instance(self):
|
|
|
|
""" instance (can be None) on which test function was collected. """
|
2012-09-18 16:54:12 +08:00
|
|
|
# unittest support hack, see _pytest.unittest.TestCaseFunction
|
|
|
|
try:
|
|
|
|
return self._pyfuncitem._testcase
|
|
|
|
except AttributeError:
|
|
|
|
return py.builtin._getimself(self.function)
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
@scopeproperty()
|
|
|
|
def module(self):
|
|
|
|
""" python module object where the test function was collected. """
|
|
|
|
return self._pyfuncitem.getparent(pytest.Module).obj
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
@scopeproperty()
|
2012-07-19 01:49:14 +08:00
|
|
|
def fspath(self):
|
|
|
|
""" the file system path of the test module which collected this test. """
|
|
|
|
return self._pyfuncitem.fspath
|
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
@property
|
|
|
|
def keywords(self):
|
2012-10-18 19:52:32 +08:00
|
|
|
""" keywords/markers dictionary for the underlying node. """
|
2012-10-18 21:06:55 +08:00
|
|
|
return self.node.keywords
|
2012-09-17 22:36:10 +08:00
|
|
|
|
|
|
|
@property
|
|
|
|
def session(self):
|
|
|
|
""" pytest session object. """
|
|
|
|
return self._pyfuncitem.session
|
|
|
|
|
|
|
|
def addfinalizer(self, finalizer):
|
|
|
|
"""add finalizer/teardown function to be called after the
|
|
|
|
last test within the requesting test context finished
|
|
|
|
execution. """
|
2012-10-05 20:24:44 +08:00
|
|
|
# XXX usually this method is shadowed by fixturedef specific ones
|
2012-09-17 22:36:10 +08:00
|
|
|
self._addfinalizer(finalizer, scope=self.scope)
|
|
|
|
|
|
|
|
def _addfinalizer(self, finalizer, scope):
|
|
|
|
if scope != "function" and hasattr(self, "param"):
|
|
|
|
# parametrized resources are sorted by param
|
|
|
|
# so we rather store finalizers per (argname, param)
|
2012-10-09 20:35:17 +08:00
|
|
|
colitem = (self.fixturename, self.param)
|
2012-09-17 22:36:10 +08:00
|
|
|
else:
|
|
|
|
colitem = self._getscopeitem(scope)
|
|
|
|
self._pyfuncitem.session._setupstate.addfinalizer(
|
|
|
|
finalizer=finalizer, colitem=colitem)
|
2012-07-19 01:49:14 +08:00
|
|
|
|
|
|
|
def applymarker(self, marker):
|
|
|
|
""" Apply a marker to a single test function invocation.
|
|
|
|
This method is useful if you don't want to have a keyword/marker
|
|
|
|
on all function invocations.
|
|
|
|
|
|
|
|
:arg marker: a :py:class:`_pytest.mark.MarkDecorator` object
|
|
|
|
created by a call to ``py.test.mark.NAME(...)``.
|
|
|
|
"""
|
2012-10-18 19:52:32 +08:00
|
|
|
try:
|
|
|
|
self.node.keywords[marker.markname] = marker
|
|
|
|
except AttributeError:
|
|
|
|
raise ValueError(marker)
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2012-09-17 22:36:10 +08:00
|
|
|
def raiseerror(self, msg):
|
2012-10-05 20:24:44 +08:00
|
|
|
""" raise a FixtureLookupError with the given message. """
|
2012-10-05 20:24:45 +08:00
|
|
|
raise self._fixturemanager.FixtureLookupError(None, self, msg)
|
2012-09-17 22:36:10 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
def _fillfixtures(self):
|
2012-09-24 23:04:34 +08:00
|
|
|
item = self._pyfuncitem
|
2012-10-05 20:24:44 +08:00
|
|
|
fixturenames = getattr(item, "fixturenames", self.fixturenames)
|
|
|
|
for argname in fixturenames:
|
2012-09-24 23:04:34 +08:00
|
|
|
if argname not in item.funcargs:
|
|
|
|
item.funcargs[argname] = self.getfuncargvalue(argname)
|
2012-09-17 22:36:10 +08:00
|
|
|
|
2012-07-19 01:49:14 +08:00
|
|
|
def cached_setup(self, setup, teardown=None, scope="module", extrakey=None):
|
2012-09-17 22:36:10 +08:00
|
|
|
""" (deprecated) Return a testing resource managed by ``setup`` &
|
2012-07-19 01:49:14 +08:00
|
|
|
``teardown`` calls. ``scope`` and ``extrakey`` determine when the
|
|
|
|
``teardown`` function will be called so that subsequent calls to
|
2012-10-05 20:24:44 +08:00
|
|
|
``setup`` would recreate the resource. With pytest-2.3 you often
|
2012-09-17 22:36:10 +08:00
|
|
|
do not need ``cached_setup()`` as you can directly declare a scope
|
2012-10-05 20:24:44 +08:00
|
|
|
on a fixture function and register a finalizer through
|
2012-09-17 22:36:10 +08:00
|
|
|
``request.addfinalizer()``.
|
2012-07-19 01:49:14 +08:00
|
|
|
|
|
|
|
:arg teardown: function receiving a previously setup resource.
|
|
|
|
:arg setup: a no-argument function creating a resource.
|
|
|
|
:arg scope: a string value out of ``function``, ``class``, ``module``
|
|
|
|
or ``session`` indicating the caching lifecycle of the resource.
|
|
|
|
:arg extrakey: added to internal caching key of (funcargname, scope).
|
|
|
|
"""
|
|
|
|
if not hasattr(self.config, '_setupcache'):
|
|
|
|
self.config._setupcache = {} # XXX weakref?
|
2012-10-09 20:35:17 +08:00
|
|
|
cachekey = (self.fixturename, self._getscopeitem(scope), extrakey)
|
2012-07-19 01:49:14 +08:00
|
|
|
cache = self.config._setupcache
|
|
|
|
try:
|
|
|
|
val = cache[cachekey]
|
|
|
|
except KeyError:
|
2012-07-20 20:16:46 +08:00
|
|
|
__tracebackhide__ = True
|
2012-07-23 16:55:09 +08:00
|
|
|
if scopemismatch(self.scope, scope):
|
|
|
|
raise ScopeMismatchError("You tried to access a %r scoped "
|
|
|
|
"resource with a %r scoped request object" %(
|
|
|
|
(scope, self.scope)))
|
2012-07-20 20:16:46 +08:00
|
|
|
__tracebackhide__ = False
|
2012-07-19 01:49:14 +08:00
|
|
|
val = setup()
|
|
|
|
cache[cachekey] = val
|
|
|
|
if teardown is not None:
|
|
|
|
def finalizer():
|
|
|
|
del cache[cachekey]
|
|
|
|
teardown(val)
|
|
|
|
self._addfinalizer(finalizer, scope=scope)
|
|
|
|
return val
|
|
|
|
|
|
|
|
def getfuncargvalue(self, argname):
|
2012-10-09 20:35:17 +08:00
|
|
|
""" Dynamically retrieve a named fixture function argument.
|
|
|
|
|
|
|
|
As of pytest-2.3, it is easier and usually better to access other
|
|
|
|
fixture values by stating it as an input argument in the fixture
|
|
|
|
function. If you only can decide about using another fixture at test
|
|
|
|
setup time, you may use this function to retrieve it inside a fixture
|
|
|
|
function body.
|
2012-07-19 01:49:14 +08:00
|
|
|
"""
|
|
|
|
try:
|
|
|
|
return self._funcargs[argname]
|
|
|
|
except KeyError:
|
|
|
|
pass
|
2012-10-01 15:23:39 +08:00
|
|
|
try:
|
2012-10-16 19:59:12 +08:00
|
|
|
fixturedef = self._getnextfixturedef(argname)
|
2012-10-05 20:24:44 +08:00
|
|
|
except FixtureLookupError:
|
2012-10-01 15:23:39 +08:00
|
|
|
if argname == "request":
|
|
|
|
return self
|
|
|
|
raise
|
2013-11-21 20:15:32 +08:00
|
|
|
result = self._getfuncargvalue(fixturedef)
|
|
|
|
self._funcargs[argname] = result
|
|
|
|
return result
|
|
|
|
|
|
|
|
def _get_fixturestack(self):
|
|
|
|
current = self
|
|
|
|
l = []
|
|
|
|
while 1:
|
|
|
|
fixturedef = getattr(current, "_fixturedef", None)
|
|
|
|
if fixturedef is None:
|
|
|
|
l.reverse()
|
|
|
|
return l
|
|
|
|
l.append(fixturedef)
|
|
|
|
current = current._parent_request
|
2012-07-23 16:55:09 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
def _getfuncargvalue(self, fixturedef):
|
2012-11-19 19:42:10 +08:00
|
|
|
try:
|
|
|
|
return fixturedef.cached_result # set by fixturedef.execute()
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
2013-11-21 19:21:52 +08:00
|
|
|
# prepare a subrequest object before calling fixture function
|
|
|
|
# (latter managed by fixturedef)
|
2012-10-05 20:24:44 +08:00
|
|
|
argname = fixturedef.argname
|
2012-07-20 20:16:28 +08:00
|
|
|
node = self._pyfuncitem
|
2013-11-21 19:21:52 +08:00
|
|
|
scope = fixturedef.scope
|
2012-07-20 20:16:28 +08:00
|
|
|
try:
|
|
|
|
param = node.callspec.getparam(argname)
|
|
|
|
except (AttributeError, ValueError):
|
2013-11-21 20:15:32 +08:00
|
|
|
param = NOTSET
|
2012-07-20 20:16:28 +08:00
|
|
|
else:
|
2013-11-21 19:21:52 +08:00
|
|
|
# if a parametrize invocation set a scope it will override
|
|
|
|
# the static scope defined with the fixture function
|
|
|
|
paramscopenum = node.callspec._arg2scopenum.get(argname)
|
|
|
|
if paramscopenum is not None and \
|
|
|
|
paramscopenum != scopenum_subfunction:
|
2012-10-07 03:01:13 +08:00
|
|
|
scope = scopes[paramscopenum]
|
|
|
|
|
2013-11-21 20:15:32 +08:00
|
|
|
subrequest = SubRequest(self, scope, param, fixturedef)
|
|
|
|
|
2012-11-16 17:03:51 +08:00
|
|
|
# check if a higher-level scoped fixture accesses a lower level one
|
2012-07-20 20:16:46 +08:00
|
|
|
if scope is not None:
|
|
|
|
__tracebackhide__ = True
|
2012-07-23 16:55:09 +08:00
|
|
|
if scopemismatch(self.scope, scope):
|
|
|
|
# try to report something helpful
|
2013-11-21 20:15:32 +08:00
|
|
|
lines = subrequest._factorytraceback()
|
2012-07-23 16:55:09 +08:00
|
|
|
raise ScopeMismatchError("You tried to access the %r scoped "
|
2013-11-21 19:21:52 +08:00
|
|
|
"fixture %r with a %r scoped request object, "
|
2012-07-23 16:55:09 +08:00
|
|
|
"involved factories\n%s" %(
|
|
|
|
(scope, argname, self.scope, "\n".join(lines))))
|
2012-07-20 20:16:46 +08:00
|
|
|
__tracebackhide__ = False
|
2012-11-16 17:03:51 +08:00
|
|
|
|
2013-07-17 16:29:11 +08:00
|
|
|
try:
|
|
|
|
# perform the fixture call
|
2013-11-21 19:21:52 +08:00
|
|
|
val = fixturedef.execute(request=subrequest)
|
2013-07-17 16:29:11 +08:00
|
|
|
finally:
|
|
|
|
# if the fixture function failed it might still have
|
|
|
|
# registered finalizers so we can register
|
|
|
|
# prepare finalization according to scope
|
|
|
|
# (XXX analyse exact finalizing mechanics / cleanup)
|
|
|
|
self.session._setupstate.addfinalizer(fixturedef.finish,
|
2013-11-21 19:21:52 +08:00
|
|
|
subrequest.node)
|
2013-07-17 16:29:11 +08:00
|
|
|
self._fixturemanager.addargfinalizer(fixturedef.finish, argname)
|
|
|
|
for subargname in fixturedef.argnames: # XXX all deps?
|
|
|
|
self._fixturemanager.addargfinalizer(fixturedef.finish,
|
|
|
|
subargname)
|
2012-07-20 20:16:46 +08:00
|
|
|
return val
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2012-08-08 17:48:53 +08:00
|
|
|
def _factorytraceback(self):
|
|
|
|
lines = []
|
2013-11-21 20:15:32 +08:00
|
|
|
for fixturedef in self._get_fixturestack():
|
2012-10-05 20:24:44 +08:00
|
|
|
factory = fixturedef.func
|
2012-08-08 17:48:53 +08:00
|
|
|
fs, lineno = getfslineno(factory)
|
|
|
|
p = self._pyfuncitem.session.fspath.bestrelpath(fs)
|
|
|
|
args = inspect.formatargspec(*inspect.getargspec(factory))
|
|
|
|
lines.append("%s:%d: def %s%s" %(
|
|
|
|
p, lineno, factory.__name__, args))
|
|
|
|
return lines
|
|
|
|
|
2012-07-19 01:49:14 +08:00
|
|
|
def _getscopeitem(self, scope):
|
|
|
|
if scope == "function":
|
|
|
|
return self._pyfuncitem
|
|
|
|
elif scope == "session":
|
2012-10-18 21:06:55 +08:00
|
|
|
return self.session
|
2012-07-19 01:49:14 +08:00
|
|
|
elif scope == "class":
|
|
|
|
x = self._pyfuncitem.getparent(pytest.Class)
|
|
|
|
if x is not None:
|
|
|
|
return x
|
2013-07-07 03:30:24 +08:00
|
|
|
# fallback to function
|
|
|
|
return self._pyfuncitem
|
2012-07-19 01:49:14 +08:00
|
|
|
if scope == "module":
|
|
|
|
return self._pyfuncitem.getparent(pytest.Module)
|
|
|
|
raise ValueError("unknown finalization scope %r" %(scope,))
|
|
|
|
|
|
|
|
def __repr__(self):
|
2012-10-09 20:35:17 +08:00
|
|
|
return "<FixtureRequest for %r>" %(self.node)
|
2012-07-19 01:49:14 +08:00
|
|
|
|
2013-11-21 20:15:32 +08:00
|
|
|
|
2013-11-21 19:21:52 +08:00
|
|
|
class SubRequest(FixtureRequest):
|
|
|
|
""" a sub request for handling getting a fixture from a
|
|
|
|
test function/fixture. """
|
2013-11-21 19:42:22 +08:00
|
|
|
def __init__(self, request, scope, param, fixturedef):
|
2013-11-21 19:21:52 +08:00
|
|
|
self._parent_request = request
|
2013-11-21 19:42:22 +08:00
|
|
|
self.fixturename = fixturedef.argname
|
2013-11-21 20:15:32 +08:00
|
|
|
if param is not NOTSET:
|
2013-11-21 19:21:52 +08:00
|
|
|
self.param = param
|
|
|
|
self.scope = scope
|
2013-11-21 20:15:32 +08:00
|
|
|
self._fixturedef = fixturedef
|
2013-11-21 19:42:22 +08:00
|
|
|
self.addfinalizer = fixturedef.addfinalizer
|
2013-11-21 19:21:52 +08:00
|
|
|
self._pyfuncitem = request._pyfuncitem
|
|
|
|
self._funcargs = request._funcargs
|
|
|
|
self._arg2fixturedefs = request._arg2fixturedefs
|
|
|
|
self._arg2index = request._arg2index
|
|
|
|
self.fixturenames = request.fixturenames
|
|
|
|
self._fixturemanager = request._fixturemanager
|
|
|
|
|
|
|
|
def __repr__(self):
|
2013-11-21 20:15:32 +08:00
|
|
|
return "<SubRequest %r for %r>" % (self.fixturename, self._pyfuncitem)
|
|
|
|
|
2013-11-21 19:21:52 +08:00
|
|
|
|
2012-07-20 20:16:46 +08:00
|
|
|
class ScopeMismatchError(Exception):
|
2012-10-05 20:24:44 +08:00
|
|
|
""" A fixture function tries to use a different fixture function which
|
2012-07-20 20:16:46 +08:00
|
|
|
which has a lower scope (e.g. a Session one calls a function one)
|
|
|
|
"""
|
2012-07-23 16:55:09 +08:00
|
|
|
|
2012-10-07 03:01:13 +08:00
|
|
|
scopes = "session module class function subfunction".split()
|
|
|
|
scopenum_subfunction = scopes.index("subfunction")
|
2013-11-21 21:16:44 +08:00
|
|
|
scopenum_function = scopes.index("function")
|
2012-07-23 16:55:09 +08:00
|
|
|
def scopemismatch(currentscope, newscope):
|
|
|
|
return scopes.index(newscope) > scopes.index(currentscope)
|
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class FixtureLookupError(LookupError):
|
2012-10-05 20:24:45 +08:00
|
|
|
""" could not return a requested Fixture (missing or invalid). """
|
|
|
|
def __init__(self, argname, request, msg=None):
|
|
|
|
self.argname = argname
|
|
|
|
self.request = request
|
2013-11-21 20:15:32 +08:00
|
|
|
self.fixturestack = request._get_fixturestack()
|
2012-08-01 15:23:39 +08:00
|
|
|
self.msg = msg
|
2012-10-05 20:24:45 +08:00
|
|
|
|
|
|
|
def formatrepr(self):
|
|
|
|
tblines = []
|
|
|
|
addline = tblines.append
|
|
|
|
stack = [self.request._pyfuncitem.obj]
|
|
|
|
stack.extend(map(lambda x: x.func, self.fixturestack))
|
|
|
|
msg = self.msg
|
|
|
|
if msg is not None:
|
|
|
|
stack = stack[:-1] # the last fixture raise an error, let's present
|
|
|
|
# it at the requesting side
|
|
|
|
for function in stack:
|
|
|
|
fspath, lineno = getfslineno(function)
|
|
|
|
lines, _ = inspect.getsourcelines(function)
|
|
|
|
addline("file %s, line %s" % (fspath, lineno+1))
|
|
|
|
for i, line in enumerate(lines):
|
|
|
|
line = line.rstrip()
|
|
|
|
addline(" " + line)
|
|
|
|
if line.lstrip().startswith('def'):
|
|
|
|
break
|
|
|
|
|
|
|
|
if msg is None:
|
|
|
|
fm = self.request._fixturemanager
|
|
|
|
available = []
|
2012-10-16 22:13:12 +08:00
|
|
|
for name, fixturedef in fm._arg2fixturedefs.items():
|
2013-11-21 19:21:52 +08:00
|
|
|
parentid = self.request._pyfuncitem.parent.nodeid
|
|
|
|
faclist = list(fm._matchfactories(fixturedef, parentid))
|
2012-10-05 20:24:45 +08:00
|
|
|
if faclist:
|
|
|
|
available.append(name)
|
|
|
|
msg = "fixture %r not found" % (self.argname,)
|
|
|
|
msg += "\n available fixtures: %s" %(", ".join(available),)
|
|
|
|
msg += "\n use 'py.test --fixtures [testpath]' for help on them."
|
|
|
|
|
|
|
|
return FixtureLookupErrorRepr(fspath, lineno, tblines, msg, self.argname)
|
2012-08-01 15:23:39 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class FixtureLookupErrorRepr(TerminalRepr):
|
2012-10-05 20:24:45 +08:00
|
|
|
def __init__(self, filename, firstlineno, tblines, errorstring, argname):
|
|
|
|
self.tblines = tblines
|
2012-08-01 15:23:39 +08:00
|
|
|
self.errorstring = errorstring
|
|
|
|
self.filename = filename
|
|
|
|
self.firstlineno = firstlineno
|
2012-10-05 20:24:45 +08:00
|
|
|
self.argname = argname
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
def toterminal(self, tw):
|
2012-10-05 20:24:45 +08:00
|
|
|
#tw.line("FixtureLookupError: %s" %(self.argname), red=True)
|
|
|
|
for tbline in self.tblines:
|
|
|
|
tw.line(tbline.rstrip())
|
2012-08-01 15:23:39 +08:00
|
|
|
for line in self.errorstring.split("\n"):
|
|
|
|
tw.line(" " + line.strip(), red=True)
|
|
|
|
tw.line()
|
|
|
|
tw.line("%s:%d" % (self.filename, self.firstlineno+1))
|
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class FixtureManager:
|
2012-10-17 17:20:45 +08:00
|
|
|
"""
|
|
|
|
pytest fixtures definitions and information is stored and managed
|
|
|
|
from this class.
|
|
|
|
|
|
|
|
During collection fm.parsefactories() is called multiple times to parse
|
|
|
|
fixture function definitions into FixtureDef objects and internal
|
|
|
|
data structures.
|
|
|
|
|
|
|
|
During collection of test functions, metafunc-mechanics instantiate
|
|
|
|
a FuncFixtureInfo object which is cached per node/func-name.
|
|
|
|
This FuncFixtureInfo object is later retrieved by Function nodes
|
|
|
|
which themselves offer a fixturenames attribute.
|
|
|
|
|
|
|
|
The FuncFixtureInfo object holds information about fixtures and FixtureDefs
|
|
|
|
relevant for a particular function. An initial list of fixtures is
|
|
|
|
assembled like this:
|
|
|
|
|
|
|
|
- ini-defined usefixtures
|
|
|
|
- autouse-marked fixtures along the collection chain up from the function
|
|
|
|
- usefixtures markers at module/class/function level
|
|
|
|
- test function funcargs
|
|
|
|
|
|
|
|
Subsequently the funcfixtureinfo.fixturenames attribute is computed
|
|
|
|
as the closure of the fixtures needed to setup the initial fixtures,
|
|
|
|
i. e. fixtures needed by fixture functions themselves are appended
|
|
|
|
to the fixturenames list.
|
|
|
|
|
|
|
|
Upon the test-setup phases all fixturenames are instantiated, retrieved
|
|
|
|
by a lookup of their FuncFixtureInfo.
|
|
|
|
"""
|
|
|
|
|
2012-08-01 15:23:39 +08:00
|
|
|
_argprefix = "pytest_funcarg__"
|
2012-10-05 20:24:44 +08:00
|
|
|
FixtureLookupError = FixtureLookupError
|
|
|
|
FixtureLookupErrorRepr = FixtureLookupErrorRepr
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
def __init__(self, session):
|
|
|
|
self.session = session
|
|
|
|
self.config = session.config
|
2012-10-16 22:13:12 +08:00
|
|
|
self._arg2fixturedefs = {}
|
2012-09-12 18:51:45 +08:00
|
|
|
self._seenplugins = set()
|
2012-08-01 15:23:39 +08:00
|
|
|
self._holderobjseen = set()
|
|
|
|
self._arg2finish = {}
|
2012-10-16 22:13:12 +08:00
|
|
|
self._nodeid_and_autousenames = [("", self.config.getini("usefixtures"))]
|
2012-09-12 18:51:45 +08:00
|
|
|
session.config.pluginmanager.register(self, "funcmanage")
|
2012-08-01 15:23:39 +08:00
|
|
|
|
2012-10-17 17:20:45 +08:00
|
|
|
self._nodename2fixtureinfo = {}
|
|
|
|
|
2012-11-20 05:17:59 +08:00
|
|
|
def getfixtureinfo(self, node, func, cls, funcargs=True):
|
2012-10-17 17:20:45 +08:00
|
|
|
key = (node, func.__name__)
|
|
|
|
try:
|
|
|
|
return self._nodename2fixtureinfo[key]
|
|
|
|
except KeyError:
|
|
|
|
pass
|
2012-11-20 05:17:59 +08:00
|
|
|
if funcargs and not hasattr(node, "nofuncargs"):
|
2012-10-17 17:20:45 +08:00
|
|
|
if cls is not None:
|
|
|
|
startindex = 1
|
|
|
|
else:
|
|
|
|
startindex = None
|
|
|
|
argnames = getfuncargnames(func, startindex)
|
|
|
|
else:
|
|
|
|
argnames = ()
|
|
|
|
usefixtures = getattr(func, "usefixtures", None)
|
|
|
|
initialnames = argnames
|
|
|
|
if usefixtures is not None:
|
|
|
|
initialnames = usefixtures.args + initialnames
|
|
|
|
fm = node.session._fixturemanager
|
|
|
|
names_closure, arg2fixturedefs = fm.getfixtureclosure(initialnames,
|
|
|
|
node)
|
|
|
|
fixtureinfo = FuncFixtureInfo(argnames, names_closure,
|
|
|
|
arg2fixturedefs)
|
|
|
|
self._nodename2fixtureinfo[key] = fixtureinfo
|
|
|
|
return fixtureinfo
|
|
|
|
|
2012-08-01 15:23:39 +08:00
|
|
|
### XXX this hook should be called for historic events like pytest_configure
|
2012-09-12 18:51:45 +08:00
|
|
|
### so that we don't have to do the below pytest_configure hook
|
2012-08-01 15:23:39 +08:00
|
|
|
def pytest_plugin_registered(self, plugin):
|
2012-09-12 18:51:45 +08:00
|
|
|
if plugin in self._seenplugins:
|
|
|
|
return
|
2013-07-06 23:56:54 +08:00
|
|
|
nodeid = None
|
2012-08-01 15:23:39 +08:00
|
|
|
try:
|
|
|
|
p = py.path.local(plugin.__file__)
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
if p.basename.startswith("conftest.py"):
|
|
|
|
nodeid = p.dirpath().relto(self.session.fspath)
|
2012-10-17 19:42:40 +08:00
|
|
|
if p.sep != "/":
|
2012-10-17 19:45:03 +08:00
|
|
|
nodeid = nodeid.replace(p.sep, "/")
|
2012-10-16 22:13:12 +08:00
|
|
|
self.parsefactories(plugin, nodeid)
|
2012-09-12 18:51:45 +08:00
|
|
|
self._seenplugins.add(plugin)
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
@pytest.mark.tryfirst
|
2012-09-12 18:51:45 +08:00
|
|
|
def pytest_configure(self, config):
|
|
|
|
plugins = config.pluginmanager.getplugins()
|
2012-08-01 15:23:39 +08:00
|
|
|
for plugin in plugins:
|
|
|
|
self.pytest_plugin_registered(plugin)
|
|
|
|
|
2012-10-16 22:13:12 +08:00
|
|
|
def _getautousenames(self, nodeid):
|
|
|
|
""" return a tuple of fixture names to be used. """
|
|
|
|
autousenames = []
|
|
|
|
for baseid, basenames in self._nodeid_and_autousenames:
|
|
|
|
if nodeid.startswith(baseid):
|
|
|
|
if baseid:
|
2012-11-20 05:17:55 +08:00
|
|
|
i = len(baseid)
|
2012-10-16 22:13:12 +08:00
|
|
|
nextchar = nodeid[i:i+1]
|
|
|
|
if nextchar and nextchar not in ":/":
|
|
|
|
continue
|
|
|
|
autousenames.extend(basenames)
|
|
|
|
# make sure autousenames are sorted by scope, scopenum 0 is session
|
|
|
|
autousenames.sort(
|
|
|
|
key=lambda x: self._arg2fixturedefs[x][-1].scopenum)
|
|
|
|
return autousenames
|
2012-10-06 01:20:40 +08:00
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
def getfixtureclosure(self, fixturenames, parentnode):
|
2012-10-06 01:20:40 +08:00
|
|
|
# collect the closure of all fixtures , starting with the given
|
2012-10-05 20:24:44 +08:00
|
|
|
# fixturenames as the initial set. As we have to visit all
|
2012-10-16 19:59:12 +08:00
|
|
|
# factory definitions anyway, we also return a arg2fixturedefs
|
2012-10-05 20:24:44 +08:00
|
|
|
# mapping so that the caller can reuse it and does not have
|
|
|
|
# to re-discover fixturedefs again for each fixturename
|
|
|
|
# (discovering matching fixtures for a given name/node is expensive)
|
|
|
|
|
2012-09-24 23:04:34 +08:00
|
|
|
parentid = parentnode.nodeid
|
2012-10-16 22:13:12 +08:00
|
|
|
fixturenames_closure = self._getautousenames(parentid)
|
2012-09-24 23:04:34 +08:00
|
|
|
def merge(otherlist):
|
|
|
|
for arg in otherlist:
|
2012-10-05 20:24:44 +08:00
|
|
|
if arg not in fixturenames_closure:
|
|
|
|
fixturenames_closure.append(arg)
|
|
|
|
merge(fixturenames)
|
2012-10-16 19:59:12 +08:00
|
|
|
arg2fixturedefs = {}
|
2012-09-24 23:04:34 +08:00
|
|
|
lastlen = -1
|
2012-10-05 20:24:44 +08:00
|
|
|
while lastlen != len(fixturenames_closure):
|
|
|
|
lastlen = len(fixturenames_closure)
|
|
|
|
for argname in fixturenames_closure:
|
2012-10-16 19:59:12 +08:00
|
|
|
if argname in arg2fixturedefs:
|
2012-09-24 23:04:34 +08:00
|
|
|
continue
|
2012-10-16 19:59:12 +08:00
|
|
|
fixturedefs = self.getfixturedefs(argname, parentid)
|
|
|
|
arg2fixturedefs[argname] = fixturedefs
|
2013-11-14 01:25:55 +08:00
|
|
|
if fixturedefs:
|
2013-11-12 20:45:36 +08:00
|
|
|
merge(fixturedefs[-1].argnames)
|
2012-10-16 19:59:12 +08:00
|
|
|
return fixturenames_closure, arg2fixturedefs
|
2012-09-24 23:04:34 +08:00
|
|
|
|
2012-08-01 15:23:39 +08:00
|
|
|
def pytest_generate_tests(self, metafunc):
|
2012-10-05 20:24:44 +08:00
|
|
|
for argname in metafunc.fixturenames:
|
2012-10-16 19:59:12 +08:00
|
|
|
faclist = metafunc._arg2fixturedefs[argname]
|
2012-08-01 15:23:39 +08:00
|
|
|
if faclist is None:
|
2012-10-05 20:24:44 +08:00
|
|
|
continue # will raise FixtureLookupError at setup time
|
|
|
|
for fixturedef in faclist:
|
|
|
|
if fixturedef.params is not None:
|
2013-10-21 19:33:36 +08:00
|
|
|
metafunc.parametrize(argname, fixturedef.params,
|
|
|
|
indirect=True, scope=fixturedef.scope)
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
def pytest_collection_modifyitems(self, items):
|
|
|
|
# separate parametrized setups
|
|
|
|
items[:] = parametrize_sorted(items, set(), {}, 0)
|
|
|
|
|
2013-11-21 21:16:44 +08:00
|
|
|
@pytest.mark.trylast
|
2012-08-01 15:23:39 +08:00
|
|
|
def pytest_runtest_teardown(self, item, nextitem):
|
2012-11-16 17:03:51 +08:00
|
|
|
# XXX teardown needs to be normalized for parametrized and
|
|
|
|
# no-parametrized functions
|
2012-08-01 15:23:39 +08:00
|
|
|
try:
|
|
|
|
cs1 = item.callspec
|
|
|
|
except AttributeError:
|
|
|
|
return
|
2012-10-05 16:21:35 +08:00
|
|
|
|
|
|
|
# determine which fixtures are not needed anymore for the next test
|
|
|
|
keylist = []
|
2012-08-01 15:23:39 +08:00
|
|
|
for name in cs1.params:
|
|
|
|
try:
|
|
|
|
if name in nextitem.callspec.params and \
|
|
|
|
cs1.params[name] == nextitem.callspec.params[name]:
|
|
|
|
continue
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
2012-10-05 16:21:35 +08:00
|
|
|
key = (-cs1._arg2scopenum[name], name, cs1.params[name])
|
|
|
|
keylist.append(key)
|
|
|
|
|
|
|
|
# sort by scope (function scope first, then higher ones)
|
|
|
|
keylist.sort()
|
|
|
|
for (scopenum, name, param) in keylist:
|
2013-11-21 21:16:44 +08:00
|
|
|
#if -scopenum >= scopenum_function:
|
|
|
|
# continue # handled by runner.pytest_runtest_teardown
|
2012-10-05 16:21:35 +08:00
|
|
|
item.session._setupstate._callfinalizers((name, param))
|
2012-11-16 17:03:51 +08:00
|
|
|
l = self._arg2finish.pop(name, None)
|
2012-08-01 15:23:39 +08:00
|
|
|
if l is not None:
|
2012-10-22 18:16:54 +08:00
|
|
|
for fin in reversed(l):
|
2012-08-01 15:23:39 +08:00
|
|
|
fin()
|
|
|
|
|
2013-11-21 20:15:32 +08:00
|
|
|
def parsefactories(self, node_or_obj, nodeid=NOTSET, unittest=False):
|
|
|
|
if nodeid is not NOTSET:
|
2012-10-16 22:13:12 +08:00
|
|
|
holderobj = node_or_obj
|
|
|
|
else:
|
|
|
|
holderobj = node_or_obj.obj
|
|
|
|
nodeid = node_or_obj.nodeid
|
2012-08-01 15:23:39 +08:00
|
|
|
if holderobj in self._holderobjseen:
|
|
|
|
return
|
|
|
|
self._holderobjseen.add(holderobj)
|
2012-10-16 22:13:12 +08:00
|
|
|
autousenames = []
|
2012-08-01 15:23:39 +08:00
|
|
|
for name in dir(holderobj):
|
2012-10-18 21:09:20 +08:00
|
|
|
obj = getattr(holderobj, name, None)
|
2012-08-01 15:23:39 +08:00
|
|
|
if not callable(obj):
|
|
|
|
continue
|
2012-10-05 20:24:44 +08:00
|
|
|
# fixture functions have a pytest_funcarg__ prefix (pre-2.3 style)
|
2012-10-05 16:21:35 +08:00
|
|
|
# or are "@pytest.fixture" marked
|
2013-04-29 16:31:51 +08:00
|
|
|
marker = getfixturemarker(obj)
|
2012-10-05 16:21:35 +08:00
|
|
|
if marker is None:
|
|
|
|
if not name.startswith(self._argprefix):
|
2012-08-19 20:57:07 +08:00
|
|
|
continue
|
2012-10-05 16:21:35 +08:00
|
|
|
marker = defaultfuncargprefixmarker
|
|
|
|
name = name[len(self._argprefix):]
|
|
|
|
elif not isinstance(marker, FixtureFunctionMarker):
|
|
|
|
# magic globals with __getattr__ might have got us a wrong
|
|
|
|
# fixture attribute
|
2012-08-01 15:23:39 +08:00
|
|
|
continue
|
2012-10-05 16:21:35 +08:00
|
|
|
else:
|
|
|
|
assert not name.startswith(self._argprefix)
|
2012-10-05 20:24:44 +08:00
|
|
|
fixturedef = FixtureDef(self, nodeid, name, obj,
|
2012-10-05 16:21:35 +08:00
|
|
|
marker.scope, marker.params,
|
2013-09-27 18:28:34 +08:00
|
|
|
yieldctx=marker.yieldctx,
|
2012-10-05 16:21:35 +08:00
|
|
|
unittest=unittest)
|
2012-10-16 22:13:12 +08:00
|
|
|
faclist = self._arg2fixturedefs.setdefault(name, [])
|
2013-07-06 23:56:54 +08:00
|
|
|
if not fixturedef.has_location:
|
2013-08-02 01:58:28 +08:00
|
|
|
# All fixturedefs with no location are at the front
|
|
|
|
# so this inserts the current fixturedef after the
|
|
|
|
# existing fixturedefs from external plugins but
|
|
|
|
# before the fixturedefs provided in conftests.
|
|
|
|
i = len([f for f in faclist if not f.has_location])
|
2013-07-06 23:56:54 +08:00
|
|
|
else:
|
|
|
|
i = len(faclist) # append
|
|
|
|
faclist.insert(i, fixturedef)
|
2012-10-12 20:52:36 +08:00
|
|
|
if marker.autouse:
|
2012-10-16 22:13:12 +08:00
|
|
|
autousenames.append(name)
|
|
|
|
if autousenames:
|
2013-08-02 01:58:28 +08:00
|
|
|
self._nodeid_and_autousenames.append((nodeid or '', autousenames))
|
2012-08-01 15:23:39 +08:00
|
|
|
|
2012-10-16 19:59:12 +08:00
|
|
|
def getfixturedefs(self, argname, nodeid):
|
2012-08-01 15:23:39 +08:00
|
|
|
try:
|
2012-10-16 22:13:12 +08:00
|
|
|
fixturedefs = self._arg2fixturedefs[argname]
|
2012-08-01 15:23:39 +08:00
|
|
|
except KeyError:
|
2012-09-24 23:04:34 +08:00
|
|
|
return None
|
2012-08-01 15:23:39 +08:00
|
|
|
else:
|
2012-10-16 19:59:12 +08:00
|
|
|
return tuple(self._matchfactories(fixturedefs, nodeid))
|
2012-08-01 15:23:39 +08:00
|
|
|
|
2012-10-16 19:59:12 +08:00
|
|
|
def _matchfactories(self, fixturedefs, nodeid):
|
|
|
|
for fixturedef in fixturedefs:
|
2012-10-05 20:24:44 +08:00
|
|
|
if nodeid.startswith(fixturedef.baseid):
|
|
|
|
yield fixturedef
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
def addargfinalizer(self, finalizer, argname):
|
|
|
|
l = self._arg2finish.setdefault(argname, [])
|
|
|
|
l.append(finalizer)
|
|
|
|
|
|
|
|
def removefinalizer(self, finalizer):
|
|
|
|
for l in self._arg2finish.values():
|
|
|
|
try:
|
|
|
|
l.remove(finalizer)
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
|
2013-09-26 18:57:21 +08:00
|
|
|
def fail_fixturefunc(fixturefunc, msg):
|
|
|
|
fs, lineno = getfslineno(fixturefunc)
|
|
|
|
location = "%s:%s" % (fs, lineno+1)
|
|
|
|
source = py.code.Source(fixturefunc)
|
|
|
|
pytest.fail(msg + ":\n\n" + str(source.indent()) + "\n" + location,
|
|
|
|
pytrace=False)
|
|
|
|
|
|
|
|
def call_fixture_func(fixturefunc, request, kwargs, yieldctx):
|
|
|
|
if yieldctx:
|
|
|
|
if not is_generator(fixturefunc):
|
|
|
|
fail_fixturefunc(fixturefunc,
|
2013-09-30 19:42:39 +08:00
|
|
|
msg="yield_fixture requires yield statement in function")
|
2013-05-05 20:48:37 +08:00
|
|
|
iter = fixturefunc(**kwargs)
|
|
|
|
next = getattr(iter, "__next__", None)
|
|
|
|
if next is None:
|
|
|
|
next = getattr(iter, "next")
|
|
|
|
res = next()
|
|
|
|
def teardown():
|
|
|
|
try:
|
|
|
|
next()
|
|
|
|
except StopIteration:
|
|
|
|
pass
|
|
|
|
else:
|
2013-09-26 18:57:21 +08:00
|
|
|
fail_fixturefunc(fixturefunc,
|
2013-09-30 19:42:39 +08:00
|
|
|
"yield_fixture function has more than one 'yield'")
|
2013-05-05 20:48:37 +08:00
|
|
|
request.addfinalizer(teardown)
|
|
|
|
else:
|
2013-09-30 19:56:54 +08:00
|
|
|
if is_generator(fixturefunc):
|
|
|
|
fail_fixturefunc(fixturefunc,
|
|
|
|
msg="pytest.fixture functions cannot use ``yield``. "
|
|
|
|
"Instead write and return an inner function/generator "
|
|
|
|
"and let the consumer call and iterate over it.")
|
2013-05-05 20:48:37 +08:00
|
|
|
res = fixturefunc(**kwargs)
|
|
|
|
return res
|
|
|
|
|
2012-10-05 20:24:44 +08:00
|
|
|
class FixtureDef:
|
2012-10-05 16:21:35 +08:00
|
|
|
""" A container for a factory definition. """
|
2012-11-14 16:40:01 +08:00
|
|
|
def __init__(self, fixturemanager, baseid, argname, func, scope, params,
|
2013-09-26 18:57:21 +08:00
|
|
|
yieldctx, unittest=False):
|
2012-11-14 16:40:01 +08:00
|
|
|
self._fixturemanager = fixturemanager
|
2013-07-06 23:56:54 +08:00
|
|
|
self.baseid = baseid or ''
|
|
|
|
self.has_location = baseid is not None
|
2012-08-01 15:23:39 +08:00
|
|
|
self.func = func
|
2012-10-05 16:21:35 +08:00
|
|
|
self.argname = argname
|
|
|
|
self.scope = scope
|
|
|
|
self.scopenum = scopes.index(scope or "function")
|
|
|
|
self.params = params
|
2012-09-18 19:46:24 +08:00
|
|
|
startindex = unittest and 1 or None
|
2012-10-16 20:19:38 +08:00
|
|
|
self.argnames = getfuncargnames(func, startindex=startindex)
|
2013-09-26 18:57:21 +08:00
|
|
|
self.yieldctx = yieldctx
|
2012-10-05 16:21:35 +08:00
|
|
|
self.unittest = unittest
|
2012-08-01 15:23:39 +08:00
|
|
|
self._finalizer = []
|
|
|
|
|
|
|
|
def addfinalizer(self, finalizer):
|
|
|
|
self._finalizer.append(finalizer)
|
|
|
|
|
|
|
|
def finish(self):
|
|
|
|
while self._finalizer:
|
|
|
|
func = self._finalizer.pop()
|
|
|
|
func()
|
|
|
|
# check neccesity of next commented call
|
2012-10-05 20:24:44 +08:00
|
|
|
self._fixturemanager.removefinalizer(self.finish)
|
2012-11-19 19:42:10 +08:00
|
|
|
try:
|
|
|
|
del self.cached_result
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
2012-10-05 16:21:35 +08:00
|
|
|
|
|
|
|
def execute(self, request):
|
|
|
|
kwargs = {}
|
2012-10-16 20:19:38 +08:00
|
|
|
for newname in self.argnames:
|
2012-10-05 16:21:35 +08:00
|
|
|
kwargs[newname] = request.getfuncargvalue(newname)
|
|
|
|
if self.unittest:
|
|
|
|
result = self.func(request.instance, **kwargs)
|
|
|
|
else:
|
2012-10-20 15:59:20 +08:00
|
|
|
fixturefunc = self.func
|
|
|
|
# the fixture function needs to be bound to the actual
|
|
|
|
# request.instance so that code working with "self" behaves
|
|
|
|
# as expected. XXX request.instance should maybe return None
|
|
|
|
# instead of raising AttributeError
|
|
|
|
try:
|
|
|
|
if request.instance is not None:
|
|
|
|
fixturefunc = getimfunc(self.func)
|
|
|
|
if fixturefunc != self.func:
|
|
|
|
fixturefunc = fixturefunc.__get__(request.instance)
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
2013-09-26 18:57:21 +08:00
|
|
|
result = call_fixture_func(fixturefunc, request, kwargs,
|
|
|
|
self.yieldctx)
|
2012-11-19 19:42:10 +08:00
|
|
|
assert not hasattr(self, "cached_result")
|
2012-10-05 16:21:35 +08:00
|
|
|
self.cached_result = result
|
|
|
|
return result
|
2012-08-01 15:23:39 +08:00
|
|
|
|
2012-10-05 16:21:35 +08:00
|
|
|
def __repr__(self):
|
2013-07-06 23:56:54 +08:00
|
|
|
return ("<FixtureDef name=%r scope=%r baseid=%r module=%r>" %
|
|
|
|
(self.argname, self.scope, self.baseid, self.func.__module__))
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
def getfuncargnames(function, startindex=None):
|
|
|
|
# XXX merge with main.py's varnames
|
2012-10-08 17:22:31 +08:00
|
|
|
#assert not inspect.isclass(function)
|
2012-11-06 18:04:11 +08:00
|
|
|
realfunction = function
|
|
|
|
while hasattr(realfunction, "__wrapped__"):
|
|
|
|
realfunction = realfunction.__wrapped__
|
2012-10-08 17:22:31 +08:00
|
|
|
if startindex is None:
|
2012-09-18 20:00:47 +08:00
|
|
|
startindex = inspect.ismethod(function) and 1 or 0
|
2012-11-06 18:04:11 +08:00
|
|
|
if realfunction != function:
|
|
|
|
startindex += len(getattr(function, "patchings", []))
|
|
|
|
function = realfunction
|
2012-09-18 20:00:47 +08:00
|
|
|
argnames = inspect.getargs(py.code.getrawcode(function))[0]
|
2012-08-01 15:23:39 +08:00
|
|
|
defaults = getattr(function, 'func_defaults',
|
|
|
|
getattr(function, '__defaults__', None)) or ()
|
|
|
|
numdefaults = len(defaults)
|
|
|
|
if numdefaults:
|
2012-10-16 19:47:59 +08:00
|
|
|
return tuple(argnames[startindex:-numdefaults])
|
|
|
|
return tuple(argnames[startindex:])
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
# algorithm for sorting on a per-parametrized resource setup basis
|
|
|
|
|
|
|
|
def parametrize_sorted(items, ignore, cache, scopenum):
|
|
|
|
if scopenum >= 3:
|
|
|
|
return items
|
2013-06-28 18:57:10 +08:00
|
|
|
|
|
|
|
# we pick the first item which has a arg/param combo in the
|
|
|
|
# requested scope and sort other items with the same combo
|
|
|
|
# into "newitems" which then is a list of all items using this
|
|
|
|
# arg/param.
|
|
|
|
|
2013-06-28 18:54:10 +08:00
|
|
|
similar_items = []
|
|
|
|
other_items = []
|
2012-08-01 15:23:39 +08:00
|
|
|
slicing_argparam = None
|
2013-06-28 18:57:10 +08:00
|
|
|
slicing_index = 0
|
2012-08-01 15:23:39 +08:00
|
|
|
for item in items:
|
|
|
|
argparamlist = getfuncargparams(item, ignore, scopenum, cache)
|
|
|
|
if slicing_argparam is None and argparamlist:
|
|
|
|
slicing_argparam = argparamlist[0]
|
2013-06-28 18:54:10 +08:00
|
|
|
slicing_index = len(other_items)
|
2012-08-01 15:23:39 +08:00
|
|
|
if slicing_argparam in argparamlist:
|
2013-06-28 18:54:10 +08:00
|
|
|
similar_items.append(item)
|
2012-08-01 15:23:39 +08:00
|
|
|
else:
|
2013-06-28 18:54:10 +08:00
|
|
|
other_items.append(item)
|
2013-06-28 18:57:10 +08:00
|
|
|
|
|
|
|
if (len(similar_items) + slicing_index) > 1:
|
2012-08-01 15:23:39 +08:00
|
|
|
newignore = ignore.copy()
|
|
|
|
newignore.add(slicing_argparam)
|
2013-06-28 18:54:10 +08:00
|
|
|
part2 = parametrize_sorted(
|
|
|
|
similar_items + other_items[slicing_index:],
|
|
|
|
newignore, cache, scopenum)
|
|
|
|
part1 = parametrize_sorted(
|
|
|
|
other_items[:slicing_index], newignore,
|
|
|
|
cache, scopenum+1)
|
|
|
|
return part1 + part2
|
2012-08-01 15:23:39 +08:00
|
|
|
else:
|
2013-06-28 18:54:10 +08:00
|
|
|
other_items = parametrize_sorted(other_items, ignore, cache, scopenum+1)
|
|
|
|
return other_items + similar_items
|
2012-08-01 15:23:39 +08:00
|
|
|
|
|
|
|
def getfuncargparams(item, ignore, scopenum, cache):
|
|
|
|
""" return list of (arg,param) tuple, sorted by broader scope first. """
|
|
|
|
assert scopenum < 3 # function
|
|
|
|
try:
|
|
|
|
cs = item.callspec
|
|
|
|
except AttributeError:
|
|
|
|
return []
|
|
|
|
if scopenum == 0:
|
|
|
|
argparams = [x for x in cs.params.items() if x not in ignore
|
|
|
|
and cs._arg2scopenum[x[0]] == scopenum]
|
|
|
|
elif scopenum == 1: # module
|
|
|
|
argparams = []
|
|
|
|
for argname, param in cs.params.items():
|
|
|
|
if cs._arg2scopenum[argname] == scopenum:
|
|
|
|
key = (argname, param, item.fspath)
|
|
|
|
if key in ignore:
|
|
|
|
continue
|
|
|
|
argparams.append(key)
|
|
|
|
elif scopenum == 2: # class
|
|
|
|
argparams = []
|
|
|
|
for argname, param in cs.params.items():
|
|
|
|
if cs._arg2scopenum[argname] == scopenum:
|
|
|
|
l = cache.setdefault(item.fspath, [])
|
|
|
|
try:
|
|
|
|
i = l.index(item.cls)
|
|
|
|
except ValueError:
|
|
|
|
i = len(l)
|
|
|
|
l.append(item.cls)
|
|
|
|
key = (argname, param, item.fspath, i)
|
|
|
|
if key in ignore:
|
|
|
|
continue
|
|
|
|
argparams.append(key)
|
|
|
|
#elif scopenum == 3:
|
|
|
|
# argparams = []
|
|
|
|
# for argname, param in cs.params.items():
|
|
|
|
# if cs._arg2scopenum[argname] == scopenum:
|
|
|
|
# key = (argname, param, getfslineno(item.obj))
|
|
|
|
# if key in ignore:
|
|
|
|
# continue
|
|
|
|
# argparams.append(key)
|
|
|
|
return argparams
|
|
|
|
|
2012-08-13 19:37:14 +08:00
|
|
|
|
|
|
|
def xunitsetup(obj, name):
|
|
|
|
meth = getattr(obj, name, None)
|
2013-04-29 16:31:51 +08:00
|
|
|
if getfixturemarker(meth) is None:
|
|
|
|
return meth
|
|
|
|
|
|
|
|
def getfixturemarker(obj):
|
|
|
|
""" return fixturemarker or None if it doesn't exist or raised
|
|
|
|
exceptions."""
|
|
|
|
try:
|
|
|
|
return getattr(obj, "_pytestfixturefunction", None)
|
|
|
|
except KeyboardInterrupt:
|
|
|
|
raise
|
|
|
|
except Exception:
|
|
|
|
# some objects raise errors like request (from flask import request)
|
|
|
|
# we don't expect them to be fixture functions
|
|
|
|
return None
|
|
|
|
|