使用PostgreSQL在SQLAlchemy测试中回滚数据库事务

Question

我正在构建一个Pyramid Web应用程序，该应用程序建立在SQLAlchemy的顶部，并且仅依赖PostgreSQL作为其数据库后端。

什么是具有单元测试结构的方法，以便

• 为了加快测试速度，数据库事务会在teardown()或测试套件的其他清理钩子上回滚。

• 可以使用其他加快测试速度的技巧，例如，如果SQLAlchemy和PostgreSQL具有与SQLite相对应的:in:memory:数据库，

• 如果标准库unittest框架之外的特定功能使编写测试用例更加容易，则可以选择自定义测试运行器py.test 。

Answer 1

由于此问题与您的其他问题有点类似，因此我将在此处复制答案的相关部分：

所有测试用例类都应从基类的子类中定义，该基类定义了通用的tearDown方法：

class BaseTest(unittest.TestCase):

    def setUp(self):
        # This makes things nicer if the previous test fails
        # - without this all subsequent tests fail
        self.tearDown()

        self.config = testing.setUp()

    def tearDown(self):
        testing.tearDown()
        session.expunge_all()
        session.rollback()

Answer 2

对于测试，您可以在RAMdisk上初始化PostgreSQL数据目录（可以在/dev/shm/下创建一个目录，该目录在现代Linux发行版中作为tmpfs挂载）。 您可以在非标准端口上运行Postgres。

PGTEMP=`mktemp -d /dev/shm/pgtemp.XXXXXX`
initdb -D "$PGTEMP"
postgres -D "$PGTEMP" -k "$PGTEMP" -p 54321