test_state: close StateManagerRedis connection in test to avoid warning

reflex-dev · Oct 7, 2024 · ebd2e69 · ebd2e69
1 parent 4919191
commit ebd2e69
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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):