如何使用unittest测试Flask中返回render_template的路由

测试逻辑说明 #

你可以直接使用Flask内置的测试客户端完成路由测试，不需要启动正式的Web服务，测试核心逻辑是模拟发送GET/POST请求到目标路由，校验响应的状态码、返回内容是否符合预期。

完整测试代码 #

假设你的Flask应用实例定义在app.py文件中，新建测试文件test_home_route.py，内容如下：

import unittest
# 导入你自己的Flask应用实例，替换为你实际的应用文件名称
from app import app

class HomeRouteTestCase(unittest.TestCase):
    def setUp(self):
        # 开启测试模式，禁用错误捕获方便排查问题
        app.config.update({
            "TESTING": True,
        })
        # 初始化测试客户端
        self.client = app.test_client()

    def test_get_request_access(self):
        # 测试GET请求访问主路由
        resp = self.client.get("/")
        # 校验响应状态码为200（请求成功）
        self.assertEqual(resp.status_code, 200)
        # 可选：校验返回的是home.html的内容，比如home.html存在唯一文本"欢迎来到首页"，就加下面的校验
        # self.assertIn(b"欢迎来到首页", resp.data)

    def test_post_request_access(self):
        # 测试POST请求访问主路由（适配你路由配置的methods支持POST方法）
        resp = self.client.post("/")
        self.assertEqual(resp.status_code, 200)

if __name__ == "__main__":
    unittest.main()

执行方法 #

直接运行命令：
python test_home_route.py
即可看到测试用例的执行结果。

补充说明 #

• 如果需要严格验证渲染的模板确实是home.html，可以使用Flask的测试信号捕获模板渲染事件，或者给home.html添加独有的特征文本做内容校验即可。
• 如果你的Flask应用使用应用工厂模式构建，只需要在setUp方法中先调用工厂函数生成应用实例，再创建测试客户端即可。

内容的提问来源于stack exchange，提问作者John