在Python中测试flash消息的内容

import unittest
from flask import Flask, flash, redirect, render_template, request, url_for

class FlashMessageTest(unittest.TestCase):
    def setUp(self):
        self.app = Flask(__name__)
        self.app.secret_key = 'test_secret_key'

    def test_flash_message(self):
        with self.app.test_request_context():
            with self.app.test_client() as client:
                # 发送GET请求
                response = client.get('/')

                # 检查flash消息是否为空
                self.assertIsNone(request.session.get('_flashes'))

                # 发送POST请求
                response = client.post('/', data={'message': 'Test message'})

                # 检查flash消息是否包含正确的内容
                with self.app.test_request_context():
                    self.assertEqual(request.session['_flashes'][0][1], 'Test message')

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