✨ Feature: 迁移至结构化并发框架 AnyIO (#3053)

tests/conftest.py

@@ -1,8 +1,10 @@
 import os
 import threading
 from pathlib import Path
-from typing import TYPE_CHECKING
+from functools import wraps
 from collections.abc import Generator
+from typing_extensions import ParamSpec
+from typing import TYPE_CHECKING, TypeVar, Callable
 
 import pytest
 from nonebug import NONEBOT_INIT_KWARGS
@@ -20,6 +22,9 @@ os.environ["CONFIG_OVERRIDE"] = "new"
 if TYPE_CHECKING:
     from nonebot.plugin import Plugin
 
+P = ParamSpec("P")
+R = TypeVar("R")
+
 collect_ignore = ["plugins/", "dynamic/", "bad_plugins/"]
 
 
@@ -38,14 +43,36 @@ def load_driver(request: pytest.FixtureRequest) -> Driver:
     return DriverClass(Env(environment=global_driver.env), global_driver.config)
 
 
+@pytest.fixture(scope="session", params=[pytest.param("asyncio"), pytest.param("trio")])
+def anyio_backend(request: pytest.FixtureRequest):
+    return request.param
+
+
+def run_once(func: Callable[P, R]) -> Callable[P, R]:
+    result = ...
+
+    @wraps(func)
+    def _wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
+        nonlocal result
+        if result is not Ellipsis:
+            return result
+
+        result = func(*args, **kwargs)
+        return result
+
+    return _wrapper
+
+
 @pytest.fixture(scope="session", autouse=True)
-def load_plugin(nonebug_init: None) -> set["Plugin"]:
+@run_once
+def load_plugin(anyio_backend, nonebug_init: None) -> set["Plugin"]:
     # preload global plugins
     return nonebot.load_plugins(str(Path(__file__).parent / "plugins"))
 
 
 @pytest.fixture(scope="session", autouse=True)
-def load_builtin_plugin(nonebug_init: None) -> set["Plugin"]:
+@run_once
+def load_builtin_plugin(anyio_backend, nonebug_init: None) -> set["Plugin"]:
     # preload builtin plugins
     return nonebot.load_builtin_plugins("echo", "single_session")