mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-08 04:43:00 +08:00
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
import base64
|
||
import hashlib
|
||
import platform
|
||
from io import UnsupportedOperation
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
import respx
|
||
from httpx import Response
|
||
from nonebot.adapters.onebot.v11.message import MessageSegment
|
||
from nonebug.app import App
|
||
|
||
|
||
@pytest.fixture
|
||
def ms_list():
|
||
msg_segments: list[MessageSegment] = []
|
||
msg_segments.append(MessageSegment.text("【Zc】每早合约日替攻略！"))
|
||
msg_segments.append(
|
||
MessageSegment.image(
|
||
file="http://i0.hdslb.com/bfs/live/new_room_cover/cf7d4d3b2f336c6dba299644c3af952c5db82612.jpg",
|
||
cache=0,
|
||
)
|
||
)
|
||
msg_segments.append(MessageSegment.text("来源: Bilibili直播 魔法Zc目录"))
|
||
msg_segments.append(MessageSegment.text("详情: https://live.bilibili.com/3044248"))
|
||
|
||
return msg_segments
|
||
|
||
|
||
@pytest.fixture
|
||
def pic_hash():
|
||
platform_name = platform.system()
|
||
if platform_name == "Windows":
|
||
return "58723fdc24b473b6dbd8ec8cbc3b7e46160c83df"
|
||
elif platform_name == "Linux":
|
||
return "4d540798108762df76de34f7bdbc667dada6b5cb"
|
||
elif platform_name == "Darwin":
|
||
return "a482bf8317d56e5ddc71437584343ace29ff545c"
|
||
else:
|
||
raise UnsupportedOperation(f"未支持的平台{platform_name}")
|
||
|
||
|
||
@pytest.fixture
|
||
def expect_md():
|
||
return "【Zc】每早合约日替攻略！<br>\n来源: Bilibili直播 魔法Zc目录<br>详情: https://live.bilibili.com/3044248<br>"
|
||
|
||
|
||
def test_gene_md(app: App, expect_md, ms_list):
|
||
from nonebot_bison.post.custom_post import CustomPost
|
||
|
||
cp = CustomPost(message_segments=ms_list)
|
||
cp_md = cp._generate_md()
|
||
assert cp_md == expect_md
|
||
|
||
|
||
@respx.mock
|
||
@pytest.mark.asyncio
|
||
async def test_gene_pic(app: App, ms_list, pic_hash):
|
||
from nonebot_bison.post.custom_post import CustomPost
|
||
|
||
pic_router = respx.get(
|
||
"http://i0.hdslb.com/bfs/live/new_room_cover/cf7d4d3b2f336c6dba299644c3af952c5db82612.jpg"
|
||
)
|
||
|
||
pic_path = Path(__file__).parent / "platforms" / "static" / "custom_post_pic.jpg"
|
||
with open(pic_path, mode="rb") as f:
|
||
mock_pic = f.read()
|
||
|
||
pic_router.mock(return_value=Response(200, stream=mock_pic))
|
||
|
||
cp = CustomPost(message_segments=ms_list)
|
||
cp_pic_bytes: list[MessageSegment] = await cp.generate_pic_messages()
|
||
|
||
pure_b64 = base64.b64decode(
|
||
cp_pic_bytes[0].data.get("file").replace("base64://", "")
|
||
)
|
||
sha1obj = hashlib.sha1()
|
||
sha1obj.update(pure_b64)
|
||
sha1hash = sha1obj.hexdigest()
|
||
assert sha1hash == pic_hash
|