tests/helpers.py


1
2
3
4
5
6
7
8
9
10

from unittest.mock import MagicMock


__all__ = ('AsyncMock',)


# TODO: Remove me on 3.8
class AsyncMock(MagicMock):
    async def __call__(self, *args, **kwargs):
        return super(AsyncMock, self).__call__(*args, **kwargs)