2018-10-08 23:54:55 +02:00
|
|
|
"""Common test utilities."""
|
2021-01-01 22:31:56 +01:00
|
|
|
from unittest.mock import Mock
|
2018-10-08 23:54:55 +02:00
|
|
|
|
|
|
|
|
|
|
|
class AsyncMock(Mock):
|
|
|
|
"""Implements Mock async."""
|
|
|
|
|
|
|
|
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)
|