aboutsummaryrefslogtreecommitdiffstats
path: root/tests/helpers.py
diff options
context:
space:
mode:
authorGravatar Chris G <[email protected]>2019-10-19 20:12:59 -0600
committerGravatar GitHub <[email protected]>2019-10-19 20:12:59 -0600
commit046bb4badf459fa318070f824234af17688c256f (patch)
treeaad3b2129a11acd364b29f4fa0497d645fcd6369 /tests/helpers.py
parentBugfix - ensure .py attachment is prioritized over other non-whitelisted (diff)
parentPluralize "infractions" as necessary. (#545) (diff)
Merge branch 'master' into antimalware-cog
Diffstat (limited to 'tests/helpers.py')
-rw-r--r--tests/helpers.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/helpers.py b/tests/helpers.py
index 2908294f7..25059fa3a 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -7,6 +7,10 @@ __all__ = ('AsyncMock', 'async_test')
# TODO: Remove me on 3.8
+# Allows you to mock a coroutine. Since the default `__call__` of `MagicMock`
+# is not a coroutine, trying to mock a coroutine with it will result in errors
+# as the default `__call__` is not awaitable. Use this class for monkeypatching
+# coroutines instead.
class AsyncMock(MagicMock):
async def __call__(self, *args, **kwargs):
return super(AsyncMock, self).__call__(*args, **kwargs)