如何使用unittest测试Flask中的外键

pip install Flask

import unittest
from flask import Flask
from flask_sqlalchemy import SQLAlchemy

class TestFlaskApp(unittest.TestCase):
    def setUp(self):
        # 创建Flask应用
        self.app = Flask(__name__)
        self.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
        self.app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

        # 创建数据库对象
        self.db = SQLAlchemy(self.app)

        # 定义数据库模型
        class User(self.db.Model):
            id = self.db.Column(self.db.Integer, primary_key=True)
            name = self.db.Column(self.db.String(50))
            email = self.db.Column(self.db.String(50))
            address = self.db.relationship('Address', backref='user')

        class Address(self.db.Model):
            id = self.db.Column(self.db.Integer, primary_key=True)
            address = self.db.Column(self.db.String(100))
            user_id = self.db.Column(self.db.Integer, self.db.ForeignKey('user.id'))

        # 创建所有数据库表
        self.db.create_all()

    def tearDown(self):
        # 删除所有数据库表
        self.db.drop_all()

    def test_foreign_key(self):
        # 创建用户
        user = User(name='John Doe', email='john@example.com')
        self.db.session.add(user)
        self.db.session.commit()

        # 创建地址
        address = Address(address='123 Main St', user=user)
        self.db.session.add(address)
        self.db.session.commit()

        # 检查外键是否正常工作
        self.assertEqual(address.user, user)

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

python test_app.py

.
----------------------------------------------------------------------
Ran 1 test in 0.123s

OK