diff options
author | 2020-11-21 09:37:57 +0200 | |
---|---|---|
committer | 2020-11-21 09:37:57 +0200 | |
commit | 95429b10aab81d849ab86241f411bbc122ea1594 (patch) | |
tree | 707b7f6055e8cd6d11f1034ed6296470bc437416 /tests/helpers.py | |
parent | EH tests: Fix InWhitelistCheckFailure import path (diff) | |
parent | Merge 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.py | 21 |
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. |