Skip to content

Commit

Permalink
Cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
kclowes committed Nov 11, 2019
1 parent d27990e commit 4d2a0d4
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 13 deletions.
21 changes: 10 additions & 11 deletions tests/core/method-class/test_method.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ def test_default_input_munger_with_input_parameters_exception():


@pytest.mark.parametrize(
"method_config,args,kwargs,expected_req_result,expected_result_formatters_len",
"method_config,args,kwargs,expected_request_result,expected_result_formatters_len",
(
(
{
Expand Down Expand Up @@ -224,17 +224,17 @@ def test_process_params(
method_config,
args,
kwargs,
expected_req_result,
expected_request_result,
expected_result_formatters_len):

if isclass(expected_req_result) and issubclass(expected_req_result, Exception):
with pytest.raises(expected_req_result):
if isclass(expected_request_result) and issubclass(expected_request_result, Exception):
with pytest.raises(expected_request_result):
method = Method(**method_config)
req_params, output_formatter = method.process_params(object(), *args, **kwargs)
request_params, output_formatter = method.process_params(object(), *args, **kwargs)
else:
method = Method(**method_config)
req_params, output_formatter = method.process_params(object(), *args, **kwargs)
assert req_params == expected_req_result
request_params, output_formatter = method.process_params(object(), *args, **kwargs)
assert request_params == expected_request_result
first_formatter = (output_formatter[0].first,)
all_other_formatters = output_formatter[0].funcs
assert len(first_formatter + all_other_formatters) == expected_result_formatters_len
Expand All @@ -256,7 +256,7 @@ def formatter(params):

class FakeModule(ModuleV2):
method = Method(
'eth_getBalance',
'eth_method',
mungers=[keywords],
request_formatters=return_exception_raising_formatter)

Expand All @@ -275,8 +275,7 @@ def test_munger_class_method_access_raises_friendly_error():


def test_munger_arguments_by_keyword(dummy_w3):
addr = '0x' + '00' * 20
with pytest.raises(Success):
dummy_w3.fake.method(addr, keyword_two='latest')
dummy_w3.fake.method(keyword_one=1, keyword_two='latest')
with pytest.raises(Success):
dummy_w3.fake.method(addr, 'latest')
dummy_w3.fake.method(1, keyword_two=2)
2 changes: 0 additions & 2 deletions web3/method.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,8 +91,6 @@ def __init__(

self.json_rpc_method = json_rpc_method
self.mungers = mungers or [default_munger]
# TODO - decide if this request_formatters
# (and result_formatters) is worth keeping for testing
self.request_formatters = request_formatters or get_request_formatters
self.result_formatters = result_formatters or get_result_formatters
self.error_formatters = get_error_formatters
Expand Down

0 comments on commit 4d2a0d4

Please sign in to comment.