2018-10-08 23:54:55 +02:00
|
|
|
"""Common test utilities."""
|
2020-05-03 11:27:19 -07:00
|
|
|
from tests.async_mock import Mock
|
2018-10-08 23:54:55 +02:00
|
|
|
|
|
|
|
|
|
|
|
class AsyncMock(Mock):
|
|
|
|
"""Implements Mock async."""
|
|
|
|
|
2019-11-26 08:40:08 +02:00
|
|
|
# pylint: disable=useless-super-delegation
|
2018-10-08 23:54:55 +02:00
|
|
|
async def __call__(self, *args, **kwargs):
|
|
|
|
"""Hack for async support for Mock."""
|
2019-09-25 00:38:20 +02:00
|
|
|
return super().__call__(*args, **kwargs)
|