python 2.6 - django TestCase - assertRaises ValidationError clean() 方法答案

【问题标题】：python 2.6 - django TestCase - assertRaises ValidationError clean() methodpython 2.6 - django TestCase - assertRaises ValidationError clean() 方法
【发布时间】：2013-12-19 16:49:31
【问题描述】：

Django 1.5 和 Python 2.6。

该模型有一个 clean() 方法来验证 job.company_id 必须等于 job.location.company_id

我正在尝试为此编写一个测试，但测试不是通过/失败，而是以来自模型clean() 方法的验证错误消息结束。

这里是代码（不相关的部分省略）：

在models.py中：

class Job(models.Model):
    title = models.CharField(max_length=200, verbose_name="Job title")
    company = models.ForeignKey(Company)    
    location = models.ForeignKey(Location, blank=True, null=True)

    def clean(self):
        from django.core.exceptions import ValidationError
        '''
        Location.company_id must equal Job.company_id
        '''
        if (self.company_id != Location.objects.get(pk=self.location_id).company_id):
            raise ValidationError('Location is not valid for company')

在tests.py中：

class job_cannot_have_invalid_location_and_can_have_valid_location(TestCase):
    def test_jobs_and_locations(self):
        job2 = Job.objects.create(company_id=company2.id)
        location1 = Location.objects.create(company_id=company1.id)
        job2.location_id = location1.id
        self.assertRaises(ValidationError, job2.clean())

当我运行 python manage.py 测试时：

.E.
======================================================================
ERROR: test_jobs_and_locations     (companies.tests.job_cannot_have_invalid_location_and_can_have_valid_location)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/home/djangobox/jobboard/companies/tests.py", line 60, in test_jobs_and_locations
    self.assertRaises(ValidationError, job2.clean())
  File "/home/djangobox/jobboard/companies/models.py", line 151, in clean
    raise ValidationError('Location is not valid for company')
ValidationError: [u'Location is not valid for company']

【问题讨论】：

标签： python django python-2.6 testcase

【解决方案1】：

问题在于您对assertRaises 的使用。它应该接受异常和可调用检查：http://docs.python.org/2/library/unittest.html#unittest.TestCase.assertRaises

但是，您没有传递可调用的job2.clean，而是传递了可调用的job2.clean() 的返回值。将调用更改为

self.assertRaises(ValidationError, job2.clean)

应该修复异常和您的测试用例。

【讨论】：