ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-17 19:40:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ Feature: 添加正则匹配文本注入 (#1457)

Browse Source 
Co-authored-by: Ju4tCode <42488585+yanyongyu@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Akirami
2022-12-09 14:42:54 +08:00
committed by GitHub
parent 8176cd189c
commit 36d7b44741
7 changed files with 47 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_rule.py
View File

@ -328,7 +328,7 @@ async def test_shell_command(app: App):
@pytest.mark.asyncio
@pytest.mark.parametrize(
"pattern,type,text,expected,matched,group,dict",
"pattern,type,text,expected,matched,string,group,dict",
[
(
r"(?P<key>key\d)",
@ -336,11 +336,12 @@ async def test_shell_command(app: App):
"_key1_",
True,
"key1",
"key1",
("key1",),
{"key": "key1"},
),
(r"foo", "message", None, False, None, None, None),
(r"foo", "notice", "foo", False, None, None, None),
(r"foo", "message", None, False, None, None, None, None),
(r"foo", "notice", "foo", False, None, None, None, None),
],
)
async def test_regex(
@ -350,12 +351,13 @@ async def test_regex(
text: Optional[str],
expected: bool,
matched: Optional[str],
string: Optional[str],
group: Optional[Tuple[str, ...]],
dict: Optional[Dict[str, str]],
):
from nonebot.typing import T_State
from nonebot.rule import RegexRule, regex
from nonebot.consts import REGEX_DICT, REGEX_GROUP, REGEX_MATCHED
from nonebot.consts import REGEX_STR, REGEX_DICT, REGEX_GROUP, REGEX_MATCHED
test_regex = regex(pattern)
dependent = list(test_regex.checkers)[0]
@ -369,6 +371,7 @@ async def test_regex(
state = {}
assert await dependent(event=event, state=state) == expected
assert state.get(REGEX_MATCHED) == matched
assert state.get(REGEX_STR) == string
assert state.get(REGEX_GROUP) == group
assert state.get(REGEX_DICT) == dict