1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
from typing import Type, Union
import pytest
from nonebug import App
from nonebot.matcher import Matcher
from nonebot.adapters.onebot.v11 import Bot, Message, MessageSegment
from ATRI.permission import toggle_master
from .utils import group_message_event
async def __test_wrapper(
app: App,
test_matcher: Type[Matcher],
test_command: str,
test_arg: Union[str, MessageSegment],
test_reply: str,
):
async with app.test_matcher(test_matcher) as ctx:
bot = ctx.create_bot(base=Bot)
event = group_message_event(message=Message(test_command))
ctx.receive_event(bot, event)
ctx.should_call_send(event, "要操作的目标是？", True)
event = group_message_event(message=Message(test_arg))
ctx.receive_event(bot, event)
ctx.should_call_send(event, test_reply, True)
ctx.should_finished()
@pytest.mark.asyncio
async def test_manage(app: App):
await toggle_master("1145141919")
from ATRI.plugins.manage import block_user
await __test_wrapper(app, block_user, "封禁用户", "114514", "用户 114514 危！")
from ATRI.plugins.manage import unblock_user
await __test_wrapper(app, unblock_user, "解封用户", "114514", "用户 114514 已解封")
from ATRI.plugins.manage import block_group
await __test_wrapper(app, block_group, "封禁群", "114514", "群 114514 危！")
from ATRI.plugins.manage import unblock_group
await __test_wrapper(app, unblock_group, "解封群", "114514", "群 114514 已解封")
from ATRI.plugins.manage import toggle_global_service
await __test_wrapper(app, toggle_global_service, "全局控制", "涩图", "服务 涩图 已全局禁用")
await __test_wrapper(app, toggle_global_service, "全局控制", "涩图", "服务 涩图 已全局启用")
from ATRI.plugins.manage import toggle_group_service
await __test_wrapper(app, toggle_group_service, "控制", "涩图", "服务 涩图 已针对本群禁用")
await __test_wrapper(app, toggle_group_service, "控制", "涩图", "服务 涩图 已针对本群启用")
from ATRI.plugins.manage import track_error
await __test_wrapper(
app, track_error, "追踪", "abcdefg", "操作 abcdefg 失败...原因：\n未找到对应ID的信息"
)
from ATRI.plugins.manage import toggle_user_service
async with app.test_matcher(toggle_user_service) as ctx:
bot = ctx.create_bot(base=Bot)
event = group_message_event(message=Message("对用户114514禁用涩图"))
ctx.receive_event(bot, event)
ctx.should_call_send(event, "已禁止用户 114514 使用 涩图", True)
ctx.should_finished()
async with app.test_matcher(toggle_user_service) as ctx:
bot = ctx.create_bot(base=Bot)
event = group_message_event(message=Message("对用户114514启用涩图"))
ctx.receive_event(bot, event)
ctx.should_call_send(event, "已允许用户 114514 使用 涩图", True)
ctx.should_finished()
await toggle_master("1145141919")
|
