From ebd2e694a09129e578cb02e1ac5cf0ac04f71661 Mon Sep 17 00:00:00 2001 From: Masen Furer Date: Mon, 7 Oct 2024 10:36:17 -0700 Subject: [PATCH] test_state: close StateManagerRedis connection in test to avoid warning --- tests/units/test_state.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/units/test_state.py b/tests/units/test_state.py index 45d1075614..2bb6902476 100644 --- a/tests/units/test_state.py +++ b/tests/units/test_state.py @@ -2793,6 +2793,9 @@ async def test_preprocess(app_module_mock, token, test_state, expected, mocker): } assert (await state._process(events[1]).__anext__()).delta == exp_is_hydrated(state) + if isinstance(app.state_manager, StateManagerRedis): + await app.state_manager.close() + @pytest.mark.asyncio async def test_preprocess_multiple_load_events(app_module_mock, token, mocker): @@ -2840,6 +2843,9 @@ async def test_preprocess_multiple_load_events(app_module_mock, token, mocker): } assert (await state._process(events[2]).__anext__()).delta == exp_is_hydrated(state) + if isinstance(app.state_manager, StateManagerRedis): + await app.state_manager.close() + @pytest.mark.asyncio async def test_get_state(mock_app: rx.App, token: str):