diff options
| -rw-r--r-- | bot/exts/moderation/stream.py | 8 | ||||
| -rw-r--r-- | tests/bot/exts/moderation/test_stream.py | 7 |
2 files changed, 3 insertions, 12 deletions
diff --git a/bot/exts/moderation/stream.py b/bot/exts/moderation/stream.py index 7e15864bf..0ee2a5612 100644 --- a/bot/exts/moderation/stream.py +++ b/bot/exts/moderation/stream.py @@ -9,10 +9,6 @@ from bot.converters import Expiry from bot.utils.scheduling import Scheduler from bot.utils.time import format_infraction_with_duration -# Constant error messages -USER_ALREADY_ALLOWED_TO_STREAM = f"{Emojis.cross_mark}This user can already stream." -USER_ALREADY_NOT_ALLOWED_TO_STREAM = f"{Emojis.cross_mark}This user already can't stream." - class Stream(commands.Cog): """Grant and revoke streaming permissions from users.""" @@ -54,7 +50,7 @@ class Stream(commands.Cog): # Check if user already has streaming permission already_allowed = any(Roles.video == role.id for role in user.roles) if already_allowed: - await ctx.send(USER_ALREADY_ALLOWED_TO_STREAM) + await ctx.send(f"{Emojis.cross_mark} This user can already stream.") return # Schedule task to remove streaming permission from Member @@ -80,7 +76,7 @@ class Stream(commands.Cog): await user.remove_roles(discord.Object(Roles.video)) await ctx.send(f"{Emojis.check_mark} Streaming permission taken from {user.display_name}.") else: - await ctx.send(USER_ALREADY_NOT_ALLOWED_TO_STREAM) + await ctx.send(f"{Emojis.cross_mark} This user already can't stream.") def setup(bot: Bot) -> None: diff --git a/tests/bot/exts/moderation/test_stream.py b/tests/bot/exts/moderation/test_stream.py index 15956a9de..2ac274699 100644 --- a/tests/bot/exts/moderation/test_stream.py +++ b/tests/bot/exts/moderation/test_stream.py @@ -2,16 +2,11 @@ import unittest from bot.constants import Roles -from bot.exts.moderation.stream import Stream -from tests.helpers import MockBot, MockMember, MockRole +from tests.helpers import MockMember, MockRole class StreamCommandTest(unittest.IsolatedAsyncioTestCase): - def setUp(self) -> None: - self.bot = MockBot() - self.cog = Stream(self.bot) - def test_checking_if_user_has_streaming_permission(self): """ Test searching for video role in Member.roles |