Red-DiscordBot/tests/core/test_commands.py
2023-03-27 20:49:59 -04:00

72 lines
2.0 KiB
Python

import inspect
import datetime
from dateutil.relativedelta import relativedelta
import pytest
from discord.ext import commands as dpy_commands
from redbot.core import commands
from redbot.core.commands import converter
@pytest.fixture(scope="session")
def group():
@commands.group()
async def fixturegroup(*args, **kwargs):
return args, kwargs
return fixturegroup
def is_Command(obj):
return isinstance(obj, commands.Command)
def is_Group(obj):
return isinstance(obj, commands.Group)
def test_command_decorators(coroutine):
assert is_Command(commands.command(name="cmd")(coroutine))
assert is_Group(commands.group(name="grp")(coroutine))
def test_group_decorator_methods(group, coroutine):
assert is_Command(group.command(name="cmd")(coroutine))
assert is_Group(group.group(name="grp")(coroutine))
def test_bot_decorator_methods(red, coroutine):
assert is_Command(red.command(name="cmd")(coroutine))
assert is_Group(red.group(name="grp")(coroutine))
def test_dpy_commands_reexports():
dpy_attrs = set()
for attr_name, attr_value in dpy_commands.__dict__.items():
if attr_name.startswith("_") or inspect.ismodule(attr_value):
continue
dpy_attrs.add(attr_name)
missing_attrs = dpy_attrs - set(commands.__dict__.keys())
if missing_attrs:
pytest.fail(
"redbot.core.commands is missing these names from discord.ext.commands: "
+ ", ".join(missing_attrs)
)
def test_converter_timedelta():
assert converter.parse_timedelta("1 day") == datetime.timedelta(days=1)
assert converter.parse_timedelta("1 minute") == datetime.timedelta(minutes=1)
assert converter.parse_timedelta("13 days 5 minutes") == datetime.timedelta(days=13, minutes=5)
def test_converter_relativedelta():
assert converter.parse_relativedelta("1 year") == relativedelta(years=1)
assert converter.parse_relativedelta("1 year 10 days 3 seconds") == relativedelta(
years=1, days=10, seconds=3
)