Mocking a custom context manager within a function

Question

I have a custom context manager defined as:

# ~/db/query.py

@contextmanager
def get_db_connection(conn_string, **kwargs):