Python 使用 MagicMock 在 Pytest 测试中模拟两个函数

Question

这是我要测试的功能。

def service_request(url, session=configure_session()):
    response = session.get(url)

    document = json.loads(response.text)
    return document

是的，它现在是基本的（你可能会争辩说不值得测试），但如果我能让模拟工作，我将能够向它添加更多逻辑并在我进行时测试新添加的内容。我尝试了各种方法来模拟此测试中调用的另外两个函数。一个是 configure_session 函数，另一个是 requests 库中的 get 方法。

configure_session 函数构建 requests.Session 对象，该对象配置了向服务发出请求所需的证书。此方法还可以选择验证与 CA 捆绑包的 SSL 连接。在文件顶部导入的 config 文件（未显示）检索环境变量并按预期工作。

def configure_session():
    session = requests.Session()

    session.cert = (config.client_cert, config.client_key)
    if config.ssl_verify:
        session.verify = config.ca_bundle
    else:
        session.verify = False

    user_agent = f'{config.user_agent_version}'
    session.headers.update({'User-Agent': user_agent,
                     'Accept': 'application/json'})
    return session

这是我试过的最新代码。 load_fixture 方法是一种测试实用程序方法，并且按预期工作。

def test_request():
    response_fixture = load_fixture('example_response.json')

    mock_session = MagicMock()
    mock_get = MagicMock()
    mock_session.get.return_value = mock_get
    mock_get.text.return_value = response_fixture

    test_url = "https://url-for-test/with/the/path/here"

    expected_headers = { 
        'Accept': 'application/json',
        'User-Agent': 'test_user_agent_from_env_vars'
         }

    service_request(test_url, mock_session)

    mock_session.get.assert_called_with(
        test_url,
        headers=expected_headers
        )
    assert response == response_fixture

当前错误为 TypeError: the JSON object must be str, bytes or bytearray, not 'MagicMock'。

Answer 1

当 MagicMock 实例被传递到 json.loads() 时，问题出在 document = json.loads(response.text)。

这就是解决问题的方法。

mock_session = MagicMock()
mock_session.get.return_value.text = response_fixture

Python 使用 MagicMock 在 Pytest 测试中模拟两个函数

Python Mock two functions in Pytest test using MagicMock

testing

mocking

pytest

python-3.x

magicmock