mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 11:18:54 -05:00
b983d5904b
* Switch RPC libs for websockets support * Implement RPC handling for core * Black reformat * Fix docs for build on travis * Modify RPC to use a Cog base class * Refactor rpc server reference as global * Handle cogbase unload method * Add an init call to handle mutable base attributes * Move RPC server reference back to the bot object * Remove unused import * Add tests for rpc method add/removal * Add tests for rpc method add/removal and cog base unloading * Add one more test * Black reformat * Add RPC mixin...fix MRO * Correct internal rpc method names * Add rpc test html file for debugging/example purposes * Add documentation * Add get_method_info * Update docs with an example RPC call specifying parameter formatting * Make rpc methods UPPER * Black reformat * Fix doc example * Modify this to match new method naming convention * Add more tests
165 lines
3.5 KiB
Python
165 lines
3.5 KiB
Python
import random
|
|
from collections import namedtuple
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from _pytest.monkeypatch import MonkeyPatch
|
|
from redbot.core import Config
|
|
from redbot.core.bot import Red
|
|
|
|
from redbot.core.drivers import red_json
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def monkeysession(request):
|
|
mpatch = MonkeyPatch()
|
|
yield mpatch
|
|
mpatch.undo()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def override_data_path(tmpdir):
|
|
from redbot.core import data_manager
|
|
|
|
data_manager.basic_config = data_manager.basic_config_default
|
|
data_manager.basic_config["DATA_PATH"] = str(tmpdir)
|
|
|
|
|
|
@pytest.fixture()
|
|
def coroutine():
|
|
async def some_coro(*args, **kwargs):
|
|
return args, kwargs
|
|
|
|
return some_coro
|
|
|
|
|
|
@pytest.fixture()
|
|
def json_driver(tmpdir_factory):
|
|
import uuid
|
|
|
|
rand = str(uuid.uuid4())
|
|
path = Path(str(tmpdir_factory.mktemp(rand)))
|
|
driver = red_json.JSON("PyTest", identifier=str(uuid.uuid4()), data_path_override=path)
|
|
return driver
|
|
|
|
|
|
@pytest.fixture()
|
|
def config(json_driver):
|
|
conf = Config(
|
|
cog_name="PyTest", unique_identifier=json_driver.unique_cog_identifier, driver=json_driver
|
|
)
|
|
yield conf
|
|
conf._defaults = {}
|
|
|
|
|
|
@pytest.fixture()
|
|
def config_fr(json_driver):
|
|
"""
|
|
Mocked config object with force_register enabled.
|
|
"""
|
|
conf = Config(
|
|
cog_name="PyTest",
|
|
unique_identifier=json_driver.unique_cog_identifier,
|
|
driver=json_driver,
|
|
force_registration=True,
|
|
)
|
|
yield conf
|
|
conf._defaults = {}
|
|
|
|
|
|
# region Dpy Mocks
|
|
@pytest.fixture()
|
|
def guild_factory():
|
|
mock_guild = namedtuple("Guild", "id members")
|
|
|
|
class GuildFactory:
|
|
def get(self):
|
|
return mock_guild(random.randint(1, 999999999), [])
|
|
|
|
return GuildFactory()
|
|
|
|
|
|
@pytest.fixture()
|
|
def empty_guild(guild_factory):
|
|
return guild_factory.get()
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def empty_channel():
|
|
mock_channel = namedtuple("Channel", "id")
|
|
return mock_channel(random.randint(1, 999999999))
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def empty_role():
|
|
mock_role = namedtuple("Role", "id")
|
|
return mock_role(random.randint(1, 999999999))
|
|
|
|
|
|
@pytest.fixture()
|
|
def member_factory(guild_factory):
|
|
mock_member = namedtuple("Member", "id guild display_name")
|
|
|
|
class MemberFactory:
|
|
def get(self):
|
|
return mock_member(random.randint(1, 999999999), guild_factory.get(), "Testing_Name")
|
|
|
|
return MemberFactory()
|
|
|
|
|
|
@pytest.fixture()
|
|
def empty_member(member_factory):
|
|
return member_factory.get()
|
|
|
|
|
|
@pytest.fixture()
|
|
def user_factory():
|
|
mock_user = namedtuple("User", "id")
|
|
|
|
class UserFactory:
|
|
def get(self):
|
|
return mock_user(random.randint(1, 999999999))
|
|
|
|
return UserFactory()
|
|
|
|
|
|
@pytest.fixture()
|
|
def empty_user(user_factory):
|
|
return user_factory.get()
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def empty_message():
|
|
mock_msg = namedtuple("Message", "content")
|
|
return mock_msg("No content.")
|
|
|
|
|
|
@pytest.fixture()
|
|
def ctx(empty_member, empty_channel, red):
|
|
mock_ctx = namedtuple("Context", "author guild channel message bot")
|
|
return mock_ctx(empty_member, empty_member.guild, empty_channel, empty_message, red)
|
|
|
|
|
|
# endregion
|
|
|
|
|
|
# region Red Mock
|
|
@pytest.fixture()
|
|
def red(config_fr):
|
|
from redbot.core.cli import parse_cli_flags
|
|
|
|
cli_flags = parse_cli_flags(["ignore_me"])
|
|
|
|
description = "Red v3 - Alpha"
|
|
|
|
Config.get_core_conf = lambda *args, **kwargs: config_fr
|
|
|
|
red = Red(cli_flags=cli_flags, description=description, pm_help=None)
|
|
|
|
yield red
|
|
|
|
red.http._session.close()
|
|
|
|
|
|
# endregion
|