如何模拟Django模型查询
作者:互联网
我想嘲笑以下CanonPerson模型
def compute(self, is_send_emails, test_email_address):
cpses = CanonPerson.objects.filter(persons__vpd=6,
persons__country="United States",
persons__role__icontains=';IX;').prefetch_related("persons").using("global")
for cp in cpses:
...
我很失落如何模仿CanonPerson.objects.filter来返回一个可交换的集合,以便我可以继续.
这是我到目前为止所做的:
def test_X_count(self):
with mock.patch('apps.dbank.models.CanonPerson.objects.filter') as canon_patch:
mock_cp = mock.MagicMock(spec=CanonPerson)
mock_person = mock.MagicMock(spec=Person)
mock_person.vpd = 6
mock_cp.country = "United States"
mock_cp.role = ";IX;"
mock_cp.persons.add(mock_person)
canon_patch.objects.filter.return_value = [mock_cp] // ???
oi = OptinInvites()
oi.compute(False, None)
oi.get_most_recent_email.assert_called_once_with(1)
在计算函数中,我可以看到cpses是一个MagicMock类型.
然而,它不是可迭代的,并且之后的for循环,只是跳过它.
我想通过将返回值设置为[mock_cp]我会创建一个可迭代列表?
解决方法:
您为canon_patch指定返回值的行略有错误.你在哪里:
canon_patch.objects.filter.return_value = [mock_cp]
它应该是:
canon_patch.return_value = [mock_cp]
canon_patch已经是’objects.filter’的模拟.
如果您调用CanonPerson.objects.filter.objects.filter(),则原始行将返回[mock_cp].
如果你真的想要修补CanonPerson模型,那么你的补丁线看起来像:
with mock.patch('apps.dbank.models.CanonPerson') as canon_patch:
您可能还会发现需要在正在使用的位置模拟它,而不是从中导入它的位置.因此,假设您在名为my_module的模块中使用CanonPerson,则测试中的补丁代码可能如下所示:
with mock.patch('my_module.CanonPerson') as canon_patch:
标签:python,unit-testing,django,python-mock 来源: https://codeday.me/bug/20190528/1172446.html