Flask-SQLAlchemy集成测试逐次清除带外键约束的Postgres数据库数据方法

你当前的写法报错是因为reversed(meta.sorted_tables)仅能按照SQLAlchemy建表时的依赖顺序倒序删除，一旦模型存在循环外键依赖，或者外键关联关系没有被sorted_tables正确识别，就会触发外键约束的IntegrityError。同时DELETE语句逐行删除数据，性能也比批量清空方案差。

你有两种可行方案，均不需要依赖事务回滚，完全适配应用内调用db.session.commit的场景：

方案1：使用Postgres专属TRUNCATE语法（性能最优） #

Postgres的TRUNCATE支持批量清空表数据，配合CASCADE参数可以自动处理外键关联依赖，清空速度远快于逐行DELETE，还支持重置自增序列：

def tearDown(self):
    meta = db.metadata
    for table in meta.sorted_tables:
        # CASCADE自动删除关联外键数据，RESTART IDENTITY重置自增主键序列
        db.session.execute(f"TRUNCATE TABLE {table.name} CASCADE RESTART IDENTITY;")
    db.session.commit()

该方案的优势是性能最高，重置后数据库状态和新建表完全一致，自增ID会从初始值重新计数，测试结果一致性更好。

方案2：临时禁用会话级外键约束 #

如果需要保留DELETE的写法，可以在当前会话临时禁用外键约束检查，删除完成后再恢复，该配置仅对当前测试连接生效，不会影响其他数据库操作：

def tearDown(self):
    # 临时禁用当前会话的外键约束检查
    db.session.execute("SET session_replication_role = 'replica';")
    meta = db.metadata
    for table in reversed(meta.sorted_tables):
        db.session.execute(table.delete())
    # 恢复外键约束检查
    db.session.execute("SET session_replication_role = 'origin';")
    db.session.commit()

• 两种方案都不需要重建表结构，仅清空数据，完全符合你对测试性能的要求
• 优先推荐方案1，TRUNCATE的执行效率比逐行DELETE高10~100倍，数据量越大优势越明显

内容的提问来源于stack exchange，提问作者Darius Mandres