python - Python的类的默认初始化参数

我想模拟类构造函数中的默认参数：

class A (object):
    def __init__(self, connection=DefaultConnection()):
        self.connection = connection

我想在单元测试中模拟DefaultConnection，但是当作为默认值传递时它不起作用。

最佳答案

您可以使用patch来patch模块，然后可以将返回值设置为Mock。

# --- a.py (in package path x.y) --
from c import DefaultConnection

class A (object):
    def __init__(self, connection=DefaultConnection()):
        self.connection = connection

#---- a_test.py ----
from mock import patch
from a import A

@patch('x.y.a.DefaultConnection')
def test(def_conn_mock):
  conn_mock = Mock()
  def_conn_mock.return_value = conn_mock

  a_obj = A()
  ....

关于python - Python的类的默认初始化参数，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/32000970/