如何在Django中测试手动DB事务代码？

Question

我正在将遗留系统的数据传输到Django中。为了确保当前数据库的完整性，我正在手动提交所有内容。

但是，编写单元测试时，事务将不能正确回滚。由于TestCase可能使用了事务，有没有什么办法可以正确测试依赖事务的Django中的代码？

@transaction.commit_manually 
def import_records(): 
    # initial prep 
    try: 
     import_data() 
    except Exception as error: 
     rollback = True 
    except (KeyboardInterrupt, SystemExit): 
     sys.stdout.write("Import canceled\n") 
     rollback = True 
    if rollback is True: 
     transaction.rollback() 
    # save history of import 

Answer 1

我相信你正在寻找TransactionTestCase它处理的建立和拆除不同然后正常TestCase。