Python unittest模拟:是否可以在测试时模拟方法的默认参数的值?
-
21-12-2019 - |
题
我有一个接受默认参数的方法:
def build_url(endpoint, host=settings.DEFAULT_HOST):
return '{}{}'.format(host, endpoint)
我有一个练习这种方法的测试用例:
class BuildUrlTestCase(TestCase):
def test_build_url(self):
""" If host and endpoint are supplied result should be 'host/endpoint' """
result = build_url('/end', 'host')
expected = 'host/end'
self.assertEqual(result,expected)
@patch('myapp.settings')
def test_build_url_with_default(self, mock_settings):
""" If only endpoint is supplied should default to settings"""
mock_settings.DEFAULT_HOST = 'domain'
result = build_url('/end')
expected = 'domain/end'
self.assertEqual(result,expected)
如果我把一个调试点放入 build_url
并检查此属性 settings.DEFAULT_HOST
返回模拟值。然而,测试继续失败,断言表明 host
被分配的值从我的实际 settings.py
.我知道这是因为 host
关键字参数在导入时设置,不考虑我的模拟。
调试器
(Pdb) settings
<MagicMock name='settings' id='85761744'>
(Pdb) settings.DEFAULT_HOST
'domain'
(Pdb) host
'host-from-settings.com'
有没有一种方法可以在测试时复盖此值,以便我可以使用默认路径 settings
对象?
解决方案
函数将其参数默认值存储在 func_defaults
定义函数时的属性,因此您可以修补它。类似的东西
def test_build_url(self):
""" If only endpoint is supplied should default to settings"""
# Use `func_defaults` in Python2.x and `__defaults__` in Python3.x.
with patch.object(build_url, 'func_defaults', ('domain',)):
result = build_url('/end')
expected = 'domain/end'
self.assertEqual(result,expected)
我用 patch.object
作为上下文管理器而不是装饰器,以避免不必要的补丁对象作为参数传递给 test_build_url
.
其他提示
我将另一个答案应用于这个问题,但是在上下文管理器之后,修补的功能与以前不同。
我的修补功能看起来像这样:
def f(foo=True):
pass
在我的测试中,我这样做了:
with patch.object(f, 'func_defaults', (False,)):
打电话时 f
在(不在)上下文管理器之后,默认值完全消失,而不是回到以前的值。打电话来 f
没有参数给出了错误 TypeError: f() takes exactly 1 argument (0 given)
相反,我只是在我的测试之前做了这个:
f.func_defaults = (False,)
经过我的测试:
f.func_defaults = (True,)
另一种方法来做到这一点:使用functools。部分提供您想要的"默认"args。这不是 技术上讲 和复盖它们一样;call-ee看到一个明确的arg,但call-er不必提供它。这在大多数时候都足够接近,并且在上下文管理器退出后它做了正确的事情:
# mymodule.py
def myfunction(arg=17):
return arg
# test_mymodule.py
from functools import partial
from mock import patch
import mymodule
class TestMyModule(TestCase):
def test_myfunc(self):
patched = partial(mymodule.myfunction, arg=23)
with patch('mymodule.myfunction', patched):
self.assertEqual(23, mymodule.myfunction()) # Passes; default overridden
self.assertEqual(17, mymodule.myfunction()) # Also passes; original default restored
我在测试时使用它来复盖默认的配置文件位置。信用在适当的时候,我从Danilo Bargen那里得到了这个想法 这里