Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

✅ test(check): 更新自检插件测试中的渲染服务模拟

Browse Source
This commit is contained in:
webjoin111 2025-08-13 06:40:19 +08:00
parent 689505294c
commit 57343ced22
1 changed files with 15 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/builtin_plugins/check/test_check.py
View File

@ -64,9 +64,11 @@ def init_mocker(mocker: MockerFixture, tmp_path: Path):
mock_platform = mocker.patch("zhenxun.builtin_plugins.check.data_source.platform") mock_platform = mocker.patch("zhenxun.builtin_plugins.check.data_source.platform")
mock_platform.uname.return_value = platform_uname mock_platform.uname.return_value = platform_uname
mock_template_to_pic = mocker.patch("zhenxun.builtin_plugins.check.template_to_pic") mock_render_service = mocker.patch(
mock_template_to_pic_return = mocker.AsyncMock() "zhenxun.builtin_plugins.check.renderer_service.render"
mock_template_to_pic.return_value = mock_template_to_pic_return )
mock_render_service_return = mocker.AsyncMock()
mock_render_service.return_value = mock_render_service_return
mock_build_message = mocker.patch( mock_build_message = mocker.patch(
"zhenxun.builtin_plugins.check.MessageUtils.build_message" "zhenxun.builtin_plugins.check.MessageUtils.build_message"
@ -74,19 +76,14 @@ def init_mocker(mocker: MockerFixture, tmp_path: Path):
mock_build_message_return = mocker.AsyncMock() mock_build_message_return = mocker.AsyncMock()
mock_build_message.return_value = mock_build_message_return mock_build_message.return_value = mock_build_message_return
mock_template_path_new = tmp_path / "resources" / "template"
mocker.patch(
"zhenxun.builtin_plugins.check.TEMPLATE_PATH", new=mock_template_path_new
)
return ( return (
mock_psutil, mock_psutil,
mock_cpuinfo, mock_cpuinfo,
mock_platform, mock_platform,
mock_template_to_pic, mock_render_service,
mock_template_to_pic_return, mock_render_service_return,
mock_build_message, mock_build_message,
mock_build_message_return, mock_build_message_return,
mock_template_path_new,
) )
@ -105,11 +102,10 @@ async def test_check(
mock_psutil, mock_psutil,
mock_cpuinfo, mock_cpuinfo,
mock_platform, mock_platform,
mock_template_to_pic, mock_render_service,
mock_template_to_pic_return, mock_render_service_return,
mock_build_message, mock_build_message,
mock_build_message_return, mock_build_message_return,
mock_template_path_new,
) = init_mocker(mocker, tmp_path) ) = init_mocker(mocker, tmp_path)
async with app.test_matcher(_self_check_matcher) as ctx: async with app.test_matcher(_self_check_matcher) as ctx:
bot = create_bot(ctx) bot = create_bot(ctx)
@ -126,8 +122,8 @@ async def test_check(
ctx.receive_event(bot=bot, event=event) ctx.receive_event(bot=bot, event=event)
ctx.should_ignore_rule(_self_check_matcher) ctx.should_ignore_rule(_self_check_matcher)
mock_template_to_pic.assert_awaited_once() mock_render_service.assert_awaited_once()
mock_build_message.assert_called_once_with(mock_template_to_pic_return) mock_build_message.assert_called_once_with(mock_render_service_return)
mock_build_message_return.send.assert_awaited_once() mock_build_message_return.send.assert_awaited_once()
@ -161,11 +157,10 @@ async def test_check_arm(
mock_psutil, mock_psutil,
mock_cpuinfo, mock_cpuinfo,
mock_platform, mock_platform,
mock_template_to_pic, mock_render_service,
mock_template_to_pic_return, mock_render_service_return,
mock_build_message, mock_build_message,
mock_build_message_return, mock_build_message_return,
mock_template_path_new,
) = init_mocker(mocker, tmp_path) ) = init_mocker(mocker, tmp_path)
mock_platform.uname.return_value = platform_uname_arm mock_platform.uname.return_value = platform_uname_arm
@ -199,6 +194,6 @@ async def test_check_arm(
mocker.call().decode().split().__getitem__().__float__(), mocker.call().decode().split().__getitem__().__float__(),
] # type: ignore ] # type: ignore
) )
mock_template_to_pic.assert_awaited_once() mock_render_service.assert_awaited_once()
mock_build_message.assert_called_once_with(mock_template_to_pic_return) mock_build_message.assert_called_once_with(mock_render_service_return)
mock_build_message_return.send.assert_awaited_once() mock_build_message_return.send.assert_awaited_once()