Python unit test - Mocking out a decorator that has a parameter

gistfile1.txt

# Great answer by Danila Ganchar at https://stackoverflow.com/questions/47900727/mock-authentication-decorator-in-unittesting


# mock out check_auth decorator
from functools import wraps
from unittest.mock import patch


def mock_decorator(*args):
    def decorator(f):
        @wraps(f)
        def decorated_function(*args, **kwargs):
            return f(*args, **kwargs)
        return decorated_function
    return decorator


patch('myflask_app.views.check_auth', mock_decorator).start()


# alternatively, instead of using patch, you /could/ just monkey patch
#from myflask_app import views
#views.check_auth = mock_decorator

#... BELOW this is where you import the views using the decorator. Must mock it BEFORE they are first encountered.