时间戳

首页 > 解决方案 > 如何使用 patch() 模拟返回值的属性和属性

python - 如何使用 patch() 模拟返回值的属性和属性

问题描述

test_client/wclient.py

import json
import requests
client = requests.session()

def setup():
    response = REST_CLIENT.post(
        "https://placeholder.com",
        auth=(placeholder, placeholder),
        data={"grant_type": "client_credentials"},
    )

    status_code = response.status_code
    if status_code in OK_STATUS:
        payload = json.loads(response.content, object_pairs_hook=OrderedDict)
    else:
        payload = response.text
        msg = (
            "Status Code %s" % status_code
        )
        logger.error(msg)
        raise ValueError(msg)

    return payload["access_token"]

测试文件：test_client/test_client.py

import mock
import wclient
@mock.patch("test_client.wclient")
def test_taxes_pitney_bowes_setup_success(resp):
    resp.return_value.post.return_value.status_code = "200"
    wclient.pitney_bowes_setup()

Status Code <MagicMock name='REST_CLIENT.post().status_code' id='4868492200'>

如何使用 mock.patch() 模拟模块的方法和属性？我已经阅读了堆栈溢出帖子的页面，但我对执行魔术模拟的所有不同方式感到困惑。

我试过嘲笑：

resp.return_value.post.return_value.status_code
resp.return_value.post.return_value.status_code.return_value
resp.post.return_value.status_code
resp.post.return_value.status_code.return_value
resp.post.status_code
resp.post.status_code.return_value

标签： pythonunit-testingmockingattributespytest

解决方案

我认为实际进行模拟的方法有很多（请参阅Mocking Method Calls In Python中的许多方法）。我喜欢这样做并容易找到简单模拟的方式是：

对于函数： @patch('module.print', lambda x: None)

对于属性： @patch('module.cwd', os.path.join(os.getcwd(), "folder"))

这篇博文可能对您有所帮助：https ://medium.com/ukey/how-mock-patch-decorator-works-in-python-37acd8b78ae 。

如果您有更多问题，请告诉我。

编辑：要添加多个模拟，只需添加另一个属性：

import wclient
@mock.patch("test_client.wclient")
@mock.patch("another_attribute", "value")
@mock.patch("another_function", lambda x, y: x + y)
def test_taxes_pitney_bowes_setup_success(resp):
    resp.return_value.post.return_value.status_code = "200"
    wclient.pitney_bowes_setup()

推荐阅读