test: improve resource cleanup in async tests to prevent 'Event loop is closed' error

- Add `close_all_async_resources()` function to properly close all async resources
- Update `asyncTearDown()` methods in test classes to use the new cleanup function
- Add `tearDownModule()` to ensure async resources are closed after all tests run
- Update Python versions in `tests/run.sh` to include 3.8

Author: tanbro

tests/_catches.py

@@ -99,14 +99,30 @@ async def close_async_redis_client():
 
 
 ASYNC_MULTI_CACHES = {
-    "tlru": RedisFuncCache(__name__, LruTMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
-    "lru": RedisFuncCache(__name__, LruMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
-    "mru": RedisFuncCache(__name__, MruMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
-    "rr": RedisFuncCache(__name__, RrMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
-    "fifo": RedisFuncCache(__name__, FifoMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
-    "lfu": RedisFuncCache(__name__, LfuMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
+    "tlru": RedisFuncCache(__name__, LruTClusterMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
+    "lru": RedisFuncCache(__name__, LruClusterMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
+    "mru": RedisFuncCache(__name__, MruClusterMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
+    "rr": RedisFuncCache(__name__, RrClusterMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
+    "fifo": RedisFuncCache(__name__, FifoClusterMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
+    "lfu": RedisFuncCache(__name__, LfuClusterMultiplePolicy, client=ASYNC_REDIS_FACTORY, maxsize=MAXSIZE),
 }
 
+
+async def close_all_async_resources():
+    """关闭所有异步资源，防止出现 'Event loop is closed' 错误"""
+    # 关闭全局异步Redis客户端
+    await close_async_redis_client()
+
+    # 关闭所有异步缓存实例中的客户端连接
+    for cache in ASYNC_CACHES.values():
+        if hasattr(cache.client, "close"):
+            await cache.client.close()
+
+    for cache in ASYNC_MULTI_CACHES.values():
+        if hasattr(cache.client, "close"):
+            await cache.client.close()
+
+
 CLUSTER_NODES: List[ClusterNode] = []
 CLUSTER_CACHES: Dict[str, RedisFuncCache] = {}
 if REDIS_CLUSTER_NODES:

tests/run.sh

@@ -6,7 +6,7 @@ export PIP_ROOT_USER_ACTION=ignore
 export PIP_NO_WARN_SCRIPT_LOCATION=1
 
 
-PYTHON_LIST=(python3.9 python3.10 python3.11 python3.12 python3.13)
+PYTHON_LIST=(python3.8 python3.9 python3.10 python3.11 python3.12 python3.13)
 for PYTHON in ${PYTHON_LIST[@]}
 do
     echo

tests/test_async.py

@@ -3,7 +3,7 @@
 from unittest import IsolatedAsyncioTestCase
 from unittest.mock import AsyncMock, patch
 
-from ._catches import ASYNC_CACHES, ASYNC_MULTI_CACHES, CACHES, close_async_redis_client
+from ._catches import ASYNC_CACHES, ASYNC_MULTI_CACHES, CACHES, close_all_async_resources
 
 
 def _echo(x):
@@ -17,7 +17,7 @@ async def asyncSetUp(self):
 
     async def asyncTearDown(self):
         # 清理异步Redis客户端连接
-        await close_async_redis_client()
+        await asyncio.gather(*(cache.client.close() for cache in ASYNC_CACHES.values()))
 
     async def test_simple(self):
         for cache in ASYNC_CACHES.values():
@@ -42,7 +42,7 @@ async def asyncSetUp(self):
 
     async def asyncTearDown(self):
         # 清理异步Redis客户端连接
-        await close_async_redis_client()
+        await asyncio.gather(*(cache.client.close() for cache in ASYNC_MULTI_CACHES.values()))
 
     async def test_simple(self):
         for cache in ASYNC_MULTI_CACHES.values():
@@ -106,3 +106,20 @@ async def test_type_error_sync_for_async(self):
                 async def _(x):
                     await asyncio.sleep(0)
                     return x
+
+
+# 模块级别的清理函数，在所有测试运行完毕后调用
+def tearDownModule():
+    """在模块中的所有测试运行完毕后清理异步资源"""
+    try:
+        # 获取当前事件循环
+        loop = asyncio.get_event_loop()
+        # 如果事件循环还在运行，则执行清理
+        if loop.is_running():
+            loop.create_task(close_all_async_resources())
+        else:
+            # 否则直接运行直到完成
+            loop.run_until_complete(close_all_async_resources())
+    except RuntimeError:
+        # 如果没有事件循环或者事件循环已关闭，忽略错误
+        pass

tests/test_async_context.py

@@ -3,7 +3,7 @@
 from unittest.mock import AsyncMock, patch
 from uuid import uuid4
 
-from ._catches import ASYNC_CACHES
+from ._catches import ASYNC_CACHES, close_all_async_resources
 
 
 class AsyncContextTTLTest(IsolatedAsyncioTestCase):
@@ -47,3 +47,20 @@ async def echo(x):
                     mock_get.assert_called_once()
                     mock_put.assert_not_called()
                     self.assertEqual(result, val)
+
+
+# 模块级别的清理函数，在所有测试运行完毕后调用
+def tearDownModule():
+    """在模块中的所有测试运行完毕后清理异步资源"""
+    try:
+        # 获取当前事件循环
+        loop = asyncio.get_event_loop()
+        # 如果事件循环还在运行，则执行清理
+        if loop.is_running():
+            loop.create_task(close_all_async_resources())
+        else:
+            # 否则直接运行直到完成
+            loop.run_until_complete(close_all_async_resources())
+    except RuntimeError:
+        # 如果没有事件循环或者事件循环已关闭，忽略错误
+        pass