aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorGravatar Numerlor <[email protected]>2020-03-15 16:11:35 +0100
committerGravatar Numerlor <[email protected]>2020-03-15 16:11:35 +0100
commit252b385e46ef542203e69f4f6d147dadbcec8f0f (patch)
tree4d321be5577cd69fdc182fbf485ed5e6bbcc0321 /tests
parentAdd tests for `_notifier`. (diff)
Use dict instead of a set and custom class.
The FirstHash class is no longer necessary with only channels and the current loop in tuples. FirstHash was removed, along with its tests and tests were adjusted for new dict behaviour.
Diffstat (limited to 'tests')
-rw-r--r--tests/bot/cogs/moderation/test_silence.py28
1 files changed, 3 insertions, 25 deletions
diff --git a/tests/bot/cogs/moderation/test_silence.py b/tests/bot/cogs/moderation/test_silence.py
index d4719159e..6114fee21 100644
--- a/tests/bot/cogs/moderation/test_silence.py
+++ b/tests/bot/cogs/moderation/test_silence.py
@@ -2,33 +2,11 @@ import unittest
from unittest import mock
from unittest.mock import MagicMock, Mock
-from bot.cogs.moderation.silence import FirstHash, Silence, SilenceNotifier
+from bot.cogs.moderation.silence import Silence, SilenceNotifier
from bot.constants import Channels, Emojis, Guild, Roles
from tests.helpers import MockBot, MockContext, MockTextChannel
-class FirstHashTests(unittest.TestCase):
- def setUp(self) -> None:
- self.test_cases = (
- (FirstHash(0, 4), FirstHash(0, 5)),
- (FirstHash("string", None), FirstHash("string", True))
- )
-
- def test_hashes_equal(self):
- """Check hashes equal with same first item."""
-
- for tuple1, tuple2 in self.test_cases:
- with self.subTest(tuple1=tuple1, tuple2=tuple2):
- self.assertEqual(hash(tuple1), hash(tuple2))
-
- def test_eq(self):
- """Check objects are equal with same first item."""
-
- for tuple1, tuple2 in self.test_cases:
- with self.subTest(tuple1=tuple1, tuple2=tuple2):
- self.assertTrue(tuple1 == tuple2)
-
-
class SilenceNotifierTests(unittest.IsolatedAsyncioTestCase):
def setUp(self) -> None:
self.alert_channel = MockTextChannel()
@@ -41,7 +19,7 @@ class SilenceNotifierTests(unittest.IsolatedAsyncioTestCase):
channel = Mock()
with mock.patch.object(self.notifier, "_silenced_channels") as silenced_channels:
self.notifier.add_channel(channel)
- silenced_channels.add.assert_called_with(FirstHash(channel, self.notifier._current_loop))
+ silenced_channels.__setitem__.assert_called_with(channel, self.notifier._current_loop)
def test_add_channel_starts_loop(self):
"""Loop is started if `_silenced_channels` was empty."""
@@ -59,7 +37,7 @@ class SilenceNotifierTests(unittest.IsolatedAsyncioTestCase):
channel = Mock()
with mock.patch.object(self.notifier, "_silenced_channels") as silenced_channels:
self.notifier.remove_channel(channel)
- silenced_channels.remove.assert_called_with(FirstHash(channel))
+ silenced_channels.__delitem__.assert_called_with(channel)
def test_remove_channel_stops_loop(self):
"""Notifier loop is stopped if `_silenced_channels` is empty after remove."""