在 Flask 框架中,可以使用 Python 自带的 unittest 模块进行单元测试。
具体步骤如下:
创建一个测试文件,例如 test.py。
在测试文件中导入 Flask 应用程序和 unittest 模块。
创建一个测试类,并继承 unittest.TestCase。
在测试类中编写测试方法,方法名以 test_
开头。
在测试方法中,使用 Flask 提供的测试客户端进行请求和响应的测试。
运行测试文件,可以使用 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