From 57343ced2289fbf605b3f077b6d95fc175245722 Mon Sep 17 00:00:00 2001 From: webjoin111 <455457521@qq.com> Date: Wed, 13 Aug 2025 06:40:19 +0800 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20test(check):=20=E6=9B=B4=E6=96=B0?= =?UTF-8?q?=E8=87=AA=E6=A3=80=E6=8F=92=E4=BB=B6=E6=B5=8B=E8=AF=95=E4=B8=AD?= =?UTF-8?q?=E7=9A=84=E6=B8=B2=E6=9F=93=E6=9C=8D=E5=8A=A1=E6=A8=A1=E6=8B=9F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/builtin_plugins/check/test_check.py | 35 ++++++++++------------- 1 file changed, 15 insertions(+), 20 deletions(-) diff --git a/tests/builtin_plugins/check/test_check.py b/tests/builtin_plugins/check/test_check.py index c16c2aad..35baa787 100644 --- a/tests/builtin_plugins/check/test_check.py +++ b/tests/builtin_plugins/check/test_check.py @@ -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.uname.return_value = platform_uname - mock_template_to_pic = mocker.patch("zhenxun.builtin_plugins.check.template_to_pic") - mock_template_to_pic_return = mocker.AsyncMock() - mock_template_to_pic.return_value = mock_template_to_pic_return + mock_render_service = mocker.patch( + "zhenxun.builtin_plugins.check.renderer_service.render" + ) + mock_render_service_return = mocker.AsyncMock() + mock_render_service.return_value = mock_render_service_return mock_build_message = mocker.patch( "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_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 ( mock_psutil, mock_cpuinfo, mock_platform, - mock_template_to_pic, - mock_template_to_pic_return, + mock_render_service, + mock_render_service_return, mock_build_message, mock_build_message_return, - mock_template_path_new, ) @@ -105,11 +102,10 @@ async def test_check( mock_psutil, mock_cpuinfo, mock_platform, - mock_template_to_pic, - mock_template_to_pic_return, + mock_render_service, + mock_render_service_return, mock_build_message, mock_build_message_return, - mock_template_path_new, ) = init_mocker(mocker, tmp_path) async with app.test_matcher(_self_check_matcher) as ctx: bot = create_bot(ctx) @@ -126,8 +122,8 @@ async def test_check( ctx.receive_event(bot=bot, event=event) ctx.should_ignore_rule(_self_check_matcher) - mock_template_to_pic.assert_awaited_once() - mock_build_message.assert_called_once_with(mock_template_to_pic_return) + mock_render_service.assert_awaited_once() + mock_build_message.assert_called_once_with(mock_render_service_return) mock_build_message_return.send.assert_awaited_once() @@ -161,11 +157,10 @@ async def test_check_arm( mock_psutil, mock_cpuinfo, mock_platform, - mock_template_to_pic, - mock_template_to_pic_return, + mock_render_service, + mock_render_service_return, mock_build_message, mock_build_message_return, - mock_template_path_new, ) = init_mocker(mocker, tmp_path) mock_platform.uname.return_value = platform_uname_arm @@ -199,6 +194,6 @@ async def test_check_arm( mocker.call().decode().split().__getitem__().__float__(), ] # type: ignore ) - mock_template_to_pic.assert_awaited_once() - mock_build_message.assert_called_once_with(mock_template_to_pic_return) + mock_render_service.assert_awaited_once() + mock_build_message.assert_called_once_with(mock_render_service_return) mock_build_message_return.send.assert_awaited_once()