diff --git a/nonebot_bison/admin_page/__init__.py b/nonebot_bison/admin_page/__init__.py index 1064eba..17182fe 100644 --- a/nonebot_bison/admin_page/__init__.py +++ b/nonebot_bison/admin_page/__init__.py @@ -72,7 +72,7 @@ def register_get_token_handler(): get_token.__help__info__ = "获取管理bot后台的地址,该地址会在一段时间过后过期,请不要泄漏该地址" # type: ignore -def check_driver_is_fastapi() -> "Driver" | None: +def check_driver_is_fastapi() -> "Driver | None": try: from nonebot.drivers.fastapi import Driver diff --git a/tests/test_admin_page.py b/tests/test_admin_page.py index b4d7b8f..150861c 100644 --- a/tests/test_admin_page.py +++ b/tests/test_admin_page.py @@ -42,15 +42,20 @@ async def test_log(app: App, tmp_path: Path): import io import contextlib + from nonebot import get_driver + from nonebot.drivers.fastapi import Driver from nonebot.log import logger, default_format from nonebot_bison.admin_page import init_fastapi + driver = get_driver() + assert isinstance(driver, Driver) + log_path = tmp_path / "temp.log" logger.add(log_path, level="INFO", format=default_format, rotation="1 day") with contextlib.redirect_stderr(io.StringIO()) as f: - init_fastapi() + init_fastapi(driver) with log_path.open("r", encoding="utf-8") as f: log = f.read()