Python中如何模拟Flask请求以测试接收flask.request的业务函数

完全可以在本地实现该需求，以下是三种常用的实现方案：

方案1：直接构造Request对象测试（最轻量，仅验证函数逻辑） #

不需要启动服务、不需要配置路由，直接手动构造Flask的Request对象传入目标函数即可，适合快速验证函数逻辑：

import flask
from your_module import main # 替换为你实际的模块名

# 构造正常的GET请求（带JSON参数）
test_req = flask.Request.from_values(
    method="GET",
    json={"test_key": "test_value"},
    content_type="application/json"
)
result = main(test_req)
print(result) # 输出 (do_stuff的返回值, 200)

# 构造非法JSON的请求，测试异常分支
bad_req = flask.Request.from_values(
    method="GET",
    data="invalid json string",
    content_type="application/json"
)
result = main(bad_req)
print(result) # 输出 ('', 400)

方案2：使用Flask内置测试客户端（单元测试首选） #

如果需要走完整的Flask请求处理链路，但不想对外启动服务，可以用Flask自带的测试客户端写单元测试：

import flask
import unittest
from your_module import main, do_stuff # 替换为你实际的模块名

class TestMainFunction(unittest.TestCase):
    def setUp(self):
        self.app = flask.Flask(__name__)
        self.client = self.app.test_client()
        # 绑定路由到目标函数
        self.app.add_url_rule('/', view_func=lambda: main(flask.request), methods=['GET'])

    def test_valid_json_request(self):
        test_data = {"key": "test"}
        resp = self.client.get('/', json=test_data)
        self.assertEqual(resp.status_code, 200)
        self.assertEqual(resp.data.decode(), do_stuff(test_data))

    def test_invalid_json_request(self):
        resp = self.client.get('/', data="bad_json", content_type="application/json")
        self.assertEqual(resp.status_code, 400)

if __name__ == '__main__':
    unittest.main()

方案3：启动本地服务用requests发请求测试（端到端验证首选） #

如果需要模拟真实的HTTP请求场景，可以启动本地服务后用requests库发送真实请求验证：

1. 编写服务启动文件app.py：

from flask import Flask, request
from your_module import main # 替换为你实际的模块名

app = Flask(__name__)
@app.route('/', methods=['GET'])
def entry():
    return main(request)

if __name__ == '__main__':
    app.run(debug=True) # 启动后默认访问地址为 http://127.0.0.1:5000

2. 编写请求测试代码：

import requests

# 测试正常请求
resp = requests.get("http://127.0.0.1:5000", json={"test": "xxx"})
print("正常请求返回：", resp.status_code, resp.text)

# 测试异常请求
resp = requests.get("http://127.0.0.1:5000", data="bad_json", headers={"Content-Type": "application/json"})
print("异常请求返回：", resp.status_code) # 预期返回400

注意：按照HTTP规范，GET请求通常不携带请求体，部分代理、浏览器会自动丢弃GET请求的body内容，如果是生产环境使用，建议把函数里的method判断改为POST更符合通用规范。

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