在 django 中测试数据库而不是 django 模型

Question

我正在使用 mysql-server 和 MySQLdb 来访问我的数据库。我什至没有使用 django 模型来访问数据库表。

现在问题出现在测试中。所有 django 文档都在谈论使用 django 模型进行测试。在测试数据时，我必须使用cursor.execute() 函数将数据插入数据库，然后我必须从中读取。但这浪费了我的时间。

TestCase 类也在事务中运行所有数据库查询，我怀疑这对于 mysql-server 和 myqldb 是否相同？

我正在使用 MySQLdb，因为它比 django 模型快。谁能告诉我如何使测试更简单？如何在 django 中测试数据库？

Answer 1

比覆盖TestCase 的setUp 和tearDown 方法更好的设置数据库的方法是通过实现setUpTestData 来摆脱Django。

from django.db import connections

class BasicTest(TestCase):
     @classmethod
     def setUpTestData(cls):
         alias = 'default'
         cursor = connections[alias].cursor()
         query = 'your query here'
         cursor.execute(query)

Django 将在tearDownClass 上删除测试数据库。

如果您的数据库在settings.py 中没有配置条目，您需要为您的TestCase 提供自己的setUp 和tearDown

@classmethod
def setUpClass(cls):
    super(BasicTest, self).setUpClass()
    db = MySQLdb.connect(host="localhost",user="root",passwd="toor")
    cls.cursor = db.cursor()

    query = 'CREATE DATABASE IF NOT EXISTS test;'
    cls.cursor.execute(query)

    query = 'USE test;'
    cls.cursor.execute(query)

@classmethod
def tearDownClass(cls):
    query = 'DROP DATABASE test;'
    cls.cursor.execute(query)
    cls.cursor.close()

这样，您仍然可以在setUpTestData 中进行其他数据操作，但您将在此处使用的光标是cls.cursor。

参考

• TestCase.setUpTestData
• TestCase.tearDownClass