我想模拟类构造函数中的默认参数:
class A (object):
def __init__(self, connection=DefaultConnection()):
self.connection = connection
我想在我的单元测试中模拟 DefaultConnection
,但当作为默认值传入时它不起作用。
最佳答案
您可以使用补丁来 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模拟默认初始化参数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32000970/