我试图模拟Djangos model.Manager()
类中的链接调用。现在我想嘲笑values()
和filter()
方法。Django ORM - 模拟值()。filter()链
为了测试我创建了一个小测试项目:
- 创建虚拟环境
- 运行
pip install django mock mock-django nose django-nose
- 创建项目
django-admin.py startproject mocktest
- 创建应用
manage.py startapp mockme
- 添加
django_nose
和mocktest.mockme
到INSTALLED_APPS
(settings.py) - 添加
TEST_RUNNER = 'django_nose.NoseTestSuiteRunner'
到的settings.py
,进行验证,一切都设置正确我跑manage.py test
。一个测试运行,Django在您创建应用程序时创建的标准测试。
我做的下一件事是创建一个非常简单的模型。
mockme/models.py
from django.db import models
class MyModel(models.Model):
name = models.CharField(max_length=50)
我做下一件事情就是创建一个使用MyModel
一个简单的函数。这是我想稍后测试的功能。
mockme/functions.py
from models import MyModel
def chained_query():
return MyModel.objects.values('name').filter(name='Frank')
没什么特别发生在这里。该函数正在过滤MyModel对象以查找其中name='Frank'
的所有实例。对values()的调用将返回一个ValuesQuerySet
,它将只包含所有找到的MyModel实例的名称字段。
mockme/tests.py
import mock
from django.test import TestCase
from mocktest.mockme.models import MyModel
from mocktest.mockme.functions import chained_query
from mock_django.query import QuerySetMock
class SimpleTest(TestCase):
def test_chained_query(self):
# without mocked queryset the result should be 0
result = chained_query()
self.assertEquals(result.count(), 0)
# now try to mock values().filter() and reeturn
# one 'Frank'.
qsm = QuerySetMock(MyModel, MyModel(name='Frank'))
with mock.patch('django.db.models.Manager.filter', qsm):
result = chained_query()
self.assertEquals(result.count(), 1)
第一的assertEquals将评估成功。由于模型管理器尚未被模拟,因此没有实例返回。当第二的assertEquals叫我希望result
包含我添加返回值为QuerySetMock的为MyModel实例:
qsm = QuerySetMock(MyModel, MyModel(name='Frank'))
我嘲笑了filter()
方法,而不是values()
方法,因为我觉得这将是最后评估打电话,虽然我不确定。
测试将失败,因为第二个结果变量将不包含任何MyModel实例。
可以肯定的是,filter()
方法真的是嘲笑我加了一个“调试打印”声明:
from django.db import models
print models.Manager.filter
其返回:
<SharedMock name='mock.iterator' id='4514208912'>
我在做什么错?
应该说不是' '值')'? – Splatmistro