如何在 pytest 请求模拟中伪造我的响应

Question

我要测试的函数是返回strings list ：

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']