[英]How do you assert that Django's QuerySet.count() method was called on a Mock?
我正在使用Mock替换Django模型,其配置如下:
>>> config = {'objects.filter.return_value.count.return_value': 1}
>>> MockModel = Mock(**config)
我可以查询一些值,然后在结果上调用.count()
:
>>> # This works as expected
>>> MockModel.objects.filter(foo='bar').count()
1
但是,当我断言这些方法被调用时,Mock引发TypeError
因为count
是Mock call
的方法:
>>> MockModel.assert_has_calls([
call.objects.filter(foo='bar'),
call.objects.filter().count(),
])
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-78-bf4e954d856f> in <module>()
1 MockModel.assert_has_calls([
2 call.objects.filter(foo='bar'),
----> 3 call.objects.filter().count()
4 ])
TypeError: count() takes exactly one argument (0 given)
不过,您可以看到该调用确实存在于mock_calls
:
>>> MockModel.mock_calls
[
call.trait_names(),
call._getAttributeNames(),
call.objects.filter(foo='bar'),
call.objects.filter().count(), # How do you test for this?
...
]
所以我的问题是:您如何断言在模拟Django QuerySet上调用了count()
?
call
也是tuple
,因此可以与tuple
进行比较:
>>> MockModel.assert_has_calls([
call.objects.filter(foo='bar'),
('objects.filter().count',),
])
您可以确认以下内容:
>>> map(tuple, MockModel.mock_calls)
> [('objects.filter', (), {'foo': 'bar'}), ('objects.filter().count', (), {})]
来源: https : //code.google.com/p/mock/source/browse/mock.py#1988
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.