8

我有一个模型，我正在尝试在不调用数据库层的情况下测试验证。我将发布一些示例代码，而不是用文字来描述。这里的问题是与 Bar 的 ForeignKey 关系，这与我要测试的内容无关，但会阻止我运行我想要的测试。

首先，myapp/models.py：

from django.core.exceptions import ValidationError
from django.db import models


class BadFooError(ValidationError):
    pass


class Bar(models.Model):
    description = models.CharField(max_length=20)


class Foo(models.Model):
    bar = models.ForeignKey(Bar)

    a_value = models.IntegerField()

    b_value = models.BooleanField()

    def clean(self):
        super(Foo, self).clean()
        if self.b_value and self.a_value > 50:
            raise BadFooError("No good")

接下来，myapp/tests.py：

from unittest import TestCase

from mock import MagicMock

from . import models


class SimpleTest(TestCase):

    def test_avalue_bvalue_validation(self):
        foo = models.Foo()
        foo.a_value = 30
        foo.b_value = True
        foo.bar = MagicMock(spec=models.Bar)
        self.assertRaises(models.BadFooError, foo.full_clean)

    def test_method_2(self):
        foo = models.Foo()
        foo.a_value = 30
        foo.b_value = True
        foo.bar = MagicMock()
        foo.__class__ = models.Bar
        self.assertRaises(models.BadFooError, foo.full_clean)

    def test_method_3(self):
        foo = models.Foo()
        foo.a_value = 30
        foo.b_value = True
        # ignore it and it will go away ...??
        self.assertRaises(models.BadFooError, foo.full_clean)

最后，输出python manage.py test myapp

EEE
======================================================================
ERROR: test_avalue_bvalue_validation (myapp.tests.SimpleTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "~/sandbox/myapp/tests.py", line 14, in test_avalue_bvalue_validation
    foo.bar = MagicMock(spec=models.Bar)
  File "~/dsbx/local/lib/python2.7/site-packages/django/db/models/fields/related.py", line 408, in __set__
    instance._state.db = router.db_for_write(instance.__class__, instance=value)
  File "~/dsbx/local/lib/python2.7/site-packages/django/db/utils.py", line 142, in _route_db
    return hints['instance']._state.db or DEFAULT_DB_ALIAS
  File "~/dsbx/local/lib/python2.7/site-packages/mock.py", line 658, in __getattr__
    raise AttributeError("Mock object has no attribute %r" % name)
AttributeError: Mock object has no attribute '_state'

======================================================================
ERROR: test_method_2 (myapp.tests.SimpleTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "~/sandbox/myapp/tests.py", line 21, in test_method_2
    foo.bar = MagicMock()
  File "~/dsbx/local/lib/python2.7/site-packages/django/db/models/fields/related.py", line 405, in __set__
    self.field.name, self.field.rel.to._meta.object_name))
ValueError: Cannot assign "<MagicMock id='31914832'>": "Foo.bar" must be a "Bar" instance.

======================================================================
ERROR: test_method_3 (myapp.tests.SimpleTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "~/sandbox/myapp/tests.py", line 29, in test_method_3
    self.assertRaises(models.BadFooError, foo.full_clean)
  File "/usr/lib/python2.7/unittest/case.py", line 471, in assertRaises
    callableObj(*args, **kwargs)
  File "~/dsbx/local/lib/python2.7/site-packages/django/db/models/base.py", line 926, in full_clean
    raise ValidationError(errors)
ValidationError: {'bar': [u'This field cannot be null.']}

----------------------------------------------------------------------
Ran 3 tests in 0.003s

FAILED (errors=3)
Creating test database for alias 'default'...
Destroying test database for alias 'default'...

所以我的问题是... 怎么办？

4

2 回答 2

4

在我的单元测试中，我只是分配_state给一个新的 Mock 实例，就像对您的第一个示例单元测试的这个小改动一样：

def test_avalue_bvalue_validation(self):
    foo = models.Foo()
    foo.a_value = 30
    foo.b_value = True
    bar = Mock(spec=models.Bar)
    bar._state = Mock()
    foo.bar = bar
    self.assertRaises(models.BadFooError, foo.full_clean)

但是，为了将您的验证作为黑盒进行测试，我会将验证代码拉入您模型上的一个单独方法中，我将从该方法内部调用该clean()方法。然后，您可以专门对该验证代码进行单元测试。您仍然需要执行_stage = Mock()分配，以便创建 Foo 实例，但至少您将最小化对 Django 的调用。

于 2013-08-22T15:21:44.767 回答
0

好吧，现在我只是将我的断言切换为self.assertRaises(models.BadFooError, foo.clean)（不同foo.clean之处在于foo.full_clean）。虽然这可行，但它似乎并不理想。我想将验证作为黑盒进行测试。

于 2013-05-31T14:11:06.730 回答