使用 unittest 模块修补时，是否可以将 side_effect 分配给 python 中的类成员变量？

Question

是否可以修补类实例变量并强制它在每次引用时返回不同的值？ 具体来说，我有兴趣使用side_effect参数执行此操作

• 我知道在修补方法时，可以将side_effect分配给模拟方法。 如果您将side_effect设置为一个列表，它将在每次调用时遍历该列表并返回一个不同的值。
• 我想对类实例变量做同样的事情，但无法让它工作，我看不到任何文档来建议这是否可行

例子

from unittest.mock import patch

def run_test():
    myClass = MyClass()
    for i in range(2):
        print(myClass.member_variable)

class MyClass():
    def __init__(self):
        self.member_variable = None

@patch('test_me.MyClass.member_variable',side_effect=[1,2], create=True)
def test_stuff(my_mock):
    run_test()
    assert False

输出

-------------- Captured stdout call ---------------------------------------------------------------------------------------------------------------------
None
None

期望输出

-------------- Captured stdout call ---------------------------------------------------------------------------------------------------------------------
1
2

• 要清楚 - 我知道我可以将member_variable包装在get_member_variable method() 中。 那不是我的问题。 我只想知道您是否可以使用 side_effect 修补成员变量。

Answer 1

side_effect可以是函数、可迭代对象或异常（ https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.side_effect ）。 我认为这就是它不起作用的原因。

另一种测试方法是：

>>> class Class:
...     member_variable = None
...
>>> with patch('__main__.Class') as MockClass:
...     instance = MockClass.return_value
...     instance.member_variable = 'foo'
...     assert Class() is instance
...     assert Class().member_variable == 'foo'
...

这是文档： https : //docs.python.org/3/library/unittest.mock.html#unittest.mock.patch

Answer 2

在您设置的示例的情况下，我无法按照我的想法进行更改，您可能在本课程中有更多项目，并且这个想法可以帮助您。

不是对属性产生副作用的最佳选择，但它可以按我的需要工作。

PS：我最终将让我回答您的问题的代码作为示例。

例子：

# -*- coding: utf-8 -*-
# !/usr/bin/env python3

import requests

from src.metaclasses.singleton import Singleton
from src.services.logger import new_logger
from src.exceptions.too_many_retries import TooManyRetries
from src.exceptions.unavailable_url import UnavailableURL

LOG = new_logger(__name__)


class PostbackService(metaclass=Singleton):
    def __init__(self):
        self.session = requests.session()

    def make_request(self, method, url, headers, data=None, retry=0):
        r = self.session.request(method, url, data=data, headers=headers)

        if r.status_code != 200:
            if retry < 3:
                return self.make_request(method, url, headers, data, retry + 1)
            message = f"Error performing request for url: {url}"
            LOG.error(message)
            raise TooManyRetries(message)

        return r.json()

测试：

# -*- coding: utf-8 -*-
# !/usr/bin/env python3
from unittest import TestCase
from unittest.mock import patch, MagicMock

from src.services.postback import PostbackService
from src.exceptions.too_many_retries import TooManyRetries
from src.exceptions.unavailable_url import UnavailableURL


class TestPostbackService(TestCase):
    @patch("src.services.postback.requests")
    def setUp(self, mock_requests) -> None:
        mock_requests.return_value = MagicMock()
        self.pb = PostbackService()

    def test_make_request(self):
        self.pb.session.request.return_value = MagicMock()

        url = "http://mock.io"
        header = {"mock-header": "mock"}
        data = {"mock-data": "mock"}
        mock_json = {"mock-json": "mock"}

        def _def_mock(value):
            """
            Returns a MagicMock with the status code changed for each request, so you can test the retry behavior of the application.
            """
            mock = MagicMock()
            mock.status_code = value
            mock.json.return_value = mock_json
            return mock

        self.pb.session.request.side_effect = [
            _def_mock(403),
            _def_mock(404),
            _def_mock(200),
        ]
        self.assertEqual(self.pb.make_request("GET", url, header, data), mock_json)

        self.pb.session.request.side_effect = [
            _def_mock(403),
            _def_mock(404),
            _def_mock(404),
            _def_mock(404),
        ]
        with self.assertRaises(TooManyRetries):
            self.pb.make_request("GET", url, header, data)

如您所见，我通过将每个的副作用更改为我想要做的来重新创建 magicmock。 它不是漂亮的代码和超级 pythonic，但它按预期工作。 我使用@rsarai 从unittest 文档发送的链接作为创建这个magicmock 对象的基础。