def test_use_connection_explicit_redis(redis, loop):
"""Pass redis connection explicitly."""
connection = yield from find_connection(loop)
yield from use_connection(redis=connection)
assert get_current_connection() == connection
push_connection(redis) # Make test finalizer happy.
def test_use_connection_cleanup_stack(redis, loop):
"""Ensure connection stack cleanup."""
kwargs = dict(address=('localhost', 6379), loop=loop)
yield from use_connection(**kwargs)
assert len(_connection_stack) == 1
push_connection(redis) # Make test finalizer happy.
def test_use_connection(redis, loop):
"""Replace connection stack."""
kwargs = dict(address=('localhost', 6379), loop=loop)
yield from use_connection(**kwargs)
assert get_current_connection() != redis
push_connection(redis) # Make test finalizer happy.
def test_connection_stacking_with_use_connection(redis, loop):
"""Disallow use of use_connection() together with stacked contexts."""
kwargs = dict(address=('localhost', 6379), loop=loop)
with pytest.raises(AssertionError):
with (yield from Connection(**kwargs)):
with (yield from Connection(**kwargs)):
yield from use_connection(**kwargs)
