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

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;

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

How do I prepare a rollback-friendly test datebase?

testing

postgresql

transactions

rollback

pytest