def test_positional_inject():
p = Plugin()
rdict = Request(plugin=p,
req_id=1,
method='func',
params={
'a': 1,
'b': 2,
'kwa': 3,
'kwb': 4
})
rarr = Request(
plugin=p,
req_id=1,
method='func',
params=[1, 2, 3, 4],
)
def pre_args(plugin, a, b, kwa=3, kwb=4):
assert (plugin, a, b, kwa, kwb) == (p, 1, 2, 3, 4)
def in_args(a, plugin, b, kwa=3, kwb=4):
assert (plugin, a, b, kwa, kwb) == (p, 1, 2, 3, 4)
def post_args(a, b, plugin, kwa=3, kwb=4):
assert (plugin, a, b, kwa, kwb) == (p, 1, 2, 3, 4)
def post_kwargs(a, b, kwa=3, kwb=4, plugin=None):
assert (plugin, a, b, kwa, kwb) == (p, 1, 2, 3, 4)
def in_multi_args(a, request, plugin, b, kwa=3, kwb=4):
assert request in [rarr, rdict]
assert (plugin, a, b, kwa, kwb) == (p, 1, 2, 3, 4)
def in_multi_mix_args(a, plugin, b, request=None, kwa=3, kwb=4):
assert request in [rarr, rdict]
assert (plugin, a, b, kwa, kwb) == (p, 1, 2, 3, 4)
def extra_def_arg(a, b, c, d, e=42):
""" Also uses a different name for kwa and kwb
"""
assert (a, b, c, d, e) == (1, 2, 3, 4, 42)
def count(plugin, count, request):
assert count == 42 and plugin == p
funcs = [pre_args, in_args, post_args, post_kwargs, in_multi_args]
for func, request in itertools.product(funcs, [rdict, rarr]):
p._exec_func(func, request)
p._exec_func(extra_def_arg, rarr)
p._exec_func(count,
Request(
plugin=p,
req_id=1,
method='func',
params=[42],
))
# This should fail since it is missing one positional argument
with pytest.raises(TypeError):
p._exec_func(count,
Request(plugin=p, req_id=1, method='func', params=[]))
def test_methods_errors():
"""A bunch of tests that should fail calling the methods."""
call_list = []
p = Plugin(autopatch=False)
# Fails because we haven't added the method yet
request = p._parse_request({
'id': 1,
'jsonrpc': '2.0',
'method': 'test1',
'params': {}
})
with pytest.raises(ValueError):
p._dispatch_request(request)
assert call_list == []
@p.method("test1")
def test1(name):
call_list.append(test1)
# Attempting to add it twice should fail
with pytest.raises(ValueError):
p.add_method("test1", test1)
# Fails because it is missing the 'name' argument
request = p._parse_request({
'id': 1,
'jsonrpc': '2.0',
'method': 'test1',
'params': {}
})
with pytest.raises(TypeError):
p._exec_func(test1, request)
assert call_list == []
# The same with positional arguments
request = p._parse_request({
'id': 1,
'jsonrpc': '2.0',
'method': 'test1',
'params': []
})
with pytest.raises(TypeError):
p._exec_func(test1, request)
assert call_list == []
# Fails because we have a non-matching argument
request = p._parse_request({
'id': 1,
'jsonrpc': '2.0',
'method': 'test1',
'params': {
'name': 'World',
'extra': 1
}
})
with pytest.raises(TypeError):
p._exec_func(test1, request)
assert call_list == []
request = p._parse_request({
'id': 1,
'jsonrpc': '2.0',
'method': 'test1',
'params': ['World', 1]
})
with pytest.raises(TypeError):
p._exec_func(test1, request)
assert call_list == []
