当前位置: 首页 > 工具软件 > pytest-flask > 使用案例 >

在 flask 框架中,如何进行单元测试

郎仰岳
2023-12-01

在 Flask 框架中,可以使用 Python 自带的 unittest 模块进行单元测试。

具体步骤如下:

  1. 创建一个测试文件,例如 test.py。

  2. 在测试文件中导入 Flask 应用程序和 unittest 模块。

  3. 创建一个测试类,并继承 unittest.TestCase。

  4. 在测试类中编写测试方法,方法名以 test_ 开头。

  5. 在测试方法中,使用 Flask 提供的测试客户端进行请求和响应的测试。

  6. 运行测试文件,可以使用 Python 自带的 unittest 命令或者 pytest 命令。

示例代码如下:

import unittest
from app import app


class MyTestCase(unittest.TestCase):

    def setUp(self) -> None:
        app.testing = True
        self.client = app.test_client()

    def test_get(self):
        resp = self.client.get("/")
        self.assertEqual(resp.status_code, 200)
        self.assertEqual(resp.data, b"Hello World!")

    def test_post(self):
        resp = self.client.post("/post", data={"name": "yaco"})
        self.assertEqual(resp.status_code, 200)
        self.assertEqual(resp.data, b"yaco")


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

运行测试文件:

python test.py

输出结果:

.
----------------------------------------------------------------------
Ran 1 test in 0.001s

OK
 类似资料: