1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
|
import py, pytest, sys, os, textwrap
from inspect import isclass
if hasattr(sys, 'setrecursionlimit'):
# some tests fail otherwise
sys.setrecursionlimit(2000)
LOOK_FOR_PYTHON3 = 'python3.5'
PYTHON3 = os.getenv('PYTHON3') or py.path.local.sysfind(LOOK_FOR_PYTHON3)
if PYTHON3 is not None:
PYTHON3 = str(PYTHON3)
# pytest settings
rsyncdirs = ['.', '../lib-python', '../lib_pypy', '../demo']
rsyncignore = ['_cache']
try:
from hypothesis import settings, __version__
except ImportError:
pass
else:
if __version__[:2] < '3.6':
s = settings(deadline=None)
settings.register_profile('default', s)
else:
settings.register_profile('default', deadline=None)
settings.load_profile('default')
# PyPy's command line extra options (these are added
# to py.test's standard options)
#
option = None
def braindead_deindent(self):
"""monkeypatch that wont end up doing stupid in the python tokenizer"""
text = '\n'.join(self.lines)
short = py.std.textwrap.dedent(text)
newsource = py.code.Source()
newsource.lines[:] = short.splitlines()
return newsource
py.code.Source.deindent = braindead_deindent
def pytest_report_header():
return "pytest-%s from %s" % (pytest.__version__, pytest.__file__)
def pytest_addhooks(pluginmanager):
if sys.version_info < (3,):
from rpython.conftest import LeakFinder
pluginmanager.register(LeakFinder())
def pytest_configure(config):
global option
option = config.option
def py3k_skip(message):
py.test.skip('[py3k] %s' % message)
py.test.py3k_skip = py3k_skip
def pytest_addoption(parser):
from rpython.conftest import pytest_addoption
pytest_addoption(parser)
group = parser.getgroup("pypy options")
group.addoption('-A', '--runappdirect', action="store_true",
default=False, dest="runappdirect",
help="run applevel tests directly on the python interpreter " +
"specified by --python")
group.addoption('--python', type="string", default=PYTHON3,
help="python interpreter to run appdirect tests with")
group.addoption('--direct', action="store_true",
default=False, dest="rundirect",
help="run pexpect tests directly")
group.addoption('--raise-operr', action="store_true",
default=False, dest="raise_operr",
help="Show the interp-level OperationError in app-level tests")
@pytest.fixture(scope='function')
def space(request):
from pypy.tool.pytest.objspace import gettestobjspace
spaceconfig = getattr(request.cls, 'spaceconfig', {})
return gettestobjspace(**spaceconfig)
#
# Interfacing/Integrating with py.test's collection process
#
#
def ensure_pytest_builtin_helpers(helpers='skip raises'.split()):
""" hack (py.test.) raises and skip into builtins, needed
for applevel tests to run directly on cpython but
apparently earlier on "raises" was already added
to module's globals.
"""
try:
import builtins
except ImportError:
import __builtin__ as builtins
for helper in helpers:
if not hasattr(builtins, helper):
setattr(builtins, helper, getattr(py.test, helper))
def pytest_sessionstart(session):
""" before session.main() is called. """
# stick py.test raise in module globals -- carefully
ensure_pytest_builtin_helpers()
def pytest_pycollect_makemodule(path, parent):
return PyPyModule(path, parent)
def is_applevel(item):
from pypy.tool.pytest.apptest import AppTestFunction
return isinstance(item, AppTestFunction)
def pytest_collection_modifyitems(config, items):
if config.option.runappdirect:
return
for item in items:
if isinstance(item, py.test.Function):
if is_applevel(item):
item.add_marker('applevel')
else:
item.add_marker('interplevel')
class PyPyModule(py.test.collect.Module):
""" we take care of collecting classes both at app level
and at interp-level (because we need to stick a space
at the class) ourselves.
"""
def accept_regular_test(self):
if self.config.option.runappdirect:
# only collect regular tests if we are in an 'app_test' directory,
# or in test_lib_pypy
for name in self.listnames():
if "app_test" in name or "test_lib_pypy" in name:
return True
return False
return True
def funcnamefilter(self, name):
if name.startswith('test_'):
return self.accept_regular_test()
if name.startswith('app_test_'):
return True
return False
def classnamefilter(self, name):
if name.startswith('Test'):
return self.accept_regular_test()
if name.startswith('AppTest'):
return True
return False
def makeitem(self, name, obj):
if isclass(obj) and self.classnamefilter(name):
if name.startswith('AppTest'):
from pypy.tool.pytest.apptest import AppClassCollector
return AppClassCollector(name, parent=self)
elif hasattr(obj, 'func_code') and self.funcnamefilter(name):
if name.startswith('app_test_'):
assert not obj.func_code.co_flags & 32, \
"generator app level functions? you must be joking"
from pypy.tool.pytest.apptest import AppTestFunction
return AppTestFunction(name, parent=self)
return super(PyPyModule, self).makeitem(name, obj)
def skip_on_missing_buildoption(**ropts):
__tracebackhide__ = True
import sys
options = getattr(sys, 'pypy_translation_info', None)
if options is None:
py.test.skip("not running on translated pypy3 "
"(btw, i would need options: %s)" %
(ropts,))
for opt in ropts:
if not options.has_key(opt) or options[opt] != ropts[opt]:
break
else:
return
py.test.skip("need translated pypy3 with: %s, got %s"
%(ropts,options))
class LazyObjSpaceGetter(object):
def __get__(self, obj, cls=None):
from pypy.tool.pytest.objspace import gettestobjspace
space = gettestobjspace()
if cls:
cls.space = space
return space
@pytest.hookimpl(tryfirst=True)
def pytest_runtest_setup(item):
if isinstance(item, py.test.collect.Function):
appclass = item.getparent(py.test.Class)
if appclass is not None:
# Make cls.space and cls.runappdirect available in tests.
spaceconfig = getattr(appclass.obj, 'spaceconfig', None)
if spaceconfig is not None:
from pypy.tool.pytest.objspace import gettestobjspace
appclass.obj.space = gettestobjspace(**spaceconfig)
else:
appclass.obj.space = LazyObjSpaceGetter()
appclass.obj.runappdirect = option.runappdirect
def pytest_ignore_collect(path):
return path.check(link=1)
|