diff options
Diffstat (limited to 'tests/conftest.py')
-rw-r--r-- | tests/conftest.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/conftest.py b/tests/conftest.py index 87e6301..0007b84 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -74,13 +74,15 @@ def _get_client(cls, request, single_connection_client=True, **kwargs): @pytest.fixture() def r(request): - return _get_client(redis.Redis, request) + with _get_client(redis.Redis, request) as client: + yield client @pytest.fixture() def r2(request): "A second client for tests that need multiple" - return _get_client(redis.Redis, request) + with _get_client(redis.Redis, request) as client: + yield client def _gen_cluster_mock_resp(r, response): |