跨LiveServerTestCase测试方法保留数据?

时间:2018-04-17 17:27:54

标签: python django unit-testing

由于LiveServerTestCase继承自TransactionTestCase,因此默认行为是在每个测试方法结束时删除测试数据。我想使用LiveServerTestCase类,但保留测试数据从方法到方法。在此示例中,test2失败,因为数据库在test1末尾被截断。

我的理解是,如果我使用TestCase,它将在每次测试结束时回滚事务并将数据库返回到其启动条件。我可以在使用LiveServerTestCase时模仿这种行为吗?

class TestTheTests(LiveServerTestCase):

    @classmethod
    def setUpTestData(cls):
        print('running setUpTestData')
        call_command('loaddata', 'datasources.yaml' )

    @classmethod
    def setUpClass(cls):
        print('starting setUpClass')
        cls.setUpTestData() # load the data for the entire test class
        super().setUpClass()

    @classmethod
    def tearDownClass(cls):
        print('finished tearDownClass')
        super().tearDownClass()

    def setUp(self):
        self.browser = webdriver.Chrome()

    def tearDown(self):
        self.browser.quit()

当我同时运行两个测试时,此测试通过:

    def test1(self):
        print('test 1 running')
        self.assertEquals(8, DataSource.objects.count(),'There should be 8 DataSource objects in test1')

此测试失败:

    def test2(self):
        print('test 2 running')
        self.assertEquals(8, DataSource.objects.count(),'There should be 8 DataSource objects in test2')

如果未在test1末尾删除数据库记录,则两者都会通过。

1 个答案:

答案 0 :(得分:0)

摘自:https://docs.djangoproject.com/en/3.1/topics/testing/overview/#rollback-emulation

Django可以通过在TestCase或TransactionTestCase的主体中将serialized_rollback选项设置为True来为每个测试案例重新加载该数据,但是请注意,这会使该测试套件的速度降低大约3倍。

这将解决您的问题:

class TestTheTests(LiveServerTestCase):
    serialized_rollback = True
    ...