2

我正在尝试模拟一堂课。

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

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'

谢谢！

4

1 回答 1

2

您可能想要修补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）。

于 2020-06-26T08:58:44.700 回答