def METHOD_NAME(self):
    """Patch to include proxies for async methods"""
    super().METHOD_NAME()
    for attr in {"__aenter__", "__aexit__", "__anext__"}:
        if not hasattr(MagicMock, attr):
            setattr(MagicMock, attr, _MagicProxy(attr, self))