diff options
| -rw-r--r-- | tests/bot/exts/moderation/test_stream.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/tests/bot/exts/moderation/test_stream.py b/tests/bot/exts/moderation/test_stream.py new file mode 100644 index 000000000..872627fc1 --- /dev/null +++ b/tests/bot/exts/moderation/test_stream.py @@ -0,0 +1,45 @@ +import unittest +from bot.constants import TIME_FORMATS +from bot.exts.moderation.stream import Stream +from tests.helpers import MockContext, MockBot + + +class StreamCommandTest(unittest.IsolatedAsyncioTestCase): + + def setUp(self) -> None: + self.bot = MockBot() + self.cog = Stream(self.bot) + self.ctx = MockContext() + + def test_linking_time_format_from_alias_or_key(self): + FORMATS = [] + for key, entry in TIME_FORMATS.items(): + FORMATS.extend(entry["aliases"]) + FORMATS.append(key) + + test_cases = (("sec", "second"), + ("s", "second"), + ("seconds", "second"), + ("second", "second"), + ("secs", "second"), + ("min", "minute"), + ("m", "minute"), + ("minutes", "minute"), + ("hr", "hour"), + ("hrs", "hour"), + ("hours", "hour"), + ("d", "day"), + ("days", "day"), + ("yr", "year"), + ("yrs", "year"), + ("y", "year")) + + for case in test_cases: + linked = self.cog._link_from_alias(case[0])[1] + self.assertEqual(linked, case[1]) + + def test_parsing_duration_and_time_format_to_seconds(self): + test_cases = ((1, "minute", 60), (5, "second", 5), (2, "day", 172800)) + for case in test_cases: + time_in_seconds = self.cog._parse_time_to_seconds(case[0], case[1]) + self.assertEqual(time_in_seconds, case[2]) |