Django 单元测试约束

如何解决Django 单元测试约束

所以我在为 Check 和 Unique 约束编写单元测试时遇到了问题。以下是他们的定义：

# assert that a partner can have only one headquarter
constraints = [
   models.UniqueConstraint(
       fields=['partner','active'],condition=models.Q(headquarter=True),name='HQ already set.'
     )
]


# assert every partner contant has at least one of email/phone number pair defined
constraints = [            
   models.CheckConstraint(
       check= (
                 models.Q(email__isnull=False) &
                 models.Q(phone_number__isnull=False)
              ),name='E-mail or phone number should be set.'
   )
]

并且单元测试断言失败了：

from django.db.utils import IntegrityError
....
# HQ already defined so it should work
with self.assertRaises(Exception) as raised:
    partnerHQcopy.save()

self.assertEqual(IntegrityError,type(raised.exception))

.....
# The contact added has both email and phone number None
with self.assertRaises(Exception) as raised:
    partnerContact.save()

self.assertEqual(IntegrityError,type(raised.exception))

以及回溯：

======================================================================
ERROR: test_unique_hq_for_partner (partners.tests.PartnerModelsTestCases)
----------------------------------------------------------------------

.....

sqlite3.IntegrityError: UNIQUE constraint Failed: partners_partnerbranch.partner_id,partners_partnerbranch.active

.....

django.db.utils.IntegrityError: UNIQUE constraint Failed: partners_partnerbranch.partner_id,partners_partnerbranch.active

======================================================================
ERROR: test_partner_contact_no_email_no_phone (partners.tests.PartnerModelsTestCases)
----------------------------------------------------------------------
.....

sqlite3.IntegrityError: CHECK constraint Failed: E-mail or phone number should be set.

.....

django.db.utils.IntegrityError: CHECK constraint Failed: E-mail or phone number should be set.

解决方法

这些测试失败的原因是它引发了一个 sqlite3.IntegrityError，而不是一个 django.db.utils.IntegrityError，因此您应该断言正确的错误类型：

from sqlite3 import IntegrityError

# …

# HQ already defined so it should work
with self.assertRaises(IntegrityError):
    partnerHQCopy.save()

# …

# The contact added has both email and phone number None
with self.assertRaises(IntegrityError):
    partnerContact.save()