我有一个 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;
我当然可以对测试数据库进行完整备份并在每次测试时恢复一次,但这会很耗时并且会大大减慢我们的开发周期。如何拥有一个易于回滚的测试数据库?