def METHOD_NAME(test_fn):
    def wrapper(*args, **kwargs):
        with patch.object(
            feature_flag_manager,
            "get_feature_flags",
            side_effect=mock_get_feature_flags,
        ):
            test_fn(*args, **kwargs)
    return functools.update_wrapper(wrapper, test_fn)