tmtt/redis-om-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add redis "barrier" in cleanup_keys

Browse source 
To prevent pytest-xdist workers stomping each other with _delete_test_keys
This commit is contained in:
Serhii Charykov 2022-04-27 19:41:07 +03:00
parent 878490abb7
commit 675722b221
1 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/conftest.py
View file

@ -37,11 +37,18 @@ def key_prefix(request, redis):
@pytest.fixture(scope="session", autouse=True)
def cleanup_keys(request):
def cleanup_keys():
# Always use the sync Redis connection with finalizer. Setting up an
# async finalizer should work, but I'm not suer how yet!
from redis_om.connections import get_redis_connection as get_sync_redis
# Always use the sync Redis connection with finalizer. Setting up an
# async finalizer should work, but I'm not suer how yet!
from redis_om.connections import get_redis_connection as get_sync_redis
_delete_test_keys(TEST_PREFIX, get_sync_redis())
# Increment for every pytest-xdist worker
redis = get_sync_redis()
once_key = f"{TEST_PREFIX}:cleanup_keys"
redis.incr(once_key)
request.addfinalizer(cleanup_keys)
yield
# Delete keys only once
if redis.decr(once_key) == 0:
_delete_test_keys(TEST_PREFIX, redis)
redis.delete(once_key)