測試在 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 命令執行測試。