From caa816a6367e5c6d4ec15fcc69bbd2000878f7b1 Mon Sep 17 00:00:00 2001 From: Sebastian Law Date: Mon, 7 Mar 2022 14:10:28 -0800 Subject: [PATCH] Add more tests for utils Co-authored-by: Nadir Chowdhury --- tests/test_utils.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index 0cbbacf42..b1977da98 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -137,6 +137,21 @@ def test_resolve_invite(url, code): assert utils.resolve_invite(url).code == code +@pytest.mark.parametrize( + ('url', 'event_id'), + [ + ('https://discordapp.com/invite/dpy', None), + ('https://discord.com/invite/dpy', None), + ('https://discord.gg/dpy', None), + ('https://discordapp.com/invite/dpy?event=22222222', 22222222), + ('https://discord.com/invite/dpy?event=4098', 4098), + ('https://discord.gg/dpy?event=727', 727), + ], +) +def test_resolve_invite_event(url, event_id: typing.Optional[int]): + assert utils.resolve_invite(url).event == event_id + + @pytest.mark.parametrize( ('url', 'code'), [ @@ -149,7 +164,7 @@ def test_resolve_template(url, code): assert utils.resolve_template(url) == code -@pytest.mark.parametrize('mention', ['@everyone', '@here']) +@pytest.mark.parametrize('mention', ['@everyone', '@here', '<@80088516616269824>', '<@!80088516616269824>', '<@&381978264698224660>']) def test_escape_mentions(mention): assert mention not in utils.escape_mentions(mention) assert mention not in utils.escape_mentions(f"one {mention} two")