创建自定义“test”命令以运行Flask应用程序的测试套件

from flask import Flask
import unittest

class MyTestCase(unittest.TestCase):
    def setUp(self):
        # 在每个测试用例执行之前的准备工作
        self.app = Flask(__name__)
        self.app.config['TESTING'] = True
        # 添加其他必要的配置

    def tearDown(self):
        # 在每个测试用例执行之后的清理工作
        pass

    def test_example(self):
        # 编写测试用例
        with self.app.test_client() as client:
            response = client.get('/example')
            self.assertEqual(response.status_code, 200)
            # 添加其他断言

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

def run_tests():
    # 创建测试套件
    suite = unittest.TestLoader().loadTestsFromTestCase(MyTestCase)
    # 执行测试套件
    unittest.TextTestRunner(verbosity=2).run(suite)

from flask import Flask
from test import run_tests

app = Flask(__name__)

@app.cli.command()
def test():
    """运行测试套件"""
    run_tests()

# 添加其他路由和视图函数

if __name__ == '__main__':
    app.run()

flask test