Skip to content

Commit 2ad7035

Browse files
committed
Support unban callback
1 parent e4c2d1f commit 2ad7035

2 files changed

Lines changed: 26 additions & 0 deletions

File tree

discord/ext/test/backend.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -331,6 +331,12 @@ async def ban(self, user_id: int, guild_id: int, delete_message_days: int = 1,
331331

332332
delete_member(member)
333333

334+
async def unban(self, user_id: Snowflake, guild_id: Snowflake, *, reason: str | None = None) -> None:
335+
locs = _get_higher_locs(1)
336+
guild = locs.get("self", None)
337+
member = locs.get("user", None)
338+
await callbacks.dispatch_event("unban", guild, member, reason=reason)
339+
334340
async def change_my_nickname(self, guild_id: int, nickname: str, *,
335341
reason: str | None = None) -> _types.member.Nickname:
336342
locs = _get_higher_locs(1)

tests/test_ban.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
2+
import pytest
3+
import discord
4+
import discord.ext.test as dpytest
5+
6+
@pytest.mark.asyncio
7+
async def test_ban_user(bot: discord.Client):
8+
guild = bot.guilds[0]
9+
member = guild.members[0]
10+
await guild.ban(member)
11+
12+
assert guild.get_member(member.id) is None
13+
14+
15+
@pytest.mark.asyncio
16+
async def test_unban_user(bot: discord.Client):
17+
guild = bot.guilds[0]
18+
member = guild.members[0]
19+
await guild.ban(member)
20+
await guild.unban(member)

0 commit comments

Comments
 (0)