Flask-SQLAlchemy 集成測試在每次測試后使用外鍵約束清除數據庫

Question

我想知道如何使用flask-sqlalchemy和unittest編寫快速集成測試，而不必在每個測試中創建和刪除表。 我使用 Postgres 作為我的數據庫。

現在，表分別在setUpClass和tearDownClass中創建和刪除，從性能的角度來看這很好。 我需要的是一種在每個單獨的測試中刪除所有數據並“重置”數據庫的方法，而無需重新創建所有表。

我得到的最接近的是這段代碼，但由於外鍵約束，它會引發IntegrityError 。

def tearDown(self):
    meta = db.metadata
    for table in reversed(meta.sorted_tables):
        db.session.execute(table.delete())

    db.session.commit()

重要提示：由於我正在做集成測試，我不可避免地在我的應用程序代碼中點擊了db.session.commit ，這會使任何會話事務無效，因此我無法將其用作解決方案。

Answer 1

嘗試這個：

for table in reversed(meta.sorted_tables):
    db.session.execute(f"TRUNCATE {table.name} RESTART IDENTITY CASCADE;")

CASCADE在這里可以解決問題，來自postgres 文檔：

Automatically truncate all tables that have foreign-key references to any of the named tables, or to any tables added to the group due to CASCADE.

所以它告訴 postgres 刪除所有指向截斷表中行的行。

另一方面， RESTART IDENTITY ：

Automatically restart sequences owned by columns of the truncated table(s).

使您的自動增量列從頭開始。