mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
|
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
|