velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

invites: fix infinite use

This commit is contained in:
NotNite 2021-09-21 21:42:58 -04:00
parent 8e45263f97
commit 2ec293702f
No known key found for this signature in database
GPG Key ID: BD91A5402CCEB08A
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
litecord/blueprints/invites.py
View File

@ -118,7 +118,7 @@ async def _inv_check_age(inv: dict):
await delete_invite(inv["code"]) await delete_invite(inv["code"])
raise InvalidInvite("Invite is expired") raise InvalidInvite("Invite is expired")
if inv["max_uses"] != -1 and inv["uses"] >= inv["max_uses"]: if inv["max_uses"] != 0 and inv["uses"] >= inv["max_uses"]:
await delete_invite(inv["code"]) await delete_invite(inv["code"])
raise InvalidInvite("Too many uses") raise InvalidInvite("Too many uses")

2
tests/test_invites.py
View File

@ -23,7 +23,7 @@ from litecord.enums import MessageType
pytestmark = pytest.mark.asyncio pytestmark = pytest.mark.asyncio
async def _create_invite(test_cli_user, guild, channel, max_uses=100): async def _create_invite(test_cli_user, guild, channel, max_uses=0):
resp = await test_cli_user.post( resp = await test_cli_user.post(
f'/api/v9/channels/{channel["id"]}/invites', json={"max_uses": max_uses} f'/api/v9/channels/{channel["id"]}/invites', json={"max_uses": max_uses}
) )