python - 如何测试 Django 数据库模式？

Question

我想编写测试来显示数据库是否与我的 models.py 文件同步。其实我已经写过了，只是发现django每次运行测试时都会根据models.py文件创建一个新的数据库。有什么方法可以使 models.py 测试 使用现有的数据库架构？ mysql/postgresql 中的那个，而不是/myapp/models.py 中的那个？

我不关心数据库中的数据，我只关心它的模式，即我希望我的测试能够注意到数据库中的表是否比我的模式中的字段少模型.py 文件。

如果这有任何相关性，我正在使用 unittest 框架(实际上是它的 django 扩展)。

谢谢

Answer 1

我们所做的是覆盖默认的 test_runner，这样它就不会创建新的数据库来进行测试。这样，它就可以针对我们当前的本地数据库运行测试。但是，如果您使用此方法，请务必小心，因为您在测试中对数据所做的任何更改都将是永久性的。我确保我们所有的测试都将所有更改恢复到原始状态，并在服务器上保留我们数据库的原始版本并进行备份。

因此，为此，您需要将 run_test 方法从 django.test.simple 复制到项目中的某个位置——我将我的放在 myproject/test/test_runner.py 中

然后对该方法进行以下更改:

// change
old_name = settings.DATABASE_NAME
from django.db import connection
connection.creation.create_test_db(verbosity, autoclobber=not interactive)
result = unittest.TextTestRunner(verbosity=verbosity).run(suite)
connection.creation.destroy_test_db(old_name, verbosity)

// to:
result = unittest.TextTestRunner(verbosity=verbosity).run(suite)

确保在顶部执行所有必要的导入，然后在您的设置文件中进行设置:

TEST_RUNNER = 'myproject.test.test_runner.run_tests'

现在，当您运行 ./manage.py test 时，Django 将针对数据库的当前状态运行测试，而不是根据您当前的模型定义创建新版本。

您可以做的另一件事是在本地创建数据库的副本，然后像这样检查新的 run_test() 方法:

if settings.DATABASE_NAME != 'my_test_db': 
    sys.exit("You cannot run tests using the %s database. Please switch DATABASE_NAME to my_test_db in settings.py" % settings.DATABASE_NAME) 

这样就没有对主数据库运行测试的危险。