python - 龙卷风。类 Django 的测试运行程序和测试数据库

Question

我喜欢 django 单元测试，因为它们会在运行时创建和删除测试数据库。

存在哪些为龙卷风创建测试数据库的方法？

UPD：我对在测试运行时创建 postgresql 测试数据库感兴趣。

Answer 1

我发现最简单的方法就是对测试数据库使用 SQL 转储。创建一个数据库，用夹具填充它，然后将其写入文件。只需load_fixtures在运行测试之前调用（或在您想要重置数据库时）。这种方法当然可以改进，但它已经足够满足我的需要了。

import os
import unittest2

import tornado.database

settings = dict(
    db_host="127.0.0.1:3306",
    db_name="testdb",
    db_user="testdb",
    db_password="secret",
    db_fixtures_file=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'fixtures.sql'),
)

def load_fixtures():
    """Fixtures are stored in an SQL dump.
    """
    os.system("psql %s --user=%s --password=%s < %s" % (settings['db_name'], 
        settings['db_user'], settings['db_password'], settings['db_fixtures_file']))

    return tornado.database.Connection(
        host=settings['db_host'], database=settings['db_name'],
        user=settings['db_user'], password=settings['db_password'])