mirror of
https://github.com/KimigaiiWuyi/GenshinUID.git
synced 2025-07-03 13:33:29 +00:00
124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
from nonebug import App
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
argnames="alias",
|
|
argvalues=[
|
|
"钟离推荐",
|
|
"钟离攻略",
|
|
"岩王爷攻略", # alias
|
|
],
|
|
)
|
|
async def test_get_guide_pic(app: App, alias, load_guide: None):
|
|
from utils import make_event
|
|
from nonebot.adapters.onebot.v11 import Bot, Message, MessageSegment
|
|
|
|
from GenshinUID.genshinuid_guide import get_guide_pic
|
|
|
|
async with app.test_matcher(get_guide_pic) as ctx:
|
|
bot = ctx.create_bot(base=Bot)
|
|
|
|
event = make_event(message=Message(alias))
|
|
ctx.receive_event(bot, event)
|
|
ctx.should_call_send(
|
|
event,
|
|
MessageSegment.image(
|
|
"https://file.microgg.cn/MiniGG/guide/钟离.jpg"
|
|
),
|
|
True,
|
|
)
|
|
ctx.should_finished()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_guide_pic_traveler(app: App, load_guide: None):
|
|
from utils import make_event
|
|
from nonebot.adapters.onebot.v11 import Bot, Message, MessageSegment
|
|
|
|
from GenshinUID.genshinuid_guide import get_guide_pic
|
|
|
|
async with app.test_matcher(get_guide_pic) as ctx:
|
|
bot = ctx.create_bot(base=Bot)
|
|
|
|
event = make_event(message=Message("旅行者风推荐"))
|
|
ctx.receive_event(bot, event)
|
|
ctx.should_call_send(
|
|
event,
|
|
MessageSegment.image(
|
|
"https://file.microgg.cn/MiniGG/guide/旅行者-风.jpg"
|
|
),
|
|
True,
|
|
)
|
|
ctx.should_finished()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_guide_pic_failed(app: App, load_guide: None):
|
|
from utils import make_event
|
|
from nonebot.adapters.onebot.v11 import Bot, Message
|
|
|
|
from GenshinUID.genshinuid_guide import get_guide_pic
|
|
|
|
async with app.test_matcher(get_guide_pic) as ctx:
|
|
bot = ctx.create_bot(base=Bot)
|
|
|
|
event = make_event(message=Message("蔡徐坤攻略"))
|
|
ctx.receive_event(bot, event)
|
|
ctx.should_finished()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
argnames="alias",
|
|
argvalues=[
|
|
"参考面板钟离",
|
|
"参考面板岩王爷", # alias
|
|
],
|
|
)
|
|
async def test_get_bluekun_pic(app: App, alias, load_guide: None):
|
|
from utils import make_event
|
|
from nonebot.adapters.onebot.v11 import Bot, Message, MessageSegment
|
|
|
|
from GenshinUID.genshinuid_guide import get_bluekun_pic
|
|
|
|
with open(Path("../GenshinUID/genshinuid_guide/img/钟离.jpg"), "rb") as f:
|
|
data = f.read()
|
|
async with app.test_matcher(get_bluekun_pic) as ctx:
|
|
bot = ctx.create_bot(base=Bot)
|
|
|
|
event = make_event(message=Message(alias))
|
|
ctx.receive_event(bot, event)
|
|
ctx.should_call_send(event, MessageSegment.image(data), True)
|
|
ctx.should_finished()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
argnames="alias",
|
|
argvalues=[
|
|
"岩",
|
|
"冰",
|
|
],
|
|
)
|
|
async def test_get_bluekun_pic_element(app: App, alias, load_guide: None):
|
|
from utils import make_event
|
|
from nonebot.adapters.onebot.v11 import Bot, Message, MessageSegment
|
|
|
|
from GenshinUID.genshinuid_guide import get_bluekun_pic
|
|
|
|
with open(
|
|
Path(f"../GenshinUID/genshinuid_guide/img/{alias}.jpg"), "rb"
|
|
) as f:
|
|
data = f.read()
|
|
async with app.test_matcher(get_bluekun_pic) as ctx:
|
|
bot = ctx.create_bot(base=Bot)
|
|
|
|
event = make_event(message=Message(f"参考面板{alias}"))
|
|
ctx.receive_event(bot, event)
|
|
ctx.should_call_send(event, MessageSegment.image(data), True)
|
|
ctx.should_finished()
|