python - 如何对返回 JSON 对象的方法进行单元测试？

Question

我有两个 Python 脚本:我们称它们为 program.py 和 program_utils.py。

program.py 看起来像这样:

from flask import Flask, request, jsonify
import program_utils

app = Flask(__name__)
@app.route('/mypage')
def my_func():
  # Do some stuff here and get a URL parameter
  my_var = request.args.get('my_var')
  # Get JSON object from another function in program_utils
  return program_utils.get_json_object(my_var)

program_utils.py 中的 get_json_object() 看起来像这样:

def get_json_object(my_var):
  # Do some calls to other methods to create the JSON object using my_var
  return json_object

我的问题是如何在 program_utils 中为 get_json_object 创建单元测试以确保它以正确的格式返回对象？我曾尝试编写常规单元测试(将方法视为返回字符串)，但出现运行时错误“在应用程序上下文之外工作”。

Answer 1

您使用测试客户端来测试 Flask View 。使用 app.test_client() 创建它，然后使用它来模拟对您的路由的请求。响应只有原始数据，如果您想将其作为 JSON 进行比较，则需要加载它。

c = app.test_client()
rv = c.get('/mypage', query_string={'my_var': 'my_value'})
assert json.loads(rv.get_data()) == expected_data

Flask docs有一整节专门介绍这个。 Werkzeug docs进入更多细节。