diff options
author | 2020-11-10 13:40:31 +0100 | |
---|---|---|
committer | 2020-11-10 13:40:31 +0100 | |
commit | ac82f852119e74fe88f818e1fb3cbdf34f3b54ba (patch) | |
tree | af983f028fef0f91c5e88e40a69743f8069322cc /tests/helpers.py | |
parent | Move InventoryURL converer to the converters file (diff) | |
parent | Merge pull request #1273 from python-discord/sebastiaan/bugs/codeblock-langua... (diff) |
Merge remote-tracking branch 'upstream/master' into doc-imp
# Conflicts:
# LICENSE-THIRD-PARTY
# bot/exts/info/doc.py
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 e47fdf28f..870f66197 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. |