14

我试图模拟 Djangosmodel.Manager()类的链式调用。现在我想模拟values()andfilter()方法。

为了测试我创建了一个小测试项目：

  1. 创建虚拟环境
  2. 跑pip install django mock mock-django nose django-nose
  3. 创建项目django-admin.py startproject mocktest
  4. 创建应用manage.py startapp mockme
  5. 添加django_nose和mocktest.mockme到INSTALLED_APPS(settings.py)
  6. 添加TEST_RUNNER = 'django_nose.NoseTestSuiteRunner'到 settings.py

为了验证一切都设置正确，我跑了manage.py test。运行一项测试，这是 Django 在您创建应用程序时创建的标准测试。

接下来我要做的是创建一个非常简单的模型。

模拟/模型.py

from django.db import models

class MyModel(models.Model):
    name = models.CharField(max_length=50)

接下来我做的是创建一个简单的函数，它使用MyModel. 这就是我稍后要测试的功能。

模拟/functions.py

from models import MyModel

def chained_query():
    return MyModel.objects.values('name').filter(name='Frank')

这里没有什么特别的事情发生。该函数正在过滤 MyModel 对象以查找name='Frank'. 对 values() 的调用将返回一个ValuesQuerySet仅包含所有找到的 MyModel 实例的名称字段。

模拟/测试.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'>

我究竟做错了什么？

4

2 回答 2

12

尝试这个：

import mock
from mocktest.mockme.models import MyModel

class SimpleTest(TestCase):
    def test_chained_query(self):
        my_model_value_mock = mock.patch(MyModel.objects, 'value')
        my_model_value_mock.return_value.filter.return_value.count.return_value = 10000
        self.assertTrue(my_model_value_mock.return_value.filter.return_value.count.called)
于 2014-07-25T16:44:22.167 回答
7

@Gin 的回答让我大部分时间都在那里，但在我的情况下，我正在修补MyModel.objects，我正在嘲笑的查询看起来像这样：

MyModel.objects.filter(arg1=user, arg2=something_else).order_by('-something').first()

所以这对我有用：

@patch('MyModel.objects')
def test_a_function(mock, a_fixture):
    mock.filter.return_value.order_by.return_value.first.return_value = a_fixture
    result = the_func_im_testing(arg1, arg2)
    assert result == 'value'

此外，修补属性的顺序很重要，并且必须与您在测试函数中调用它们的顺序相匹配。

于 2019-05-18T16:06:18.237 回答