与 pytest.raises(Exception) 不适用于 Flask app.post

Question

我有一个这样的文件夹

python
├── foo
│   ├── __init__.py
│   └── web
│       ├── __init__.py
│       ├── api
│       │   ├── __init__.py
│       │   └── helper_api.py
│       └── server.py         
└── test_server.py
    ├── test_helper_api.py

helper_api.py 就像

@helper_api.route("/helper", methods=['POST'])
def helper():
    data = request.get_json(force=True, silent=True, cache=True)
    if data is None:
        raise QueryParseException("JSON-formatted query is required, none found")

在 test_helper_api.py 中，我有

import ......
from foo.web.api.helper_api import QueryParseException

class TestClass(object):
    @pytest.fixture
    def client(self, request):
        self.client = server.app.test_client()
        return self.client

    def test_helper_api(self, client):
        with pytest.raises(QueryParseException):
            client.post('/helper')

当我运行测试类时，代码在 client.post 处失败，引发异常，但 pytest 未能捕获它。

    ---------------------------------------------------- Captured log call -----------------------------------------------------
app.py                    1761 ERROR    Exception on /helper [POST]
Traceback (most recent call last):
  File "/Users/...personal path.../flask/app.py", line 2292, in wsgi_app
    response = self.full_dispatch_request()
  .......
  File "/Users/...personal path.../foo-1.0.0-py3.6.egg/foo/web/api/helper_api.py", line 12, in precheck
    raise QueryParseException("JSON-formatted query is required, none found")
foo.web.api.helper_api.QueryParseException: JSON-formatted query is required, none found

为什么 pytest 没有捕获这个异常？

Answer 1

Flask 测试客户端是一个测试客户端，用于测试当您向端点发出请求时用户端会发生什么。 pytest 没有也不应该捕获异常的原因是错误发生在服务器端，用户不应该有服务器端异常。

相反，要测试您的

QueryParseException

是否在服务器端正确引发，您可以断言

client.post('/helper')

的状态代码。同时，在服务器端，您可以给出错误消息和状态代码，让客户端知道出了什么问题。

Answer 2

由于这是“Flask 测试客户端引发异常”的最高结果，因此 Flask 可能会引发异常，而不是捕获异常并返回 HTTP 错误。

您可以将 Flask 应用程序

DEBUG

、

TESTING

或

PROPAGATE_EXCEPTIONS

设置为

true

，这将导致 Flask 绕过异常捕获代码并重新引发异常。

# Import your Flask app, it may be named differently
from app import app

# Set TESTING to true
app.config.update({
    "TESTING": True,
})

# Will now raise an exception instead of http error
app.test_client().get("/something-that-raises-an-exception")

参考https://flask.palletsprojects.com/en/3.0.x/testing/#fixtures