如何创建单元测试来检查在Flask中制作的API的响应？

Question

我正在尝试测试我在Flask中实现的JSON API

这是我的视图功能

@app.route("/dummy")
def dummy(): 
    return {"dummy":"dummy-value"}

在我的Unittest中,我正在测试使用

def setUp(self):
    self.app = my_app.app.test_client()

def test_dummy(self):
     response = self.app.get("/dummy")
     self.assertEqual(response['dummy'], "dummy-value")

但是,当我运行它时,我得到了错误 TypeError: 'dict' object is not callable

Answer 1

使用jsonify()修复错误'dict' object is not callable

from flask import jsonify
@app.route("/dummy")
def dummy(): 
    return jsonify({"dummy":"dummy-value"})

对于测试,您必须从HTTP响应中提取JSON

import json

class MyAppCase(unittest.TestCase):
    def setUp(self):
        my_app.app.config['TESTING'] = True
        self.app = my_app.app.test_client()

    def test_dummy(self):
        response = self.app.get("/dummy")
        data = json.loads(response.get_data(as_text=True))

        self.assertEqual(data['dummy'], "dummy-value")

现在这适合我.