From 2d8b138e86f49d7100aabd33b750e0636704276a Mon Sep 17 00:00:00 2001 From: Azide Date: Tue, 12 Sep 2023 16:26:32 +0800 Subject: [PATCH] =?UTF-8?q?:bug:=20=E6=9B=B4=E6=94=B9=E6=A0=87=E6=B3=A8?= =?UTF-8?q?=E7=B1=BB=E5=9E=8B=E7=9A=84=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nonebot_bison/admin_page/__init__.py | 2 +- tests/test_admin_page.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) 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()