aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorGravatar Numerlor <[email protected]>2022-06-21 19:48:36 +0200
committerGravatar Numerlor <[email protected]>2022-06-21 19:48:36 +0200
commit67003153c718925844447127f291501adddb49c0 (patch)
tree814076445e7aeab29c305d836f4e1b9f63266cff /tests
parentstop 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.py26
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))))