时间戳

首页 > 解决方案 > 如何准备回滚友好的测试数据库？

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

问题描述

我有一个 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;

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

标签： postgresqltestingtransactionspytestrollback

解决方案

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

DROP DATABASE testdb;
CREATE DATABASE testdb TEMPLATE test_template;

推荐阅读