python - 测试 Flask 路由是否存在

Question

我正在创建大量 Flask 路由 using regular expressions .我想要一个单元测试来检查是否存在正确的路由以及不正确的路由 404。

这样做的一种方法是启动本地服务器并使用 urllib2.urlopen 等。但是，我希望能够在 Travis 上运行此测试，并且我假设这不是一个选项。

还有其他方法可以让我在我的应用程序上测试路由吗？

Answer 1

使用 Flask.test_client() object在你的单元测试中。该方法返回 FlaskClient instance (a werkzeug.test.TestClient subclass )，使得测试路由变得微不足道。

调用 TestClient 的结果是 Response object , 看看它是 200 还是 404 响应测试 Response.status_code attribute :

with app.test_client() as c:
    response = c.get('/some/path/that/exists')
    self.assertEquals(response.status_code, 200)

或

with app.test_client() as c:
    response = c.get('/some/path/that/doesnt/exist')
    self.assertEquals(response.status_code, 404)

参见 Testing Flask Applications chapter Flask 文档。