Question

类定义如下-

class User:
    def __init__(self):
            self.__client = ManagerClient()
            self.__data = None
    
    def get_data(self):
        if self.__data is None:
            self.__get_password()
        return self.__data

    def __get_password(self):
            self.__password = self.__client.fetch_secret()
        


class ManagerClient:
     def fetch_secret():
         pass

我想对get_data()进行单元测试，因为User类的构造函数正在初始化manager客户端的对象，而该对象稍后用于调用fetch_secret()，我应该如何模拟？

Answer 1

从技术上讲，在为单元测试测试 get_data 方法时，我们并不关心 __get_password 或其实现。我们只是想测试 get_data 的各种逻辑分支。按照这种思路，我们将简单地模拟 __get_password 及其操作。如果我们想测试 __get_password，我们会为此使用单独的测试。

另一件需要注意的事情是因为您对属性和方法使用双下划线，它们在文本中被 _classname__attribute 替换。这就是为什么我们使用如下所示的措辞来修补和模拟的原因。你可以看到更多关于here的信息。

另外作为旁注，您的 get_data 函数从根本上是不健全的。根据其逻辑，它有两种不同的返回类型，它可以返回某些内容或在对象上设置属性，这被认为是不好的做法。

鉴于以下项目结构：

stackoverflow/
├── mypackage
│   ├── __init__.py
│   └── users.py
└── tests
    ├── __init__.py
    └── test_users.py

我们将编写的测试如下所示。

from unittest.mock import MagicMock, patch

from mypackage.users import User


@patch("mypackage.users.ManagerClient")
def test_get_data_valid(mock_client):
    expected = "fizz"
    klass = User()

    klass._User__data = expected
    actual = klass.get_data()
    assert actual == expected


@patch("mypackage.users.ManagerClient")
@patch.object(User, "_User__get_password")
def test_get_data_none(mock_pwd, mock_client):
    klass = User()
    klass.get_data()

    mock_pwd.assert_called_once()


@patch("mypackage.users.ManagerClient")
def test_get_password(mock_client):
    mock_client.configure_mock(
        **{
            "fetch_secret.return_value": "buzz",
            "return_value": mock_client
        }
    )

    klass = User()
    klass._User__get_password()

    assert klass._User__password == "buzz"

======================================= test session starts ========================================
platform darwin -- Python 3.9.1, pytest-6.2.2, py-1.10.0, pluggy-0.13.1
rootdir: ***/stackoverflow
collected 3 items                                                                                  

tests/test_users.py ...                                                                      [100%]

======================================== 3 passed in 0.06s =========================================

根据要求，我们在每次测试中模拟 ManagerClient，这样它实际上不会被初始化。这是通过 patching 相关类和方法实现的。我们还通过为 __get_password 设置一个我们可以测试的返回值来测试 fetch_secret 方法。我们在最终测试中让 mock_client 返回自身的原因是当您初始化一个 Mock 对象时，它返回一个新对象。为简单起见，我们只需将 Mock 的返回值设置为自身，这样我们就不必创建新对象。

如何模拟构造函数pytest

1 个答案: