3 files changed, 128 insertions, 0 deletions
diff --git a/tests/utils/__init__.py b/tests/utils/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/tests/utils/__init__.py
diff --git a/tests/utils/test_checks.py b/tests/utils/test_checks.py
new file mode 100644
index 000000000..7121acebd
--- /dev/null
+++ b/tests/utils/test_checks.py
@@ -0,0 +1,66 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from bot.utils import checks
+
+
+[email protected]()
+def context():
+    return MagicMock()
+
+
+def test_with_role_check_without_guild(context):
+    context.guild = None
+
+    assert not checks.with_role_check(context)
+
+
+def test_with_role_check_with_guild_without_required_role(context):
+    context.guild = True
+    context.author.roles = []
+
+    assert not checks.with_role_check(context)
+
+
+def test_with_role_check_with_guild_with_required_role(context):
+    context.guild = True
+    role = MagicMock()
+    role.id = 42
+    context.author.roles = (role,)
+
+    assert checks.with_role_check(context, role.id)
+
+
+def test_without_role_check_without_guild(context):
+    context.guild = None
+
+    assert not checks.without_role_check(context)
+
+
+def test_without_role_check_with_unwanted_role(context):
+    context.guild = True
+    role = MagicMock()
+    role.id = 42
+    context.author.roles = (role,)
+
+    assert not checks.without_role_check(context, role.id)
+
+
+def test_without_role_check_without_unwanted_role(context):
+    context.guild = True
+    role = MagicMock()
+    role.id = 42
+    context.author.roles = (role,)
+
+    assert checks.without_role_check(context, role.id + 10)
+
+
+def test_in_channel_check_for_correct_channel(context):
+    context.channel.id = 42
+    assert checks.in_channel_check(context, context.channel.id)
+
+
+def test_in_channel_check_for_incorrect_channel(context):
+    context.channel.id = 42
+    assert not checks.in_channel_check(context, context.channel.id + 10)
diff --git a/tests/utils/test_time.py b/tests/utils/test_time.py
new file mode 100644
index 000000000..4baa6395c
--- /dev/null
+++ b/tests/utils/test_time.py
@@ -0,0 +1,62 @@
+import asyncio
+from datetime import datetime, timezone
+from unittest.mock import patch
+
+import pytest
+from dateutil.relativedelta import relativedelta
+
+from bot.utils import time
+from tests.helpers import AsyncMock
+
+
+[email protected](
+    ('delta', 'precision', 'max_units', 'expected'),
+    (
+        (relativedelta(days=2), 'seconds', 1, '2 days'),
+        (relativedelta(days=2, hours=2), 'seconds', 2, '2 days and 2 hours'),
+        (relativedelta(days=2, hours=2), 'seconds', 1, '2 days'),
+        (relativedelta(days=2, hours=2), 'days', 2, '2 days'),
+
+        # Does not abort for unknown units, as the unit name is checked
+        # against the attribute of the relativedelta instance.
+        (relativedelta(days=2, hours=2), 'elephants', 2, '2 days and 2 hours'),
+
+        # Very high maximum units, but it only ever iterates over
+        # each value the relativedelta might have.
+        (relativedelta(days=2, hours=2), 'hours', 20, '2 days and 2 hours'),
+    )
+)
+def test_humanize_delta(
+        delta: relativedelta,
+        precision: str,
+        max_units: int,
+        expected: str
+):
+    assert time.humanize_delta(delta, precision, max_units) == expected
+
+
+[email protected]('max_units', (-1, 0))
+def test_humanize_delta_raises_for_invalid_max_units(max_units: int):
+    with pytest.raises(ValueError, match='max_units must be positive'):
+        time.humanize_delta(relativedelta(days=2, hours=2), 'hours', max_units)
+
+
+[email protected](
+    ('stamp', 'expected'),
+    (
+        ('Sun, 15 Sep 2019 12:00:00 GMT', datetime(2019, 9, 15, 12, 0, 0, tzinfo=timezone.utc)),
+    )
+)
+def test_parse_rfc1123(stamp: str, expected: str):
+    assert time.parse_rfc1123(stamp) == expected
+
+
+@patch('asyncio.sleep', new_callable=AsyncMock)
+def test_wait_until(sleep_patch):
+    start = datetime(2019, 1, 1, 0, 0)
+    then = datetime(2019, 1, 1, 0, 10)
+
+    # No return value
+    assert asyncio.run(time.wait_until(then, start)) is None
+
+    sleep_patch.assert_called_once_with(10 * 60)