python - 请求库中的模拟 session

Question

在我的 python 代码中，我有全局 requests.session 实例:

import requests
session = requests.session()

如何使用Mock 来模拟它？这种操作有装饰器吗？我试过以下:

session.get = mock.Mock(side_effect=self.side_effects)

但是(如预期的那样)此代码不会在每次测试后将 session.get 返回到原始状态，就像 @mock.patch 装饰器所做的那样。

Answer 1

由于requests.session()返回的是Session类的一个实例，所以也可以使用patch.object()

from requests import Session
from unittest.mock import patch

@patch.object(Session, 'get')
def test_foo(mock_get):
    mock_get.return_value = 'bar'