from typing import cast from pathlib import Path from collections.abc import Callable from nonebug import App from respx import MockRouter from pytest_mock import MockerFixture from nonebot.adapters.onebot.v11 import Bot from nonebot.adapters.onebot.v11.message import Message from tests.config import BotId, UserId, GroupId, MessageId from tests.utils import ( get_content_bytes, get_response_json, _v11_group_message_event, _v11_private_message_send, ) def init_mocked_api(mocked_api: MockRouter) -> None: mocked_api.get( "https://api.github.com/repos/HibiKier/zhenxun_bot/releases/latest", name="release_latest", ).respond(200, json=get_response_json("release_latest.json")) mocked_api.get( "https://raw.githubusercontent.com/HibiKier/zhenxun_bot/dev/__version__", name="dev_branch_version", ).respond(200, text="__version__: v0.2.2") mocked_api.get( "https://raw.githubusercontent.com/HibiKier/zhenxun_bot/main/__version__", name="main_branch_version", ).respond(200, text="__version__: v0.2.2") mocked_api.get( "https://api.github.com/repos/HibiKier/zhenxun_bot/tarball/v0.2.2", name="release_download_url", ).respond( 302, headers={ "Location": "https://codeload.github.com/HibiKier/zhenxun_bot/legacy.tar.gz/refs/tags/v0.2.2" }, ) mocked_api.get( "https://codeload.github.com/HibiKier/zhenxun_bot/legacy.tar.gz/refs/tags/v0.2.2", name="release_download_url_redirect", ).respond( 200, content=get_content_bytes("download_latest_file.tar.gz"), ) async def test_check_update_release( app: App, mocker: MockerFixture, mocked_api: MockRouter, create_bot: Callable, tmp_path: Path, ) -> None: """ 测试检查更新 """ from zhenxun.builtin_plugins.auto_update import _matcher init_mocked_api(mocked_api) mocker.patch( "zhenxun.builtin_plugins.auto_update._data_source.REPLACE_FOLDERS", return_value=[], ) mocker.patch( "zhenxun.builtin_plugins.auto_update._data_source.install_requirement", return_value=None, ) async with app.test_matcher(_matcher) as ctx: bot = create_bot(ctx) bot = cast(Bot, bot) raw_message = "检查更新 release" event = _v11_group_message_event( raw_message, self_id=BotId.QQ_BOT, user_id=UserId.SUPERUSER, group_id=GroupId.GROUP_ID_LEVEL_5, message_id=MessageId.MESSAGE_ID, to_me=True, ) ctx.receive_event(bot, event) ctx.should_call_api( "send_msg", _v11_private_message_send( message="检测真寻已更新,版本更新:v0.2.2 -> v0.2.2\n开始更新...", user_id=UserId.SUPERUSER, ), ) ctx.should_call_send( event=event, message=Message( "版本更新完成\n" "版本: v0.2.2 -> v0.2.2\n" "请重新启动真寻以完成更新!" ), result=None, bot=bot, ) ctx.should_finished(_matcher) assert mocked_api["release_latest"].called assert mocked_api["release_download_url"].called assert mocked_api["release_download_url_redirect"].called