如何准备回滚友好的测试数据库？

Question

我有一个 PostgreSQL 数据库，其中填充了用于使用 pytest 进行自动化测试的假数据。 基本上，我将对测试数据库运行每个测试，检查结果，回滚所有内容以使测试环境不会改变，然后运行下一个测试，依此类推。 由于显而易见的原因，正在测试的应用程序代码（它是 Flask 应用程序 FWIW）涉及许多对BEGIN / COMMIT ，我不想改变它。

目前，我的灯具看起来像

@pytest.fixture
def app():
    app = create_app(TestConfig)
    app.config["TESTING"] = True

    cur = app.db_postgres.cursor()
    cur.execute("BEGIN")

    with app.app_context():
        yield app

    cur.execute("ROLLBACK")

我试图解决事务的问题，但 PostgreSQL 13 似乎不支持自主事务。 更具体地说，第二个COMMIT; 给我一个WARNING: there is already a transaction in progress中，并且INSERT操作只要COMMIT; 被执行，渲染最终的ROLLBACK; 无用。

-- Setup
BEGIN;

-- Application code can do whatever they want
BEGIN;
INSERT INTO test_foo(bar) VALUES (42);
COMMIT;
SELECT bar FROM test_foo;  -- assert the result is 42

-- Teardown
ROLLBACK;

我当然可以对测试数据库进行完整备份并在每次测试时恢复一次，但这会很耗时并且会大大减慢我们的开发周期。 如何拥有一个易于回滚的测试数据库？

Answer 1

要么设计你的测试，让它们自己清理，要么创建一个模板数据库test_template并在测试之间运行它：

DROP DATABASE testdb;
CREATE DATABASE testdb TEMPLATE test_template;