aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Matteo Bertucci <[email protected]>2021-04-24 17:04:48 +0200
committerGravatar Matteo Bertucci <[email protected]>2021-04-24 17:05:36 +0200
commit1fdd5aabd4ef5e356f358fdb6e9b26a5b5da99ce (patch)
tree9f0640734a6853ce11d2802000fbbfe4361f3f9c
parentTests: members shouldn't have any public flags (diff)
Tests: simplify public flags handling
Co_authored-by: Numerlor <[email protected]>
-rw-r--r--tests/bot/exts/info/test_information.py10
1 files changed, 2 insertions, 8 deletions
diff --git a/tests/bot/exts/info/test_information.py b/tests/bot/exts/info/test_information.py
index d2ecee033..770660fe3 100644
--- a/tests/bot/exts/info/test_information.py
+++ b/tests/bot/exts/info/test_information.py
@@ -281,13 +281,10 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):
"""The embed should use the string representation of the user if they don't have a nick."""
ctx = helpers.MockContext(channel=helpers.MockTextChannel(id=1))
user = helpers.MockMember()
- public_flags = unittest.mock.MagicMock()
- public_flags.__iter__.return_value = iter(())
- public_flags.verified_bot = False
+ user.public_flags = unittest.mock.MagicMock(verified_bot=False)
user.nick = None
user.__str__ = unittest.mock.Mock(return_value="Mr. Hemlock")
user.colour = 0
- user.public_flags = public_flags
embed = await self.cog.create_user_embed(ctx, user)
@@ -301,13 +298,10 @@ class UserEmbedTests(unittest.IsolatedAsyncioTestCase):
"""The embed should use the nick if it's available."""
ctx = helpers.MockContext(channel=helpers.MockTextChannel(id=1))
user = helpers.MockMember()
- public_flags = unittest.mock.MagicMock()
- public_flags.__iter__.return_value = iter(())
- public_flags.verified_bot = False
+ user.public_flags = unittest.mock.MagicMock(verified_bot=False)
user.nick = "Cat lover"
user.__str__ = unittest.mock.Mock(return_value="Mr. Hemlock")
user.colour = 0
- user.public_flags = public_flags
embed = await self.cog.create_user_embed(ctx, user)