diff --git a/tests/builtin_plugins/auto_update/test_check_update.py b/tests/builtin_plugins/auto_update/test_check_update.py index 0555806c..8a505401 100644 --- a/tests/builtin_plugins/auto_update/test_check_update.py +++ b/tests/builtin_plugins/auto_update/test_check_update.py @@ -13,7 +13,11 @@ from pytest_mock import MockerFixture from respx import MockRouter from tests.config import BotId, GroupId, MessageId, UserId -from tests.utils import _v11_group_message_event, _v11_private_message_send +from tests.utils import ( + _v11_group_message_event, + _v11_private_message_send, + get_reply_cq, +) from tests.utils import get_response_json as _get_response_json @@ -311,6 +315,12 @@ async def test_check_update_release( to_me=True, ) ctx.receive_event(bot, event) + ctx.should_call_send( + event=event, + message=Message(f"{get_reply_cq(MessageId.MESSAGE_ID)}正在进行检查更新..."), + result=None, + bot=bot, + ) ctx.should_call_api( "send_msg", _v11_private_message_send( @@ -401,6 +411,12 @@ async def test_check_update_main( to_me=True, ) ctx.receive_event(bot, event) + ctx.should_call_send( + event=event, + message=Message(f"{get_reply_cq(MessageId.MESSAGE_ID)}正在进行检查更新..."), + result=None, + bot=bot, + ) ctx.should_call_api( "send_msg", _v11_private_message_send( diff --git a/tests/utils.py b/tests/utils.py index f05aa8e9..9ec6af24 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -5,6 +5,10 @@ from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message, MessageSegme from nonebot.adapters.onebot.v11.event import Sender +def get_reply_cq(uid: int | str) -> str: + return f"[CQ:reply,id={uid}]" + + def get_response_json(base_path: Path, file: str) -> dict: try: return json.loads( diff --git a/zhenxun/builtin_plugins/auto_update/__init__.py b/zhenxun/builtin_plugins/auto_update/__init__.py index 4576ff0b..764fc39c 100644 --- a/zhenxun/builtin_plugins/auto_update/__init__.py +++ b/zhenxun/builtin_plugins/auto_update/__init__.py @@ -11,7 +11,6 @@ from nonebot_plugin_alconna import ( on_alconna, store_true, ) -from nonebot_plugin_session import EventSession from nonebot_plugin_uninfo import Uninfo from zhenxun.configs.utils import PluginExtraData