From 2ec293702f24c59bd55ebf873d803950c432a37d Mon Sep 17 00:00:00 2001 From: NotNite Date: Tue, 21 Sep 2021 21:42:58 -0400 Subject: [PATCH] invites: fix infinite use --- litecord/blueprints/invites.py | 2 +- tests/test_invites.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litecord/blueprints/invites.py b/litecord/blueprints/invites.py index 150787a..c31820a 100644 --- a/litecord/blueprints/invites.py +++ b/litecord/blueprints/invites.py @@ -118,7 +118,7 @@ async def _inv_check_age(inv: dict): await delete_invite(inv["code"]) 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"]) raise InvalidInvite("Too many uses") diff --git a/tests/test_invites.py b/tests/test_invites.py index f37ba09..7f4b8f7 100644 --- a/tests/test_invites.py +++ b/tests/test_invites.py @@ -23,7 +23,7 @@ from litecord.enums import MessageType 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( f'/api/v9/channels/{channel["id"]}/invites', json={"max_uses": max_uses} )