aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorGravatar Bast <[email protected]>2020-09-24 14:16:10 -0700
committerGravatar Bast <[email protected]>2020-09-24 14:16:10 -0700
commitce80892eb3928c7c312a221c9d0271698f1563f4 (patch)
tree69d8a294f81e340724667659d4d2494171d52820 /tests
parentAdd BOT vs USER token detection, properly handling bot tokens for bots in the... (diff)
Change the mod alert message component for the user token detection
Clean up mock usage, docstrings, unnecessarily split-lined function calls
Diffstat (limited to 'tests')
-rw-r--r--tests/bot/exts/filters/test_token_remover.py51
1 files changed, 14 insertions, 37 deletions
diff --git a/tests/bot/exts/filters/test_token_remover.py b/tests/bot/exts/filters/test_token_remover.py
index 90d40d1df..5f28ab571 100644
--- a/tests/bot/exts/filters/test_token_remover.py
+++ b/tests/bot/exts/filters/test_token_remover.py
@@ -22,12 +22,8 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):
self.msg = MockMessage(id=555, content="hello world")
self.msg.channel.mention = "#lemonade-stand"
- self.msg.guild.get_member = MagicMock(
- return_value=MagicMock(
- bot=False,
- __str__=MagicMock(return_value="Woody"),
- ),
- )
+ self.msg.guild.get_member.return_value.bot = False
+ self.msg.guild.get_member.return_value.__str__.return_value = "Woody"
self.msg.author.__str__ = MagicMock(return_value=self.msg.author.name)
self.msg.author.avatar_url_as.return_value = "picture-lemon.png"
@@ -212,7 +208,7 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):
is_valid_timestamp,
is_maybe_valid_hmac,
):
- """None should be returned if no matches have valid user IDs or timestamps."""
+ """None should be returned if no matches have valid user IDs, HMACs, and timestamps."""
token_re.finditer.return_value = [mock.create_autospec(Match, spec_set=True, instance=True)]
token_cls.return_value = mock.create_autospec(Token, spec_set=True, instance=True)
is_valid_id.return_value = False
@@ -281,10 +277,7 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):
return_value = TokenRemover.format_log_message(self.msg, token)
- self.assertEqual(
- return_value,
- log_message.format.return_value,
- )
+ self.assertEqual(return_value, log_message.format.return_value)
log_message.format.assert_called_once_with(
author=self.msg.author,
author_id=self.msg.author.id,
@@ -296,42 +289,29 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):
@autospec("bot.exts.filters.token_remover", "UNKNOWN_USER_LOG_MESSAGE")
def test_format_userid_log_message_unknown(self, unknown_user_log_message):
- """
- Should correctly format the user ID portion of the log message when the user ID is
- not found in the server.
- """
+ """Should correctly format the user ID portion when the actual user it belongs to is unknown."""
token = Token("NDcyMjY1OTQzMDYyNDEzMzMy", "XsySD_", "s45jqDV_Iisn-symw0yDRrk_jf4")
unknown_user_log_message.format.return_value = " Partner"
msg = MockMessage(id=555, content="hello world")
- msg.guild.get_member = MagicMock(return_value=None)
+ msg.guild.get_member.return_value = None
return_value = TokenRemover.format_userid_log_message(msg, token)
- self.assertEqual(
- return_value,
- (unknown_user_log_message.format.return_value, False),
- )
- unknown_user_log_message.format.assert_called_once_with(
- user_id=472265943062413332,
- )
+ self.assertEqual(return_value, (unknown_user_log_message.format.return_value, False))
+ unknown_user_log_message.format.assert_called_once_with(user_id=472265943062413332)
@autospec("bot.exts.filters.token_remover", "KNOWN_USER_LOG_MESSAGE")
def test_format_userid_log_message_bot(self, known_user_log_message):
- """
- Should correctly format the user ID portion of the log message when the user ID is
- not found in the server.
- """
+ """Should correctly format the user ID portion when the ID belongs to a known bot."""
token = Token("NDcyMjY1OTQzMDYyNDEzMzMy", "XsySD_", "s45jqDV_Iisn-symw0yDRrk_jf4")
known_user_log_message.format.return_value = " Partner"
msg = MockMessage(id=555, content="hello world")
- msg.guild.get_member = MagicMock(return_value=MagicMock(__str__=MagicMock(return_value="Sam"), bot=True))
+ msg.guild.get_member.return_value.__str__.return_value = "Sam"
+ msg.guild.get_member.return_value.bot = True
return_value = TokenRemover.format_userid_log_message(msg, token)
- self.assertEqual(
- return_value,
- (known_user_log_message.format.return_value, False),
- )
+ self.assertEqual(return_value, (known_user_log_message.format.return_value, False))
known_user_log_message.format.assert_called_once_with(
user_id=472265943062413332,
@@ -341,16 +321,13 @@ class TokenRemoverTests(unittest.IsolatedAsyncioTestCase):
@autospec("bot.exts.filters.token_remover", "KNOWN_USER_LOG_MESSAGE")
def test_format_log_message_user_token_user(self, user_token_message):
- """Should correctly format the log message with info from the message and token."""
+ """Should correctly format the user ID portion when the ID belongs to a known user."""
token = Token("NDY3MjIzMjMwNjUwNzc3NjQx", "XsySD_", "s45jqDV_Iisn-symw0yDRrk_jf4")
user_token_message.format.return_value = "Partner"
return_value = TokenRemover.format_userid_log_message(self.msg, token)
- self.assertEqual(
- return_value,
- (user_token_message.format.return_value, True),
- )
+ self.assertEqual(return_value, (user_token_message.format.return_value, True))
user_token_message.format.assert_called_once_with(
user_id=467223230650777641,
user_name="Woody",