Source code
Revision control
Copy as Markdown
Other Tools
import pytest
from pluggy import HookCallError, HookspecMarker, HookimplMarker
from pluggy._hooks import HookImpl
from pluggy._callers import _multicall
hookspec = HookspecMarker("example")
hookimpl = HookimplMarker("example")
def MC(methods, kwargs, firstresult=False):
caller = _multicall
hookfuncs = []
for method in methods:
f = HookImpl(None, "<temp>", method, method.example_impl)
hookfuncs.append(f)
return caller("foo", hookfuncs, kwargs, firstresult)
def test_keyword_args():
@hookimpl
def f(x):
return x + 1
class A:
@hookimpl
def f(self, x, y):
return x + y
reslist = MC([f, A().f], dict(x=23, y=24))
assert reslist == [24 + 23, 24]
def test_keyword_args_with_defaultargs():
@hookimpl
def f(x, z=1):
return x + z
reslist = MC([f], dict(x=23, y=24))
assert reslist == [24]
def test_tags_call_error():
@hookimpl
def f(x):
return x
with pytest.raises(HookCallError):
MC([f], {})
def test_call_none_is_no_result():
@hookimpl
def m1():
return 1
@hookimpl
def m2():
return None
res = MC([m1, m2], {}, firstresult=True)
assert res == 1
res = MC([m1, m2], {}, {})
assert res == [1]
def test_hookwrapper():
out = []
@hookimpl(hookwrapper=True)
def m1():
out.append("m1 init")
yield None
out.append("m1 finish")
@hookimpl
def m2():
out.append("m2")
return 2
res = MC([m2, m1], {})
assert res == [2]
assert out == ["m1 init", "m2", "m1 finish"]
out[:] = []
res = MC([m2, m1], {}, firstresult=True)
assert res == 2
assert out == ["m1 init", "m2", "m1 finish"]
def test_hookwrapper_order():
out = []
@hookimpl(hookwrapper=True)
def m1():
out.append("m1 init")
yield 1
out.append("m1 finish")
@hookimpl(hookwrapper=True)
def m2():
out.append("m2 init")
yield 2
out.append("m2 finish")
res = MC([m2, m1], {})
assert res == []
assert out == ["m1 init", "m2 init", "m2 finish", "m1 finish"]
def test_hookwrapper_not_yield():
@hookimpl(hookwrapper=True)
def m1():
pass
with pytest.raises(TypeError):
MC([m1], {})
def test_hookwrapper_too_many_yield():
@hookimpl(hookwrapper=True)
def m1():
yield 1
yield 2
with pytest.raises(RuntimeError) as ex:
MC([m1], {})
assert "m1" in str(ex.value)
assert (__file__ + ":") in str(ex.value)
@pytest.mark.parametrize("exc", [ValueError, SystemExit])
def test_hookwrapper_exception(exc):
out = []
@hookimpl(hookwrapper=True)
def m1():
out.append("m1 init")
yield None
out.append("m1 finish")
@hookimpl
def m2():
raise exc
with pytest.raises(exc):
MC([m2, m1], {})
assert out == ["m1 init", "m1 finish"]