#from collections.abc import Callable #from pathlib import Path #import platform #from typing import cast #import nonebot #from nonebot.adapters.onebot.v11 import Bot #from nonebot.adapters.onebot.v11.event import GroupMessageEvent #from nonebug import App #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 #platform_uname = platform.uname_result( # system="Linux", # node="zhenxun", # release="5.15.0-1027-azure", # version="#1 SMP Debian 5.15.0-1027-azure", # machine="x86_64", #) # type: ignore #cpuinfo_get_cpu_info = {"brand_raw": "Intel(R) Core(TM) i7-10700K"} #def init_mocker(mocker: MockerFixture, tmp_path: Path): # mock_psutil = mocker.patch("zhenxun.builtin_plugins.check.data_source.psutil") # mock_cpuinfo = mocker.patch("zhenxun.builtin_plugins.check.data_source.cpuinfo") # mock_cpuinfo.get_cpu_info.return_value = cpuinfo_get_cpu_info # 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_build_message = mocker.patch( # "zhenxun.builtin_plugins.check.MessageUtils.build_message" # ) # 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_build_message, # mock_build_message_return, # mock_template_path_new, # ) #async def test_check( # app: App, # mocker: MockerFixture, # mocked_api: MockRouter, # create_bot: Callable, # tmp_path: Path, #) -> None: # """ # 测试自检 # """ # from zhenxun.builtin_plugins.check import _matcher # from zhenxun.builtin_plugins.check.data_source import __get_version # from zhenxun.configs.config import BotConfig # ( # mock_psutil, # mock_cpuinfo, # mock_platform, # mock_template_to_pic, # mock_template_to_pic_return, # mock_build_message, # mock_build_message_return, # mock_template_path_new, # ) = init_mocker(mocker, tmp_path) # async with app.test_matcher(_matcher) as ctx: # bot = create_bot(ctx) # bot: Bot = cast(Bot, bot) # raw_message = "自检" # event: GroupMessageEvent = _v11_group_message_event( # message=raw_message, # self_id=BotId.QQ_BOT, # user_id=UserId.SUPERUSER, # group_id=GroupId.GROUP_ID_LEVEL_5, # message_id=MessageId.MESSAGE_ID_3, # to_me=True, # ) # ctx.receive_event(bot=bot, event=event) # mock_template_to_pic.assert_awaited_once_with( # template_path=str((mock_template_path_new / "check").absolute()), # template_name="main.html", # templates={ # "data": { # "cpu_info": "1.0% - 1.0Ghz [1 core]", # "cpu_process": 1.0, # "ram_info": "1.0 / 1.0 GB", # "ram_process": 100.0, # "swap_info": "1.0 / 1.0 GB", # "swap_process": 100.0, # "disk_info": "1.0 / 1.0 GB", # "disk_process": 100.0, # "brand_raw": cpuinfo_get_cpu_info["brand_raw"], # "baidu": "red", # "google": "red", # "system": f"{platform_uname.system} " f"{platform_uname.release}", # "version": __get_version(), # "plugin_count": len(nonebot.get_loaded_plugins()), # "nickname": BotConfig.self_nickname, # } # }, # pages={ # "viewport": {"width": 195, "height": 750}, # "base_url": f"file://{mock_template_path_new.absolute()}", # }, # wait=2, # ) # mock_template_to_pic.assert_awaited_once() # mock_build_message.assert_called_once_with(mock_template_to_pic_return) # mock_build_message_return.send.assert_awaited_once() #async def test_check_arm( # app: App, # mocker: MockerFixture, # mocked_api: MockRouter, # create_bot: Callable, # tmp_path: Path, #) -> None: # """ # 测试自检（arm） # """ # from zhenxun.builtin_plugins.check import _matcher # from zhenxun.builtin_plugins.check.data_source import __get_version # from zhenxun.configs.config import BotConfig # platform_uname_arm = platform.uname_result( # system="Linux", # node="zhenxun", # release="5.15.0-1017-oracle", # version="#22~20.04.1-Ubuntu SMP Wed Aug 24 11:13:15 UTC 2022", # machine="aarch64", # ) # type: ignore # mock_subprocess_check_output = mocker.patch( # "zhenxun.builtin_plugins.check.data_source.subprocess.check_output" # ) # mock_environ_copy = mocker.patch( # "zhenxun.builtin_plugins.check.data_source.os.environ.copy" # ) # mock_environ_copy_return = mocker.MagicMock() # mock_environ_copy.return_value = mock_environ_copy_return # ( # mock_psutil, # mock_cpuinfo, # mock_platform, # mock_template_to_pic, # mock_template_to_pic_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 # mock_cpuinfo.get_cpu_info.return_value = {} # mock_psutil.cpu_freq.return_value = {} # async with app.test_matcher(_matcher) as ctx: # bot = create_bot(ctx) # bot: Bot = cast(Bot, bot) # raw_message = "自检" # event: GroupMessageEvent = _v11_group_message_event( # message=raw_message, # self_id=BotId.QQ_BOT, # user_id=UserId.SUPERUSER, # group_id=GroupId.GROUP_ID_LEVEL_5, # message_id=MessageId.MESSAGE_ID_3, # to_me=True, # ) # ctx.receive_event(bot=bot, event=event) # mock_template_to_pic.assert_awaited_once_with( # template_path=str((mock_template_path_new / "check").absolute()), # template_name="main.html", # templates={ # "data": { # "cpu_info": "1.0% - 0.0Ghz [1 core]", # "cpu_process": 1.0, # "ram_info": "1.0 / 1.0 GB", # "ram_process": 100.0, # "swap_info": "1.0 / 1.0 GB", # "swap_process": 100.0, # "disk_info": "1.0 / 1.0 GB", # "disk_process": 100.0, # "brand_raw": "", # "baidu": "red", # "google": "red", # "system": f"{platform_uname_arm.system} " # f"{platform_uname_arm.release}", # "version": __get_version(), # "plugin_count": len(nonebot.get_loaded_plugins()), # "nickname": BotConfig.self_nickname, # } # }, # pages={ # "viewport": {"width": 195, "height": 750}, # "base_url": f"file://{mock_template_path_new.absolute()}", # }, # wait=2, # ) # mock_subprocess_check_output.assert_has_calls( # [ # mocker.call(["lscpu"], env=mock_environ_copy_return), # mocker.call().decode(), # mocker.call().decode().splitlines(), # mocker.call().decode().splitlines().__iter__(), # mocker.call(["dmidecode", "-s", "processor-frequency"]), # mocker.call().decode(), # mocker.call().decode().split(), # mocker.call().decode().split().__getitem__(0), # 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_build_message_return.send.assert_awaited_once()