在每个测试生菜django之前创建初始数据条目和刷新

时间:2014-02-23 09:02:51

标签: python django lettuce

我是在Django中使用生菜进行BDD开发的新手,但是,我需要帮助确定如何为我的模型加载初始测试数据以及如何在每次测试之前清除它们。

2 个答案:

答案 0 :(得分:1)

我认为最简单的出发点是http://lettuce.it/tutorial/tables.html。对于“刷新”数据,我使用terrain.py代码如下:

from django.db import transaction

@before.each_feature
def begin_transaction(feature):
    #shouldn't strictly be needed, but I've gotten
    #inconsistent results without it
    transaction.rollback()
    transaction.set_autocommit(False)


@after.each_feature
def end_transaction(feature):
    transaction.rollback()

使用before.each_scenario可能更合适。使用适合你的。

答案 1 :(得分:0)

我有以下内容:

@before.each_scenario
def load_scenario_fixture(scenario):
    call_command('loaddata', 'lettuce_global', interactive=False, verbosity=0)
    fixture_path = os.path.join(scenario.feature.name.lower().replace(' ', '_'), scenario.name.lower().replace(' ', '_'))
    logger.info("Loading fixture:")
    logger.info("  " + fixture_path)
    call_command('loaddata', fixture_path, interactive=False, verbosity=0)

@after.each_scenario
def flush_database(scenario):
    logger.info("Flushing the test database ...")
    call_command('flush', interactive=False, verbosity=0)

这会在每个场景之前加载一个全局测试夹具,也会加载场景的特定夹具。夹具文件路径的格式为{app} / fixtures / {feature name} / {fixture name} 方案完成后,我只需使用Django flush命令重置。