From 029cd091f33c3a18813cb19328e233b8aa6cb380 Mon Sep 17 00:00:00 2001 From: pablonyx Date: Mon, 31 Mar 2025 10:34:11 -0700 Subject: [PATCH] flush redis in integration tests --- backend/tests/integration/common_utils/reset.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/backend/tests/integration/common_utils/reset.py b/backend/tests/integration/common_utils/reset.py index 366f50814bf..63cd8906632 100644 --- a/backend/tests/integration/common_utils/reset.py +++ b/backend/tests/integration/common_utils/reset.py @@ -22,6 +22,7 @@ from onyx.document_index.document_index_utils import get_multipass_config from onyx.document_index.vespa.index import DOCUMENT_ID_ENDPOINT from onyx.document_index.vespa.index import VespaIndex from onyx.indexing.models import IndexingSetting +from onyx.redis.redis_pool import get_redis_client from onyx.setup import setup_postgres from onyx.setup import setup_vespa from onyx.utils.logger import setup_logger @@ -237,6 +238,12 @@ def reset_vespa() -> None: time.sleep(5) +def reset_redis() -> None: + """Reset the Redis database.""" + redis_client = get_redis_client() + redis_client.flushall() + + def reset_postgres_multitenant() -> None: """Reset the Postgres database for all tenants in a multitenant setup.""" @@ -341,6 +348,8 @@ def reset_all() -> None: reset_postgres() logger.info("Resetting Vespa...") reset_vespa() + logger.info("Resetting Redis...") + reset_redis() def reset_all_multitenant() -> None: