mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 11:18:54 -05:00
54 lines
927 B
Python
54 lines
927 B
Python
import pytest
|
|
from redbot.core.rpc import RPC, RPCMixin
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
__all__ = ["rpc", "rpcmixin", "cog", "existing_func", "existing_multi_func"]
|
|
|
|
|
|
@pytest.fixture()
|
|
async def rpc():
|
|
rpc = RPC()
|
|
await rpc._pre_login()
|
|
return rpc
|
|
|
|
|
|
@pytest.fixture()
|
|
def rpcmixin():
|
|
r = RPCMixin()
|
|
r.rpc = MagicMock(spec=RPC)
|
|
return r
|
|
|
|
|
|
@pytest.fixture()
|
|
def cog():
|
|
class Cog:
|
|
async def cofunc(*args, **kwargs):
|
|
pass
|
|
|
|
async def cofunc2(*args, **kwargs):
|
|
pass
|
|
|
|
async def cofunc3(*args, **kwargs):
|
|
pass
|
|
|
|
def func(*args, **kwargs):
|
|
pass
|
|
|
|
return Cog()
|
|
|
|
|
|
@pytest.fixture()
|
|
def existing_func(rpc, cog):
|
|
rpc.add_method(cog.cofunc)
|
|
|
|
return cog.cofunc
|
|
|
|
|
|
@pytest.fixture()
|
|
def existing_multi_func(rpc, cog):
|
|
funcs = [cog.cofunc, cog.cofunc2, cog.cofunc3]
|
|
rpc.add_multi_method(*funcs)
|
|
|
|
return funcs
|