867 lines
26 KiB
Python
867 lines
26 KiB
Python
import pytest, py, os
|
|
from _pytest.core import * # noqa
|
|
from _pytest.config import get_plugin_manager
|
|
|
|
|
|
class TestBootstrapping:
|
|
def test_consider_env_fails_to_import(self, monkeypatch):
|
|
pluginmanager = PluginManager()
|
|
monkeypatch.setenv('PYTEST_PLUGINS', 'nonexisting', prepend=",")
|
|
pytest.raises(ImportError, lambda: pluginmanager.consider_env())
|
|
|
|
def test_preparse_args(self):
|
|
pluginmanager = PluginManager()
|
|
pytest.raises(ImportError, lambda:
|
|
pluginmanager.consider_preparse(["xyz", "-p", "hello123"]))
|
|
|
|
def test_plugin_prevent_register(self):
|
|
pluginmanager = PluginManager()
|
|
pluginmanager.consider_preparse(["xyz", "-p", "no:abc"])
|
|
l1 = pluginmanager.getplugins()
|
|
pluginmanager.register(42, name="abc")
|
|
l2 = pluginmanager.getplugins()
|
|
assert len(l2) == len(l1)
|
|
|
|
def test_plugin_prevent_register_unregistered_alredy_registered(self):
|
|
pluginmanager = PluginManager()
|
|
pluginmanager.register(42, name="abc")
|
|
l1 = pluginmanager.getplugins()
|
|
assert 42 in l1
|
|
pluginmanager.consider_preparse(["xyz", "-p", "no:abc"])
|
|
l2 = pluginmanager.getplugins()
|
|
assert 42 not in l2
|
|
|
|
def test_plugin_double_register(self):
|
|
pm = PluginManager()
|
|
pm.register(42, name="abc")
|
|
pytest.raises(ValueError, lambda: pm.register(42, name="abc"))
|
|
|
|
def test_plugin_skip(self, testdir, monkeypatch):
|
|
p = testdir.makepyfile(skipping1="""
|
|
import pytest
|
|
pytest.skip("hello")
|
|
""")
|
|
p.copy(p.dirpath("skipping2.py"))
|
|
monkeypatch.setenv("PYTEST_PLUGINS", "skipping2")
|
|
result = testdir.runpytest("-rw", "-p", "skipping1", "--traceconfig")
|
|
assert result.ret == 0
|
|
result.stdout.fnmatch_lines([
|
|
"WI1*skipped plugin*skipping1*hello*",
|
|
"WI1*skipped plugin*skipping2*hello*",
|
|
])
|
|
|
|
def test_consider_env_plugin_instantiation(self, testdir, monkeypatch):
|
|
pluginmanager = PluginManager()
|
|
testdir.syspathinsert()
|
|
testdir.makepyfile(xy123="#")
|
|
monkeypatch.setitem(os.environ, 'PYTEST_PLUGINS', 'xy123')
|
|
l1 = len(pluginmanager.getplugins())
|
|
pluginmanager.consider_env()
|
|
l2 = len(pluginmanager.getplugins())
|
|
assert l2 == l1 + 1
|
|
assert pluginmanager.getplugin('xy123')
|
|
pluginmanager.consider_env()
|
|
l3 = len(pluginmanager.getplugins())
|
|
assert l2 == l3
|
|
|
|
def test_consider_setuptools_instantiation(self, monkeypatch):
|
|
pkg_resources = pytest.importorskip("pkg_resources")
|
|
def my_iter(name):
|
|
assert name == "pytest11"
|
|
class EntryPoint:
|
|
name = "pytest_mytestplugin"
|
|
dist = None
|
|
def load(self):
|
|
class PseudoPlugin:
|
|
x = 42
|
|
return PseudoPlugin()
|
|
return iter([EntryPoint()])
|
|
|
|
monkeypatch.setattr(pkg_resources, 'iter_entry_points', my_iter)
|
|
pluginmanager = PluginManager()
|
|
pluginmanager.consider_setuptools_entrypoints()
|
|
plugin = pluginmanager.getplugin("mytestplugin")
|
|
assert plugin.x == 42
|
|
|
|
def test_consider_setuptools_not_installed(self, monkeypatch):
|
|
monkeypatch.setitem(py.std.sys.modules, 'pkg_resources',
|
|
py.std.types.ModuleType("pkg_resources"))
|
|
pluginmanager = PluginManager()
|
|
pluginmanager.consider_setuptools_entrypoints()
|
|
# ok, we did not explode
|
|
|
|
def test_pluginmanager_ENV_startup(self, testdir, monkeypatch):
|
|
testdir.makepyfile(pytest_x500="#")
|
|
p = testdir.makepyfile("""
|
|
import pytest
|
|
def test_hello(pytestconfig):
|
|
plugin = pytestconfig.pluginmanager.getplugin('pytest_x500')
|
|
assert plugin is not None
|
|
""")
|
|
monkeypatch.setenv('PYTEST_PLUGINS', 'pytest_x500', prepend=",")
|
|
result = testdir.runpytest(p)
|
|
assert result.ret == 0
|
|
result.stdout.fnmatch_lines(["*1 passed in*"])
|
|
|
|
def test_import_plugin_importname(self, testdir):
|
|
pluginmanager = PluginManager()
|
|
pytest.raises(ImportError, 'pluginmanager.import_plugin("qweqwex.y")')
|
|
pytest.raises(ImportError, 'pluginmanager.import_plugin("pytest_qweqwx.y")')
|
|
|
|
testdir.syspathinsert()
|
|
pluginname = "pytest_hello"
|
|
testdir.makepyfile(**{pluginname: ""})
|
|
pluginmanager.import_plugin("pytest_hello")
|
|
len1 = len(pluginmanager.getplugins())
|
|
pluginmanager.import_plugin("pytest_hello")
|
|
len2 = len(pluginmanager.getplugins())
|
|
assert len1 == len2
|
|
plugin1 = pluginmanager.getplugin("pytest_hello")
|
|
assert plugin1.__name__.endswith('pytest_hello')
|
|
plugin2 = pluginmanager.getplugin("pytest_hello")
|
|
assert plugin2 is plugin1
|
|
|
|
def test_import_plugin_dotted_name(self, testdir):
|
|
pluginmanager = PluginManager()
|
|
pytest.raises(ImportError, 'pluginmanager.import_plugin("qweqwex.y")')
|
|
pytest.raises(ImportError, 'pluginmanager.import_plugin("pytest_qweqwex.y")')
|
|
|
|
testdir.syspathinsert()
|
|
testdir.mkpydir("pkg").join("plug.py").write("x=3")
|
|
pluginname = "pkg.plug"
|
|
pluginmanager.import_plugin(pluginname)
|
|
mod = pluginmanager.getplugin("pkg.plug")
|
|
assert mod.x == 3
|
|
|
|
def test_consider_module(self, testdir):
|
|
pluginmanager = PluginManager()
|
|
testdir.syspathinsert()
|
|
testdir.makepyfile(pytest_p1="#")
|
|
testdir.makepyfile(pytest_p2="#")
|
|
mod = py.std.types.ModuleType("temp")
|
|
mod.pytest_plugins = ["pytest_p1", "pytest_p2"]
|
|
pluginmanager.consider_module(mod)
|
|
assert pluginmanager.getplugin("pytest_p1").__name__ == "pytest_p1"
|
|
assert pluginmanager.getplugin("pytest_p2").__name__ == "pytest_p2"
|
|
|
|
def test_consider_module_import_module(self, testdir):
|
|
mod = py.std.types.ModuleType("x")
|
|
mod.pytest_plugins = "pytest_a"
|
|
aplugin = testdir.makepyfile(pytest_a="#")
|
|
pluginmanager = get_plugin_manager()
|
|
reprec = testdir.make_hook_recorder(pluginmanager)
|
|
#syspath.prepend(aplugin.dirpath())
|
|
py.std.sys.path.insert(0, str(aplugin.dirpath()))
|
|
pluginmanager.consider_module(mod)
|
|
call = reprec.getcall(pluginmanager.hook.pytest_plugin_registered.name)
|
|
assert call.plugin.__name__ == "pytest_a"
|
|
|
|
# check that it is not registered twice
|
|
pluginmanager.consider_module(mod)
|
|
l = reprec.getcalls("pytest_plugin_registered")
|
|
assert len(l) == 1
|
|
|
|
def test_config_sets_conftesthandle_onimport(self, testdir):
|
|
config = testdir.parseconfig([])
|
|
assert config._conftest._onimport == config._onimportconftest
|
|
|
|
def test_consider_conftest_deps(self, testdir):
|
|
mod = testdir.makepyfile("pytest_plugins='xyz'").pyimport()
|
|
pp = PluginManager()
|
|
pytest.raises(ImportError, lambda: pp.consider_conftest(mod))
|
|
|
|
def test_pm(self):
|
|
pp = PluginManager()
|
|
class A: pass
|
|
a1, a2 = A(), A()
|
|
pp.register(a1)
|
|
assert pp.isregistered(a1)
|
|
pp.register(a2, "hello")
|
|
assert pp.isregistered(a2)
|
|
l = pp.getplugins()
|
|
assert a1 in l
|
|
assert a2 in l
|
|
assert pp.getplugin('hello') == a2
|
|
pp.unregister(a1)
|
|
assert not pp.isregistered(a1)
|
|
|
|
def test_pm_ordering(self):
|
|
pp = PluginManager()
|
|
class A: pass
|
|
a1, a2 = A(), A()
|
|
pp.register(a1)
|
|
pp.register(a2, "hello")
|
|
l = pp.getplugins()
|
|
assert l.index(a1) < l.index(a2)
|
|
a3 = A()
|
|
pp.register(a3, prepend=True)
|
|
l = pp.getplugins()
|
|
assert l.index(a3) == 0
|
|
|
|
def test_register_imported_modules(self):
|
|
pp = PluginManager()
|
|
mod = py.std.types.ModuleType("x.y.pytest_hello")
|
|
pp.register(mod)
|
|
assert pp.isregistered(mod)
|
|
l = pp.getplugins()
|
|
assert mod in l
|
|
pytest.raises(ValueError, "pp.register(mod)")
|
|
pytest.raises(ValueError, lambda: pp.register(mod))
|
|
#assert not pp.isregistered(mod2)
|
|
assert pp.getplugins() == l
|
|
|
|
def test_canonical_import(self, monkeypatch):
|
|
mod = py.std.types.ModuleType("pytest_xyz")
|
|
monkeypatch.setitem(py.std.sys.modules, 'pytest_xyz', mod)
|
|
pp = PluginManager()
|
|
pp.import_plugin('pytest_xyz')
|
|
assert pp.getplugin('pytest_xyz') == mod
|
|
assert pp.isregistered(mod)
|
|
|
|
def test_register_mismatch_method(self):
|
|
pp = get_plugin_manager()
|
|
class hello:
|
|
def pytest_gurgel(self):
|
|
pass
|
|
pytest.raises(Exception, lambda: pp.register(hello()))
|
|
|
|
def test_register_mismatch_arg(self):
|
|
pp = get_plugin_manager()
|
|
class hello:
|
|
def pytest_configure(self, asd):
|
|
pass
|
|
pytest.raises(Exception, lambda: pp.register(hello()))
|
|
|
|
def test_register(self):
|
|
pm = get_plugin_manager()
|
|
class MyPlugin:
|
|
pass
|
|
my = MyPlugin()
|
|
pm.register(my)
|
|
assert pm.getplugins()
|
|
my2 = MyPlugin()
|
|
pm.register(my2)
|
|
assert pm.getplugins()[-2:] == [my, my2]
|
|
|
|
assert pm.isregistered(my)
|
|
assert pm.isregistered(my2)
|
|
pm.unregister(my)
|
|
assert not pm.isregistered(my)
|
|
assert pm.getplugins()[-1:] == [my2]
|
|
|
|
def test_listattr(self):
|
|
plugins = PluginManager()
|
|
class api1:
|
|
x = 41
|
|
class api2:
|
|
x = 42
|
|
class api3:
|
|
x = 43
|
|
plugins.register(api1())
|
|
plugins.register(api2())
|
|
plugins.register(api3())
|
|
l = list(plugins.listattr('x'))
|
|
assert l == [41, 42, 43]
|
|
|
|
def test_hook_tracing(self):
|
|
pm = get_plugin_manager()
|
|
saveindent = []
|
|
class api1:
|
|
x = 41
|
|
def pytest_plugin_registered(self, plugin):
|
|
saveindent.append(pm.trace.root.indent)
|
|
raise ValueError(42)
|
|
l = []
|
|
pm.set_tracing(l.append)
|
|
indent = pm.trace.root.indent
|
|
p = api1()
|
|
pm.register(p)
|
|
|
|
assert pm.trace.root.indent == indent
|
|
assert len(l) == 2
|
|
assert 'pytest_plugin_registered' in l[0]
|
|
assert 'finish' in l[1]
|
|
pytest.raises(ValueError, lambda: pm.register(api1()))
|
|
assert pm.trace.root.indent == indent
|
|
assert saveindent[0] > indent
|
|
|
|
class TestPytestPluginInteractions:
|
|
|
|
def test_addhooks_conftestplugin(self, testdir):
|
|
testdir.makepyfile(newhooks="""
|
|
def pytest_myhook(xyz):
|
|
"new hook"
|
|
""")
|
|
conf = testdir.makeconftest("""
|
|
import sys ; sys.path.insert(0, '.')
|
|
import newhooks
|
|
def pytest_addhooks(pluginmanager):
|
|
pluginmanager.addhooks(newhooks)
|
|
def pytest_myhook(xyz):
|
|
return xyz + 1
|
|
""")
|
|
config = get_plugin_manager().config
|
|
config._conftest.importconftest(conf)
|
|
print(config.pluginmanager.getplugins())
|
|
res = config.hook.pytest_myhook(xyz=10)
|
|
assert res == [11]
|
|
|
|
def test_addhooks_nohooks(self, testdir):
|
|
testdir.makeconftest("""
|
|
import sys
|
|
def pytest_addhooks(pluginmanager):
|
|
pluginmanager.addhooks(sys)
|
|
""")
|
|
res = testdir.runpytest()
|
|
assert res.ret != 0
|
|
res.stderr.fnmatch_lines([
|
|
"*did not find*sys*"
|
|
])
|
|
|
|
def test_namespace_early_from_import(self, testdir):
|
|
p = testdir.makepyfile("""
|
|
from pytest import Item
|
|
from pytest import Item as Item2
|
|
assert Item is Item2
|
|
""")
|
|
result = testdir.runpython(p)
|
|
assert result.ret == 0
|
|
|
|
def test_do_ext_namespace(self, testdir):
|
|
testdir.makeconftest("""
|
|
def pytest_namespace():
|
|
return {'hello': 'world'}
|
|
""")
|
|
p = testdir.makepyfile("""
|
|
from pytest import hello
|
|
import pytest
|
|
def test_hello():
|
|
assert hello == "world"
|
|
assert 'hello' in pytest.__all__
|
|
""")
|
|
reprec = testdir.inline_run(p)
|
|
reprec.assertoutcome(passed=1)
|
|
|
|
def test_do_option_postinitialize(self, testdir):
|
|
config = testdir.parseconfigure()
|
|
assert not hasattr(config.option, 'test123')
|
|
p = testdir.makepyfile("""
|
|
def pytest_addoption(parser):
|
|
parser.addoption('--test123', action="store_true",
|
|
default=True)
|
|
""")
|
|
config._conftest.importconftest(p)
|
|
assert config.option.test123
|
|
|
|
def test_configure(self, testdir):
|
|
config = testdir.parseconfig()
|
|
l = []
|
|
class A:
|
|
def pytest_configure(self, config):
|
|
l.append(self)
|
|
|
|
config.pluginmanager.register(A())
|
|
assert len(l) == 0
|
|
config.do_configure()
|
|
assert len(l) == 1
|
|
config.pluginmanager.register(A()) # leads to a configured() plugin
|
|
assert len(l) == 2
|
|
assert l[0] != l[1]
|
|
|
|
config.do_unconfigure()
|
|
config.pluginmanager.register(A())
|
|
assert len(l) == 2
|
|
|
|
# lower level API
|
|
|
|
def test_listattr(self):
|
|
pluginmanager = PluginManager()
|
|
class My2:
|
|
x = 42
|
|
pluginmanager.register(My2())
|
|
assert not pluginmanager.listattr("hello")
|
|
assert pluginmanager.listattr("x") == [42]
|
|
|
|
def test_listattr_tryfirst(self):
|
|
class P1:
|
|
@pytest.mark.tryfirst
|
|
def m(self):
|
|
return 17
|
|
|
|
class P2:
|
|
def m(self):
|
|
return 23
|
|
class P3:
|
|
def m(self):
|
|
return 19
|
|
|
|
pluginmanager = PluginManager()
|
|
p1 = P1()
|
|
p2 = P2()
|
|
p3 = P3()
|
|
pluginmanager.register(p1)
|
|
pluginmanager.register(p2)
|
|
pluginmanager.register(p3)
|
|
methods = pluginmanager.listattr('m')
|
|
assert methods == [p2.m, p3.m, p1.m]
|
|
del P1.m.__dict__['tryfirst']
|
|
pytest.mark.trylast(getattr(P2.m, 'im_func', P2.m))
|
|
methods = pluginmanager.listattr('m')
|
|
assert methods == [p2.m, p1.m, p3.m]
|
|
|
|
|
|
def test_namespace_has_default_and_env_plugins(testdir):
|
|
p = testdir.makepyfile("""
|
|
import pytest
|
|
pytest.mark
|
|
""")
|
|
result = testdir.runpython(p)
|
|
assert result.ret == 0
|
|
|
|
def test_varnames():
|
|
def f(x):
|
|
i = 3 # noqa
|
|
class A:
|
|
def f(self, y):
|
|
pass
|
|
class B(object):
|
|
def __call__(self, z):
|
|
pass
|
|
assert varnames(f) == ("x",)
|
|
assert varnames(A().f) == ('y',)
|
|
assert varnames(B()) == ('z',)
|
|
|
|
def test_varnames_default():
|
|
def f(x, y=3):
|
|
pass
|
|
assert varnames(f) == ("x",)
|
|
|
|
def test_varnames_class():
|
|
class C:
|
|
def __init__(self, x):
|
|
pass
|
|
class D:
|
|
pass
|
|
assert varnames(C) == ("x",)
|
|
assert varnames(D) == ()
|
|
|
|
class TestMultiCall:
|
|
def test_uses_copy_of_methods(self):
|
|
l = [lambda: 42]
|
|
mc = MultiCall(l, {})
|
|
repr(mc)
|
|
l[:] = []
|
|
res = mc.execute()
|
|
return res == 42
|
|
|
|
def test_call_passing(self):
|
|
class P1:
|
|
def m(self, __multicall__, x):
|
|
assert len(__multicall__.results) == 1
|
|
assert not __multicall__.methods
|
|
return 17
|
|
|
|
class P2:
|
|
def m(self, __multicall__, x):
|
|
assert __multicall__.results == []
|
|
assert __multicall__.methods
|
|
return 23
|
|
|
|
p1 = P1()
|
|
p2 = P2()
|
|
multicall = MultiCall([p1.m, p2.m], {'x': 23})
|
|
assert "23" in repr(multicall)
|
|
reslist = multicall.execute()
|
|
assert len(reslist) == 2
|
|
# ensure reversed order
|
|
assert reslist == [23, 17]
|
|
|
|
def test_keyword_args(self):
|
|
def f(x):
|
|
return x + 1
|
|
class A:
|
|
def f(self, x, y):
|
|
return x + y
|
|
multicall = MultiCall([f, A().f], dict(x=23, y=24))
|
|
assert "'x': 23" in repr(multicall)
|
|
assert "'y': 24" in repr(multicall)
|
|
reslist = multicall.execute()
|
|
assert reslist == [24+23, 24]
|
|
assert "2 results" in repr(multicall)
|
|
|
|
def test_keyword_args_with_defaultargs(self):
|
|
def f(x, z=1):
|
|
return x + z
|
|
reslist = MultiCall([f], dict(x=23, y=24)).execute()
|
|
assert reslist == [24]
|
|
|
|
def test_tags_call_error(self):
|
|
multicall = MultiCall([lambda x: x], {})
|
|
pytest.raises(KeyError, multicall.execute)
|
|
|
|
def test_call_subexecute(self):
|
|
def m(__multicall__):
|
|
subresult = __multicall__.execute()
|
|
return subresult + 1
|
|
|
|
def n():
|
|
return 1
|
|
|
|
call = MultiCall([n, m], {}, firstresult=True)
|
|
res = call.execute()
|
|
assert res == 2
|
|
|
|
def test_call_none_is_no_result(self):
|
|
def m1():
|
|
return 1
|
|
def m2():
|
|
return None
|
|
res = MultiCall([m1, m2], {}, firstresult=True).execute()
|
|
assert res == 1
|
|
res = MultiCall([m1, m2], {}).execute()
|
|
assert res == [1]
|
|
|
|
def test_hookwrapper(self):
|
|
l = []
|
|
def m1():
|
|
l.append("m1 init")
|
|
yield None
|
|
l.append("m1 finish")
|
|
m1.hookwrapper = True
|
|
|
|
def m2():
|
|
l.append("m2")
|
|
return 2
|
|
res = MultiCall([m2, m1], {}).execute()
|
|
assert res == [2]
|
|
assert l == ["m1 init", "m2", "m1 finish"]
|
|
l[:] = []
|
|
res = MultiCall([m2, m1], {}, firstresult=True).execute()
|
|
assert res == 2
|
|
assert l == ["m1 init", "m2", "m1 finish"]
|
|
|
|
def test_hookwrapper_order(self):
|
|
l = []
|
|
def m1():
|
|
l.append("m1 init")
|
|
yield 1
|
|
l.append("m1 finish")
|
|
m1.hookwrapper = True
|
|
|
|
def m2():
|
|
l.append("m2 init")
|
|
yield 2
|
|
l.append("m2 finish")
|
|
m2.hookwrapper = True
|
|
res = MultiCall([m2, m1], {}).execute()
|
|
assert res == []
|
|
assert l == ["m1 init", "m2 init", "m2 finish", "m1 finish"]
|
|
|
|
def test_listattr_hookwrapper_ordering(self):
|
|
class P1:
|
|
@pytest.mark.hookwrapper
|
|
def m(self):
|
|
return 17
|
|
|
|
class P2:
|
|
def m(self):
|
|
return 23
|
|
|
|
class P3:
|
|
@pytest.mark.tryfirst
|
|
def m(self):
|
|
return 19
|
|
|
|
pluginmanager = PluginManager()
|
|
p1 = P1()
|
|
p2 = P2()
|
|
p3 = P3()
|
|
pluginmanager.register(p1)
|
|
pluginmanager.register(p2)
|
|
pluginmanager.register(p3)
|
|
methods = pluginmanager.listattr('m')
|
|
assert methods == [p2.m, p3.m, p1.m]
|
|
## listattr keeps a cache and deleting
|
|
## a function attribute requires clearing it
|
|
#pluginmanager._listattrcache.clear()
|
|
#del P1.m.__dict__['tryfirst']
|
|
|
|
def test_hookwrapper_not_yield(self):
|
|
def m1():
|
|
pass
|
|
m1.hookwrapper = True
|
|
|
|
mc = MultiCall([m1], {})
|
|
with pytest.raises(TypeError):
|
|
mc.execute()
|
|
|
|
def test_hookwrapper_too_many_yield(self):
|
|
def m1():
|
|
yield 1
|
|
yield 2
|
|
m1.hookwrapper = True
|
|
|
|
mc = MultiCall([m1], {})
|
|
with pytest.raises(RuntimeError) as ex:
|
|
mc.execute()
|
|
assert "m1" in str(ex.value)
|
|
assert "test_core.py:" in str(ex.value)
|
|
|
|
|
|
class TestHookRelay:
|
|
def test_happypath(self):
|
|
class Api:
|
|
def hello(self, arg):
|
|
"api hook 1"
|
|
pm = PluginManager([Api], prefix="he")
|
|
hook = pm.hook
|
|
assert hasattr(hook, 'hello')
|
|
assert repr(hook.hello).find("hello") != -1
|
|
class Plugin:
|
|
def hello(self, arg):
|
|
return arg + 1
|
|
plugin = Plugin()
|
|
pm.register(plugin)
|
|
l = hook.hello(arg=3)
|
|
assert l == [4]
|
|
assert not hasattr(hook, 'world')
|
|
pm.unregister(plugin)
|
|
assert hook.hello(arg=3) == []
|
|
|
|
def test_argmismatch(self):
|
|
class Api:
|
|
def hello(self, arg):
|
|
"api hook 1"
|
|
pm = PluginManager(Api, prefix="he")
|
|
class Plugin:
|
|
def hello(self, argwrong):
|
|
return arg + 1
|
|
with pytest.raises(PluginValidationError) as exc:
|
|
pm.register(Plugin())
|
|
assert "argwrong" in str(exc.value)
|
|
|
|
def test_only_kwargs(self):
|
|
pm = PluginManager()
|
|
class Api:
|
|
def hello(self, arg):
|
|
"api hook 1"
|
|
mcm = HookRelay(hookspecs=Api, pm=pm, prefix="he")
|
|
pytest.raises(TypeError, lambda: mcm.hello(3))
|
|
|
|
def test_firstresult_definition(self):
|
|
class Api:
|
|
def hello(self, arg):
|
|
"api hook 1"
|
|
hello.firstresult = True
|
|
pm = PluginManager([Api], "he")
|
|
class Plugin:
|
|
def hello(self, arg):
|
|
return arg + 1
|
|
pm.register(Plugin())
|
|
res = pm.hook.hello(arg=3)
|
|
assert res == 4
|
|
|
|
class TestTracer:
|
|
def test_simple(self):
|
|
from _pytest.core import TagTracer
|
|
rootlogger = TagTracer()
|
|
log = rootlogger.get("pytest")
|
|
log("hello")
|
|
l = []
|
|
rootlogger.setwriter(l.append)
|
|
log("world")
|
|
assert len(l) == 1
|
|
assert l[0] == "world [pytest]\n"
|
|
sublog = log.get("collection")
|
|
sublog("hello")
|
|
assert l[1] == "hello [pytest:collection]\n"
|
|
|
|
def test_indent(self):
|
|
from _pytest.core import TagTracer
|
|
rootlogger = TagTracer()
|
|
log = rootlogger.get("1")
|
|
l = []
|
|
log.root.setwriter(lambda arg: l.append(arg))
|
|
log("hello")
|
|
log.root.indent += 1
|
|
log("line1")
|
|
log("line2")
|
|
log.root.indent += 1
|
|
log("line3")
|
|
log("line4")
|
|
log.root.indent -= 1
|
|
log("line5")
|
|
log.root.indent -= 1
|
|
log("last")
|
|
assert len(l) == 7
|
|
names = [x[:x.rfind(' [')] for x in l]
|
|
assert names == ['hello', ' line1', ' line2',
|
|
' line3', ' line4', ' line5', 'last']
|
|
|
|
def test_readable_output_dictargs(self):
|
|
from _pytest.core import TagTracer
|
|
rootlogger = TagTracer()
|
|
|
|
out = rootlogger.format_message(['test'], [1])
|
|
assert out == ['1 [test]\n']
|
|
|
|
out2= rootlogger.format_message(['test'], ['test', {'a':1}])
|
|
assert out2 ==[
|
|
'test [test]\n',
|
|
' a: 1\n'
|
|
]
|
|
|
|
def test_setprocessor(self):
|
|
from _pytest.core import TagTracer
|
|
rootlogger = TagTracer()
|
|
log = rootlogger.get("1")
|
|
log2 = log.get("2")
|
|
assert log2.tags == tuple("12")
|
|
l = []
|
|
rootlogger.setprocessor(tuple("12"), lambda *args: l.append(args))
|
|
log("not seen")
|
|
log2("seen")
|
|
assert len(l) == 1
|
|
tags, args = l[0]
|
|
assert "1" in tags
|
|
assert "2" in tags
|
|
assert args == ("seen",)
|
|
l2 = []
|
|
rootlogger.setprocessor("1:2", lambda *args: l2.append(args))
|
|
log2("seen")
|
|
tags, args = l2[0]
|
|
assert args == ("seen",)
|
|
|
|
|
|
def test_setmyprocessor(self):
|
|
from _pytest.core import TagTracer
|
|
rootlogger = TagTracer()
|
|
log = rootlogger.get("1")
|
|
log2 = log.get("2")
|
|
l = []
|
|
log2.setmyprocessor(lambda *args: l.append(args))
|
|
log("not seen")
|
|
assert not l
|
|
log2(42)
|
|
assert len(l) == 1
|
|
tags, args = l[0]
|
|
assert "1" in tags
|
|
assert "2" in tags
|
|
assert args == (42,)
|
|
|
|
def test_default_markers(testdir):
|
|
result = testdir.runpytest("--markers")
|
|
result.stdout.fnmatch_lines([
|
|
"*tryfirst*first*",
|
|
"*trylast*last*",
|
|
])
|
|
|
|
def test_importplugin_issue375(testdir):
|
|
testdir.makepyfile(qwe="import aaaa")
|
|
excinfo = pytest.raises(ImportError, lambda: importplugin("qwe"))
|
|
assert "qwe" not in str(excinfo.value)
|
|
assert "aaaa" in str(excinfo.value)
|
|
|
|
class TestWrapMethod:
|
|
def test_basic_happypath(self):
|
|
class A:
|
|
def f(self):
|
|
return "A.f"
|
|
|
|
l = []
|
|
def f(self):
|
|
l.append(1)
|
|
box = yield
|
|
assert box.result == "A.f"
|
|
l.append(2)
|
|
undo = add_method_wrapper(A, f)
|
|
|
|
assert A().f() == "A.f"
|
|
assert l == [1,2]
|
|
undo()
|
|
l[:] = []
|
|
assert A().f() == "A.f"
|
|
assert l == []
|
|
|
|
def test_no_yield(self):
|
|
class A:
|
|
def method(self):
|
|
return
|
|
|
|
def method(self):
|
|
if 0:
|
|
yield
|
|
|
|
add_method_wrapper(A, method)
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
A().method()
|
|
|
|
assert "method" in str(excinfo.value)
|
|
assert "did not yield" in str(excinfo.value)
|
|
|
|
def test_method_raises(self):
|
|
class A:
|
|
def error(self, val):
|
|
raise ValueError(val)
|
|
|
|
l = []
|
|
def error(self, val):
|
|
l.append(val)
|
|
yield
|
|
l.append(None)
|
|
|
|
undo = add_method_wrapper(A, error)
|
|
|
|
with pytest.raises(ValueError):
|
|
A().error(42)
|
|
assert l == [42, None]
|
|
undo()
|
|
l[:] = []
|
|
with pytest.raises(ValueError):
|
|
A().error(42)
|
|
assert l == []
|
|
|
|
def test_controller_swallows_method_raises(self):
|
|
class A:
|
|
def error(self, val):
|
|
raise ValueError(val)
|
|
|
|
def error(self, val):
|
|
box = yield
|
|
box.force_result(2)
|
|
|
|
add_method_wrapper(A, error)
|
|
assert A().error(42) == 2
|
|
|
|
def test_reraise_on_controller_StopIteration(self):
|
|
class A:
|
|
def error(self, val):
|
|
raise ValueError(val)
|
|
|
|
def error(self, val):
|
|
try:
|
|
yield
|
|
except ValueError:
|
|
pass
|
|
|
|
add_method_wrapper(A, error)
|
|
with pytest.raises(ValueError):
|
|
A().error(42)
|
|
|
|
@pytest.mark.xfail(reason="if needed later")
|
|
def test_modify_call_args(self):
|
|
class A:
|
|
def error(self, val1, val2):
|
|
raise ValueError(val1+val2)
|
|
|
|
l = []
|
|
def error(self):
|
|
box = yield (1,), {'val2': 2}
|
|
assert box.excinfo[1].args == (3,)
|
|
l.append(1)
|
|
|
|
add_method_wrapper(A, error)
|
|
with pytest.raises(ValueError):
|
|
A().error()
|
|
assert l == [1]
|