python - python、postgresql中的数据库测试

Question

如何对使用 postgresql 的 python DAL 进行单元测试。

在 sqlite 中，您可以为每个测试创建内存数据库，但这不能为 postgresql 完成。

我想要一个可用于设置数据库并在测试完成后清理它的库。

我正在使用 Sqlalchemy 作为我的 ORM。

Answer 1

pg_tmp(1)是一个旨在简化此任务的实用程序。以下是您可能如何使用 SQLAlchemy 启动新连接:

from subprocess import check_output
from sqlalchemy import create_engine

url = check_output(['pg_tmp', '-t'])
engine = create_engine(url)

这将启动一个在 60 秒内自动销毁的新数据库。如果连接打开，pg_tmp 将等待所有事件连接关闭。