在没有事务错误的情况下单元测试 Django 模型的正确方法

Question

我正在编写一些带有 unique=True 和 blank=False 和 null=False 等约束的模型。我正在尝试为带有鼻子的模型编写测试。但是，如果我编写这样的测试：

from job_sites.models import Site, SiteType

@raises(IntegrityError)
def test_empty_site():
   s = Site()
   s.save()

@raises(IntegrityError)
def test_empty_site_type():
   st = SiteType()
   st.save()

我得到一个这样的 DatabaseError： DatabaseError: current transaction is aborted, commands ignored until end of transaction block 运行第一次测试后。

当我预计会出现错误时，运行 DJango 模型测试的正确方法是什么？

作为参考，模型如下所示：

class SiteType(models.Model):
    site_type_id = models.AutoField(primary_key=True)
    site_type = models.CharField(max_length=32, unique=True, blank=False, null=False, default=None)
    site_type_abbrev = models.CharField(max_length=32, blank=True)

    class Meta:
        db_table = u'site_types'

class Site(models.Model):
    site_id = models.AutoField(primary_key=True, blank=False, null=False, db_index=True)
    site_name = models.CharField(max_length=128, blank=False, null=False, db_index=True)
    site_type = models.ForeignKey(SiteType, blank=True, null=True)
    date_entered = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = u'sites'

我的约束和默认值如下所示：

ALTER TABLE site_types ADD CONSTRAINT site_types_site_type_name_minlen CHECK (char_length(site_type) > 0);
ALTER TABLE sites ALTER COLUMN date_entered SET DEFAULT now();
ALTER TABLE sites ADD CONSTRAINT sites_site_name_minlen CHECK (char_length(site_name) > 0);

Answer 1

您应该将测试创建为 Django 的 TestCase 的子类，而不是使用鼻子的简洁测试定义。这样，您的数据库等将在运行时为您设置和配置，所有事务都将得到神奇的处理。

关于如何为 Django 项目编写测试的概述：https://docs.djangoproject.com/en/dev/topics/testing/overview/

您尝试执行的操作类似于：

from django.db import IntegrityError
from django.utils import unittest
from job_sites.models import Site, SiteType

class TestMyStuff(unittest.TestCase):

    def test_empty_site(self):
        s = Site()
        assertRaises(IntegrityError, s.save())

    def test_empty_site_type(self):
        st = SiteType()
        assertRaises(IntegrityError, st.save())

（免责声明：我没有实际运行此代码，因此它可能无法正常工作。）

但是，测试这种东西可能是浪费时间。此处测试的唯一逻辑是 Django 内部的，因此您不会通过测试来了解您的应用程序的任何内容。