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 响应对象没有属性 json （它的请求对象）。所以，这就是你获得例外的原因。相反，它有通用方法 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（）代替数据：这不应该使用，最终会被弃用。
使用 get（）从字典中读取值，以便在密钥丢失时不生成异常，但获取有关丢失消息的正确断言。

同时检查this Q&A。

Python鼻子测试失败了JSON响应

1 个答案: