def test_simple_cache_middleware_populates_cache(w3):
w3.middleware_stack.add(construct_simple_cache_middleware(
cache_class=dict,
rpc_whitelist={'fake_endpoint'},
))
result = w3.manager.request_blocking('fake_endpoint', [])
assert w3.manager.request_blocking('fake_endpoint', []) == result
assert w3.manager.request_blocking('fake_endpoint', [1]) != result
def test_simple_cache_middleware_does_not_cache_endpoints_not_in_whitelist(w3):
w3.middleware_stack.add(construct_simple_cache_middleware(
cache_class=dict,
rpc_whitelist={'fake_endpoint'},
))
result_a = w3.manager.request_blocking('not_whitelisted', [])
result_b = w3.manager.request_blocking('not_whitelisted', [])
assert result_a != result_b
def test_simple_cache_middleware_pulls_from_cache(w3):
def cache_class():
return {
generate_cache_key(('fake_endpoint', [1])): {'result': 'value-a'},
}
w3.middleware_stack.add(construct_simple_cache_middleware(
cache_class=cache_class,
rpc_whitelist={'fake_endpoint'},
))
assert w3.manager.request_blocking('fake_endpoint', [1]) == 'value-a'
def test_simple_cache_middleware_does_not_cache_error_responses(w3_base):
w3 = w3_base
w3.middleware_stack.add(construct_error_generator_middleware({
'fake_endpoint': lambda *_: 'msg-{0}'.format(str(uuid.uuid4())),
}))
w3.middleware_stack.add(construct_simple_cache_middleware(
cache_class=dict,
rpc_whitelist={'fake_endpoint'},
))
with pytest.raises(ValueError) as err_a:
w3.manager.request_blocking('fake_endpoint', [])
with pytest.raises(ValueError) as err_b:
w3.manager.request_blocking('fake_endpoint', [])
assert str(err_a) != str(err_b)
def test_simple_cache_middleware_does_not_cache_none_responses(w3_base):
counter = itertools.count()
w3 = w3_base
def result_cb(method, params):
next(counter)
return None
w3.middleware_stack.add(construct_result_generator_middleware({
'fake_endpoint': result_cb,
}))
w3.middleware_stack.add(construct_simple_cache_middleware(
cache_class=dict,
rpc_whitelist={'fake_endpoint'},
))
w3.manager.request_blocking('fake_endpoint', [])
w3.manager.request_blocking('fake_endpoint', [])
assert next(counter) == 2