Дразнить асин c с помощью объекта, возвращающего объект MagicMock вместо значения

Question

Я пытаюсь смоделировать функцию client.encrypt, я попытался смоделировать приведенный ниже код тестового примера, но после запуска я возвращаю объект MagicMock вместо значения.

Ниже приведена реализация:

Class KMSClient():
  def __init__():
     # some operation. 

  async with aioboto3.client(**self.kms_config) as client:
    try:

        await client.encrypt(**body)
    except ConnectionError:
        print('Error occurred')

Контрольные примеры:

@patch('aioboto3.client', newcallable=AsyncContextManagerMock)
@testing.gen_test()
async def test_raises_kmserror(self, aiopatch):
    client = kms.KMSClient(os.environ['KEY'])

    aiopatch.return_value.aenter.encrypt.return_value = {}

    value = str(uuid.uuid4())
    await client.encrypt(value)