mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-27 18:45:05 +08:00
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
|
from utils import make_fake_message
|
||
|
|
|
||
|
|
|
||
|
|
def test_message_template():
|
||
|
|
from nonebot.adapters import MessageTemplate
|
||
|
|
|
||
|
|
Message = make_fake_message()
|
||
|
|
|
||
|
|
template = MessageTemplate("{a:custom}{b:text}{c:image}", Message)
|
||
|
|
|
||
|
|
@template.add_format_spec
|
||
|
|
def custom(input: str) -> str:
|
||
|
|
return input + "-custom!"
|
||
|
|
|
||
|
|
formatted = template.format(a="test", b="test", c="https://example.com/test")
|
||
|
|
assert formatted.extract_plain_text() == "test-custom!test"
|
||
|
|
assert str(formatted) == "test-custom!test[fake:image]"
|
||
|
|
|
||
|
|
|
||
|
|
def test_message_slice():
|
||
|
|
|
||
|
|
Message = make_fake_message()
|
||
|
|
MessageSegment = Message.get_segment_class()
|
||
|
|
|
||
|
|
message = Message(
|
||
|
|
[
|
||
|
|
MessageSegment.text("test"),
|
||
|
|
MessageSegment.image("test2"),
|
||
|
|
MessageSegment.image("test3"),
|
||
|
|
MessageSegment.text("test4"),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert message[0] == MessageSegment.text("test")
|
||
|
|
|
||
|
|
assert message[0:2] == Message(
|
||
|
|
[MessageSegment.text("test"), MessageSegment.image("test2")]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert message["image"] == Message(
|
||
|
|
[MessageSegment.image("test2"), MessageSegment.image("test3")]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert message["image", 0] == MessageSegment.image("test2")
|
||
|
|
assert message["image", 0:2] == message["image"]
|
||
|
|
|
||
|
|
assert message.index(message[0]) == 0
|
||
|
|
assert message.index("image") == 1
|
||
|
|
|
||
|
|
assert message.get("image") == message["image"]
|
||
|
|
assert message.get("image", 114514) == message["image"]
|
||
|
|
assert message.get("image", 1) == Message([message["image", 0]])
|
||
|
|
|
||
|
|
assert message.count("image") == 2
|