如何为 python 的 requests.Response.text 属性设置值？答案

【问题标题】：How to set value to python's requests.Response.text property?如何为 python 的 requests.Response.text 属性设置值？
【发布时间】：2019-01-30 17:31:12
【问题描述】：

我想在我的单元测试中模拟 requests.Response() 对象，我从下面的链接中得到了提示。

How to mock data as request.Response type in python

在这里，我可以只设置status_code 的值（不是@Property），我想设置@Property 的值text 或content

类用户名密码AuthStrategyTest(TestCase)：

def test_do_success(self):
    content = self._factory.get_reader(ReaderType.CONTENT).read('MY_TEST')
    auth_strategy = UsernamePasswordAuthStrategy(content)
    # mock send_request method response
    response = Response()
    response.status_code = 200
    # How could I achieve below line? 
    response.text = """<html>
            <body>
                <form method="post" name="NavForm">
                    <input id="csrfKey" name="csrfKey" type="hidden" value="JEK7schtDx5IVcH1eOWKN9lFH7ptcwHD/"/>
                </form>
            </body>
        </html>"""
    auth_strategy.send_request = mock.MagicMock(return_value=response)
    session, auth_result = auth_strategy.do()  # {'for_next_url_params': {'csrfKey': 'T4WNcz+hXqrxVa5R9o2HXkDm8pNZEi4k/'}}
    self.assertTrue(session, 'Test Failed! Something went wrong')
    self.assertTrue('for_next_url_params' in auth_result and 'csrfKey' in auth_result['for_next_url_params'],
                    'Test Failed! csrfKey not found')

send_request 返回response

【问题讨论】：

我们需要你的一些代码来解决问题。
@eagle33322 提供的链接有足够的信息，所以之前没有添加代码。您也可以在提供的链接中举一个例子
How to mock data as request.Response type in python的可能重复
@eagle33322 不是真的，该示例只是说明如何为status_code 设置值而不是@Property 以及我在描述中使用的相同链接

标签： python-3.x python-requests

【解决方案1】：

我已经浏览了python文档并弄清楚了......

解决方案是->

def test_do_success(self):
        content = self._factory.get_reader(ReaderType.CONTENT).read('MY_TEST')
        auth_strategy = UsernamePasswordAuthStrategy(content)
        # mock send_request method response
        response = Response()
        response.status_code = 200
        my_text = """<html>
                <body>
                    <form method="post" name="NavForm">
                        <input id="csrfKey" name="csrfKey" type="hidden" value="JEK7schtDx5IVcH1eOWKN9lFH7ptcwHD/"/>
                    </form>
                </body>
            </html>
        """
        type(response).text = mock.PropertyMock(return_value=my_text)
        auth_strategy.send_request = mock.MagicMock(return_value=response)
        session, auth_result = auth_strategy.do()
        self.assertTrue(session, 'Test Failed! Something went wrong')
        self.assertTrue('JEK7schtDx5IVcH1eOWKN9lFH7ptcwHD' in auth_result['for_next_url_params']['csrfKey'],
                        'Test Failed! csrfKey not found')

我在文本周围添加了PropertyMock，代码更改是-->

type(response).text = mock.PropertyMock(return_value=my_text)

【讨论】：

不错的答案。我想知道它是如何/为什么起作用的；您能否将其添加到您的答案或您提到的文档的链接中？
此解决方案为requests.Response 类重新定义text 属性。可以通过以下方式检查：import requests as r; a = r.request('POST', 'https://www.google.com'); type(a).text = '{"k":"v"}'; a = r.request('POST', 'https://www.google.com'); print(a.text)。即使在 a 将被重新分配之后，在重新启动过程之前，文本将始终为 '{"k":"v"}'。

【解决方案2】：

text的值可以通过使用私有属性_content来设置（应该是bytes）：

import requests as r

res = r.Response()
res.status_code = 404

utf_string = '{"error": "page not found", "check UTF8 characters": "あア"}'
bytes_string = utf_string.encode('utf-8')
res._content = bytes_string

print('Status code:', res.status_code)
print('RAW:', f'[{type(res.content)}]', res.content)
print('Private RAW:', f'[{type(res._content)}]', res._content)
print('Decoded text:', f'[{type(res.text)}]', res.text)
print('JSON converted:', f'[{type(res.json())}]', res.json())

输出：

Status code: 404
RAW: [<class 'bytes'>] b'{"error": "page not found", "check UTF8 characters": "\xe3\x81\x82\xe3\x82\xa2"}'
Private RAW: [<class 'bytes'>] b'{"error": "page not found", "check UTF8 characters": "\xe3\x81\x82\xe3\x82\xa2"}'
Decoded text: [<class 'str'>] {"error": "page not found", "check UTF8 characters": "あア"}
JSON converted: [<class 'dict'>] {'error': 'page not found', 'check UTF8 characters': 'あア'}

【讨论】：

这是一个非常有用的答案。我认为它实际上优于公认的答案，因为它看起来更简单。您可能会考虑突出显示的一条关键信息（因为它在代码视图中滚动离开页面）是第 4 代码行上的 .encode('utf-8')。也许将字符串分配给不同行上的变量，然后将 _content 分配给该变量。然后编码调用将更加明显。只是一个想法。我第一次看你的答案时错过了。
这是手动将预加载的.text 推送到响应以进行内部测试的一种快速而肮脏的方法，而不是加载完整的 API 调用并等待 5 分钟让您的服务器下载数据。