如何使用Connexion框架Mock请求头进行测试？

你的核心问题是模拟的请求头结构不符合Connexion的request对象规范——Connexion的request.headers是Werkzeug提供的Headers对象（具备实例方法get），而非普通字典。直接赋值字典会导致调用headers.get时出错，以下是两种可行的解决方案：

方案一：正确使用Connexion TestContext模拟请求对象 #

重新构造符合规范的模拟request和headers对象，确保headers.get是可调用的实例方法：

from unittest.mock import MagicMock
from connexion import request
from connexion.testing import TestContext
from your_module import get_my_header  # 替换为你的实际模块路径

def test_get_my_header():
    # 模拟Headers对象，重点实现get方法
    mock_headers = MagicMock()
    mock_headers.get.return_value = "header-value"
    
    # 构造模拟的request对象，绑定headers
    mock_request = MagicMock()
    mock_request.headers = mock_headers
    
    # 注入请求上下文并执行测试
    with TestContext(context={"request": mock_request}):
        result = get_my_header()
        assert result == "header-value"
        # 验证get方法是否被正确调用
        mock_headers.get.assert_called_once_with("my-header")

方案二：利用Flask测试上下文（更简洁） #

由于Connexion基于Flask开发，其request本质是Flask请求对象的代理，因此可以直接使用Flask的test_request_context来快速设置请求头：

from flask import Flask
from your_module import get_my_header

def test_get_my_header_flask_context():
    app = Flask(__name__)
    # 直接在测试上下文中传入headers参数
    with app.test_request_context(headers={"my-header": "header-value"}):
        result = get_my_header()
        assert result == "header-value"

关键说明 #

• 避免直接用字典替换request.headers：Connexion的headers不是普通字典，而是带有方法的对象，模拟时需要保证get方法的可调用性。
• Flask测试上下文方案更轻量：如果不需要依赖Connexion特定的TestContext功能，用Flask原生的测试上下文更简单直观。

内容的提问来源于stack exchange，提问作者marikaner