如何在 Flask 中进行单元测试

2023-04-13 flask 如何在单元测试

在 Flask 中进行单元测试的主要方法是使用 Flask 测试客户端。首先，需要安装 Flask 测试客户端库：

pip install flask-testing

然后，在测试脚本（比如 tests.py）中导入 Flask 测试客户端并创建测试类：

from flask_testing import TestCase
from app import app

class TestApp(TestCase):
    def create_app(self):
        app.config['TESTING'] = True
        return app

在这个例子中，测试类 TestApp 继承了 TestCase 类，需要实现 create_app() 方法来创建测试应用。在这个方法中，将 app.config['TESTING'] 设置为 True，以启用测试模式。返回的应用将会是测试客户端使用的应用。

接下来，可以在测试类中编写测试方法。例如，可以测试一个返回“pidancode.com”的简单路由：

def test_index(self):
    response = self.client.get('/')
    self.assertEqual(response.data.decode('utf-8'), 'pidancode.com')

在这个测试方法中，使用 self.client.get() 方法获取路由返回的响应。接下来，使用 self.assertEqual() 方法断言响应的数据等于“pidancode.com”。

最后，需要执行测试。可以使用 unittest 模块或 pytest 等测试框架来运行测试。例如，使用 unittest：

import unittest

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

完整的测试脚本示例：

from flask_testing import TestCase
from app import app

class TestApp(TestCase):
    def create_app(self):
        app.config['TESTING'] = True
        return app

    def test_index(self):
        response = self.client.get('/')
        self.assertEqual(response.data.decode('utf-8'), 'pidancode.com')

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