python - 如何在 pytest requests mock 中伪造我的回复

Question

我正在尝试测试的功能正在list返回strings：

def listForumsIds:
    response = requests.get(url)
    forums= response.json().get('forums')
    forumsIds= [forum['documentId'] for forum in forums]
    # return like: ['id1', 'id2', 'id3'.....]
    return forumsIds

我的测试功能：

@requests_mock.mock()
def test_forms(self, m):
    # I also used json='response'
    m.get('valid url', text="response", status_code=200)
    resp = listForumsIds('valid url')
    # ERROR !!!!
    assert resp == "response"

我收到如下错误：json.decoder.JSONDecodeError或str object has no attribute get

如何伪造我的响应以匹配我的函数的返回值？

Answer 1

json您必须在模拟响应的字段中传递所需的有效负载。示例，适用于您的代码：

class MyTests(unittest.TestCase):

    @requests_mock.mock()
    def test_forms(self, m):
        payload = {"forums": [{"documentId": "id1"}]}
        m.register_uri("GET", "https://www.example.com", json=payload)
        ids = listForumsIds('https://www.example.com')
        assert ids == ['id1']