mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-02 00:56:10 +08:00
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
import pytest
|
||
from nonebug import App
|
||
|
||
from .utils import fake_group_message_event, fake_private_message_event, fake_superuser
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_query_with_superuser_private(app: App):
|
||
from nonebot.adapters.onebot.v11.bot import Bot
|
||
from nonebot.adapters.onebot.v11.message import Message
|
||
from nonebot_bison.config_manager import group_manage_matcher
|
||
|
||
async with app.test_matcher(group_manage_matcher) as ctx:
|
||
bot = ctx.create_bot(base=Bot)
|
||
event = fake_private_message_event(
|
||
message=Message("群管理"),
|
||
sender=fake_superuser,
|
||
to_me=True,
|
||
user_id=fake_superuser.user_id,
|
||
)
|
||
ctx.receive_event(bot, event)
|
||
ctx.should_pass_rule()
|
||
ctx.should_pass_permission()
|
||
ctx.should_call_api(
|
||
"get_group_list", {}, [{"group_id": 101, "group_name": "test group"}]
|
||
)
|
||
ctx.should_call_send(
|
||
event, Message("请选择需要管理的群：\n1. 101 - test group\n请输入左侧序号"), True
|
||
)
|
||
event_1_err = fake_private_message_event(
|
||
message=Message("0"),
|
||
sender=fake_superuser,
|
||
to_me=True,
|
||
user_id=fake_superuser.user_id,
|
||
)
|
||
ctx.receive_event(bot, event_1_err)
|
||
ctx.should_call_send(event_1_err, "请输入正确序号", True)
|
||
ctx.should_rejected()
|
||
event_1_ok = fake_private_message_event(
|
||
message=Message("1"),
|
||
sender=fake_superuser,
|
||
to_me=True,
|
||
user_id=fake_superuser.user_id,
|
||
)
|
||
ctx.receive_event(bot, event_1_ok)
|
||
ctx.should_call_send(event_1_ok, "请输入需要使用的命令：添加订阅，查询订阅，删除订阅", True)
|
||
event_2_err = fake_private_message_event(
|
||
message=Message("222"),
|
||
sender=fake_superuser,
|
||
to_me=True,
|
||
user_id=fake_superuser.user_id,
|
||
)
|
||
ctx.receive_event(bot, event_2_err)
|
||
ctx.should_call_send(event_2_err, "请输入正确的命令", True)
|
||
ctx.should_rejected()
|
||
event_2_ok = fake_private_message_event(
|
||
message=Message("查询订阅"),
|
||
sender=fake_superuser,
|
||
to_me=True,
|
||
user_id=fake_superuser.user_id,
|
||
)
|
||
ctx.receive_event(bot, event_2_ok)
|
||
ctx.should_pass_rule()
|
||
ctx.should_pass_permission()
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_query_with_superuser_group_tome(app: App):
|
||
from nonebot.adapters.onebot.v11.bot import Bot
|
||
from nonebot.adapters.onebot.v11.message import Message
|
||
from nonebot_bison.config_manager import group_manage_matcher
|
||
|
||
async with app.test_matcher(group_manage_matcher) as ctx:
|
||
bot = ctx.create_bot(base=Bot)
|
||
event = fake_group_message_event(
|
||
message=Message("群管理"),
|
||
sender=fake_superuser,
|
||
to_me=True,
|
||
user_id=fake_superuser.user_id,
|
||
)
|
||
ctx.receive_event(bot, event)
|
||
ctx.should_pass_rule()
|
||
ctx.should_pass_permission()
|
||
ctx.should_call_send(event, Message("该功能只支持私聊使用，请私聊Bot"), True)
|