模拟类的补丁属性

Question

我试图模拟一个类。

我试图模拟的类如下所示(为简洁起见，已删除行(：

class Connection(object):
"""Connection.
"""
def __init__(self, base_url=None, creds=None, user_agent=None):
self.clients = ClientFactory(self)

如您所见，它有一个名为clients.

我测试的方法：

def _auth(self, credentials):
connection = Connection(base_url=f'https://someurl.com', creds=credentials)
return connection.clients

我的单元测试如下所示：

@patch('connection.Connection.__init__')
def test_deploy(patched_connection, fs):
patched_connection.return_value = None
patched_connection.clients = None
# Do some stuff

问题是...如何在测试中设置clients属性，因为所测试的方法需要设置它？ (我可以将其设置为None，但我只需要能够设置它。

使用当前代码，我的应用程序返回错误：

AttributeError: 'Connection' object has no attribute 'clients'

谢谢！

Answer 1

您可能希望修补Connection类本身，而不是__init__方法：

@patch('connection.Connection')
def test_deploy(patched_connection, fs):
connection_object = MagicMock()
patched_connection.return_value = connection_object
connection_object.clients = None
sut = Auth()  # create the tested object (Auth is a placeholder here)
sut._auth('')  # call the tested function
# test for how `Connection` was constructed 
patched_connection.assert_called_once_with(
base_url='https://someurl.com', creds='')

修补Connection类，并通过设置return_value设置Connection实例模拟。现在，您可以在该实例中设置所需的属性。

请注意，检查__init__调用实际上意味着检查实例创建调用，因此您可以使用Connection模拟。

当然，这是假设您不想测试Connection本身，并且_auth属于另一个类(此处称为Auth(。