Python Nose 测试在 JSON 响应上失败

Question

这是我的 Flask-Restful 应用程序中 report_runner.py 中 ReportRunner 类中的方法:

class ReportRunner(object):

  def __init__(self):
        pass

  def setup_routes(self, app):
      app.add_url_rule("/run_report", view_func=self.run_report)

  def request_report(self, key):
    # code #

  def key_exists(self, key):
    # code #

  def run_report(self):
    key = request.args.get("key", "")
    if self.key_exists(key):
      self.request_report(report_type, key)
      return jsonify(message = "Success! Your report has been created.")
    else:
      response = jsonify({"message": "Error => report key not found on server."})
      response.status_code = 404
      return response

Nose 测试调用与该路线关联的 URL

def setUp(self):
    self.setup_flask()
    self.controller = Controller()
    self.report_runner = ReportRunner()
    self.setup_route(self.report_runner)

def test_run_report(self):
    rr = Report(key = "daily_report")
    rr.save()
    self.controller.override(self.report_runner, "request_report")
    self.controller.expectAndReturn(self.report_runner.request_report("daily_report"), True )
    self.controller.replay()
    response = self.client.get("/run_report?key=daily_report")
    assert_equals({"message": "Success! Your report has been created."}, response.json)
    assert_equals(200, response.status_code)

测试失败并显示以下消息:

AttributeError: 'Response' object has no attribute 'json'

但根据docs看来你就是这样做的。我是更改方法的返回值，还是需要以不同方式构建测试？

测试现在通过了，写成这样:

json_response = json.loads(response.data)
assert_equals("Success! Your report has been created.", json_response["message"])

但我不清楚这两种方法之间的区别。

Answer 1

根据 Flask API Response 对象没有属性 json(它是 Request 对象)。所以，这就是你得到异常(exception)的原因。相反，它具有返回响应主体的字符串表示形式的通用方法 get_data()。

json_response = json.loads(response.get_data())
assert_equals("Success! Your report has been created.", json_response.get("message", "<no message>"))

因此，它与您所拥有的很接近，除了:

建议使用

• get_data() 而不是 data，如 API 所述:不应使用，最终将被弃用。

• 使用 get() 从字典中读取值以在 key 丢失时不生成异常，但获得关于丢失消息的正确断言。

检查 this Q&A还有。