测试在 Flask 中实现的 JSON API

此示例假设你知道如何使用 pytest 测试 Flask 应用程序

下面是一个 API，它使用整数值 a 和 b（例如 {"a": 1, "b": 2}）获取 JSON 输入，将它们相加并在 JSON 响应中返回 sum a + b，例如 {"sum": 3}。

# hello_add.py
from flask import Flask, request, jsonify

app = Flask(__name__)

@app.route('/add', methods=['POST'])
def add():
    data = request.get_json()
    return jsonify({'sum': data['a'] + data['b']})

使用 pytest 测试此 API

我们可以用 pytest 测试它

# test_hello_add.py
from hello_add import app
from flask import json

def test_add():        
    response = app.test_client().post(
        '/add',
        data=json.dumps({'a': 1, 'b': 2}),
        content_type='application/json',
    )

    data = json.loads(response.get_data(as_text=True))

    assert response.status_code == 200
    assert data['sum'] == 3

现在使用 py.test 命令运行测试。