aboutsummaryrefslogtreecommitdiffstats
path: root/tests/helpers.py
diff options
context:
space:
mode:
authorGravatar ks129 <[email protected]>2020-11-21 09:37:57 +0200
committerGravatar GitHub <[email protected]>2020-11-21 09:37:57 +0200
commit95429b10aab81d849ab86241f411bbc122ea1594 (patch)
tree707b7f6055e8cd6d11f1034ed6296470bc437416 /tests/helpers.py
parentEH tests: Fix InWhitelistCheckFailure import path (diff)
parentMerge pull request #1287 from python-discord/help-channel-msg (diff)
Merge branch 'master' into error-handler-test
Diffstat (limited to 'tests/helpers.py')
-rw-r--r--tests/helpers.py21
1 files changed, 2 insertions, 19 deletions
diff --git a/tests/helpers.py b/tests/helpers.py
index d35012278..ea5e2822c 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -5,7 +5,7 @@ import itertools
import logging
import unittest.mock
from asyncio import AbstractEventLoop
-from typing import Callable, Iterable, Optional
+from typing import Iterable, Optional
import discord
from aiohttp import ClientSession
@@ -14,6 +14,7 @@ from discord.ext.commands import Context
from bot.api import APIClient
from bot.async_stats import AsyncStatsClient
from bot.bot import Bot
+from tests._autospec import autospec # noqa: F401 other modules import it via this module
for logger in logging.Logger.manager.loggerDict.values():
@@ -26,24 +27,6 @@ for logger in logging.Logger.manager.loggerDict.values():
logger.setLevel(logging.CRITICAL)
-def autospec(target, *attributes: str, **kwargs) -> Callable:
- """Patch multiple `attributes` of a `target` with autospecced mocks and `spec_set` as True."""
- # Caller's kwargs should take priority and overwrite the defaults.
- kwargs = {'spec_set': True, 'autospec': True, **kwargs}
-
- # Import the target if it's a string.
- # This is to support both object and string targets like patch.multiple.
- if type(target) is str:
- target = unittest.mock._importer(target)
-
- def decorator(func):
- for attribute in attributes:
- patcher = unittest.mock.patch.object(target, attribute, **kwargs)
- func = patcher(func)
- return func
- return decorator
-
-
class HashableMixin(discord.mixins.EqualityComparable):
"""
Mixin that provides similar hashing and equality functionality as discord.py's `Hashable` mixin.