2018-10-08 21:54:55 +00:00
|
|
|
"""Common test utilities."""
|
2021-01-01 21:31:56 +00:00
|
|
|
from unittest.mock import Mock
|
2018-10-08 21:54:55 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
class AsyncMock(Mock):
|
|
|
|
|
"""Implements Mock async."""
|
|
|
|
|
|
2019-11-26 06:40:08 +00:00
|
|
|
# pylint: disable=useless-super-delegation
|
2018-10-08 21:54:55 +00:00
|
|
|
async def __call__(self, *args, **kwargs):
|
|
|
|
|
"""Hack for async support for Mock."""
|
2019-09-24 22:38:20 +00:00
|
|
|
return super().__call__(*args, **kwargs)
|
