mirror of
https://github.com/nonebot/nonebot2.git
synced 2025-07-27 16:21:28 +00:00
⚗️ add more reject case
This commit is contained in:
@ -16,9 +16,11 @@ test_got = on_message()
|
||||
@test_got.got("key1", "prompt key1")
|
||||
@test_got.got("key2", "prompt key2")
|
||||
async def got(key1: str = ArgStr(), key2: str = ArgStr()):
|
||||
if key2 == "text":
|
||||
await test_got.reject("reject", at_sender=True)
|
||||
|
||||
assert key1 == "text"
|
||||
assert key2 == "text"
|
||||
await test_got.reject("reject", at_sender=True)
|
||||
assert key2 == "text_next"
|
||||
|
||||
|
||||
test_receive = on_message()
|
||||
@ -33,3 +35,22 @@ async def receive(
|
||||
assert str(z.get_message()) == "text"
|
||||
assert x is y
|
||||
await test_receive.pause("pause", at_sender=True)
|
||||
|
||||
|
||||
test_combine = on_message()
|
||||
|
||||
|
||||
@test_combine.got("a")
|
||||
@test_combine.receive()
|
||||
@test_combine.got("b")
|
||||
async def combine(a: str = ArgStr(), b: str = ArgStr(), r: Event = Received()):
|
||||
if a == "text":
|
||||
await test_combine.reject_arg("a")
|
||||
elif b == "text":
|
||||
await test_combine.reject_arg("b")
|
||||
elif str(r.get_message()) == "text":
|
||||
await test_combine.reject_receive()
|
||||
|
||||
assert a == "text_next"
|
||||
assert b == "text_next"
|
||||
assert str(r.get_message()) == "text_next"
|
||||
|
Reference in New Issue
Block a user