Question

我有一个接受默认参数的方法：

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对象来运用默认路径？

Answer 1

函数在定义函数时将其参数默认值存储在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。

Answer 2

我对这个问题应用了另一个答案，但是在上下文管理器之后，补丁函数与以前不一样。

我的补丁功能如下所示：

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,)

Answer 3

执行此操作的另一种方法：使用functools.partial提供所需的“默认”参数。从技术上讲，这与覆盖它们不是同一回事。被叫方看到一个明确的arg，但是主叫方不必提供它。在大多数情况下，这已经足够接近了，并且在上下文管理器退出后会执行正确的操作：

# 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 here

那里得到了这个主意。

Python unittest mock：是否可以在测试时模拟方法的默认参数的值？

3 个答案: