Use decorator to mock an import within the wrapped function

Question

I want to write a decorator that, when applied, mocks an import for the mocked function, e.g., I want the following code:

from .decorator import mock_api
from .biz_logic import call_api


@mock_api
def test_api_mock():
    call_api()

Where call_api is contained in a module that either directly or indirectly imports a class called Api from the module api_library. I want the implementation to look something like this, but it doesn't work

import mock


class MockApi:
    pass


def mock_api(func):
    @mock.patch(
        "api_library.Api", mock.MagicMock(return_value=MockApi)
    )
    def wrapper(*args, **kwargs):
        func(*args, **kwargs)

    return wrapper

I'm inspired by the moto library, and want to accomplish something similiar to what they've done with convenient decorators like mock_s3