Question

我正在设置一个非常简单的Django项目，并且测试非常简单：

def test_name(self):
    t = Thing.objects.create(name='a')
    print(t.id)
    import time
    time.sleep(30)
    self.assertEqual('a', t.name)

当然，测试通过，但数据库（TEST数据库）没有填充我的Thing模型信息，即使我实际上可以看到它的id为你可以看到我的剧本。

当我连接到数据库时，Thing表总是空的（我在文档中也看到了一些关于它的评论）。

现在，我如何告诉Django实际填充数据？我正在使用mysql，并检查完整的日志，我看到Django在填充数据之前创建了一个SAVEPOINT（未提交），一旦测试通过，它就会回到之前的SAVEPOINT。

我正在使用：

Django==2.0.1
mysqlclient==1.3.12
pytest==3.3.2
pytest-django==3.1.2

我希望Django实际填充TEST数据库，使用我的测试中的信息，最后删除该数据库，这就是为什么我假设Django正在创建一个全新的数据库测试

Answer 1

如果您使用的是继承TransactionTestCase类的django TestCase类，则数据库将始终重置为原始状态。

https://docs.djangoproject.com/en/2.0/topics/testing/tools/#django.test.TestCase

https://docs.djangoproject.com/en/2.0/topics/testing/tools/#django.test.TransactionTestCase

Answer 2

纯函数测试的django.test.TransactionTestCase的吊坠是标记

pytest.mark.django_db(transaction=True)

来自pytest-django。示例（使用sqlite3后端）：

@pytest.mark.django_db(transaction=True)
def test_model_written_to_db():
    obj = MyModel.objects.create(name='foo')
    assert obj.id == 1
    conn = sqlite3.connect('/tmp/mytestdatabase.sqlite3')
    cur = conn.cursor()
    cur.execute("SELECT * FROM backend_mymodel")
    assert len(cur.fetchall()) == 1

如果要将标记自动应用于所有测试，可以使用自定义pytest_collection_modifyitems挂钩来执行此操作。在conftest.py：

中

import pytest

def pytest_collection_modifyitems(items):
    for item in items:
        item.add_marker(pytest.mark.django_db(transaction=True))

现在您可以从上面的测试中移除pytest.mark.django_db标记，它仍然会表现相同。

但是，自定义挂钩或pytest灯具不适用于unittest样式的测试用例子类django.test.TestCase，因此您最好的选择是将django.test.TransactionTestCase作为子类，在Ghariani Mohamed中建议his answer。

强制Django测试将模型写入数据库

2 个答案: