diff options
author | 2022-06-21 19:48:36 +0200 | |
---|---|---|
committer | 2022-06-21 19:48:36 +0200 | |
commit | 67003153c718925844447127f291501adddb49c0 (patch) | |
tree | 814076445e7aeab29c305d836f4e1b9f63266cff /tests | |
parent | stop cleanup task when manager is destroyed (diff) |
ensure tuples from pos arg and kwarg tuples are differentiated
Diffstat (limited to 'tests')
-rw-r--r-- | tests/botcore/utils/test_cooldown.py | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/tests/botcore/utils/test_cooldown.py b/tests/botcore/utils/test_cooldown.py index 87c433ce..00e5a052 100644 --- a/tests/botcore/utils/test_cooldown.py +++ b/tests/botcore/utils/test_cooldown.py @@ -1,23 +1,18 @@ import unittest -from collections.abc import Iterable from unittest.mock import patch -from botcore.utils.cooldown import _CommandCooldownManager - - -def create_argument_tuple(*args, **kwargs) -> Iterable[object]: - return (*args, *kwargs.items()) +from botcore.utils.cooldown import _CommandCooldownManager, _create_argument_tuple class CommandCooldownManagerTests(unittest.IsolatedAsyncioTestCase): test_call_args = ( - create_argument_tuple(0), - create_argument_tuple(a=0), - create_argument_tuple([]), - create_argument_tuple(a=[]), - create_argument_tuple(1, 2, 3, a=4, b=5, c=6), - create_argument_tuple([1], [2], [3], a=[4], b=[5], c=[6]), - create_argument_tuple([1], 2, [3], a=4, b=[5], c=6), + _create_argument_tuple(0), + _create_argument_tuple(a=0), + _create_argument_tuple([]), + _create_argument_tuple(a=[]), + _create_argument_tuple(1, 2, 3, a=4, b=5, c=6), + _create_argument_tuple([1], [2], [3], a=[4], b=[5], c=[6]), + _create_argument_tuple([1], 2, [3], a=4, b=[5], c=6), ) async def asyncSetUp(self): @@ -47,3 +42,8 @@ class CommandCooldownManagerTests(unittest.IsolatedAsyncioTestCase): with self.subTest(arguments_tuple=call_args): self.cooldown_manager.set_cooldown(0, call_args) self.assertFalse(self.cooldown_manager.is_on_cooldown(0, call_args)) + + def test_keywords_and_tuples_differentiated(self): + self.cooldown_manager.set_cooldown(0, _create_argument_tuple(("a", 0))) + self.assertFalse(self.cooldown_manager.is_on_cooldown(0, _create_argument_tuple(a=0))) + self.assertTrue(self.cooldown_manager.is_on_cooldown(0, _create_argument_tuple(("a", 0)))) |