Track history of request, responses in httpx

httpx_history.py

import httpx
import pytest

from typing import NamedTuple, Dequeue
from unittest import mock


class Transaction(NamedTuple):
    request: httpx.Request
    response: httpx.Response


class ResponseHistory(Deque[Transaction]):
    pass


@pytest.fixture()
def response_history() -> ResponseHistory:
    http_send = httpx.Client.send
    history = ResponseHistory()

    def wrap_send(client, req, *args, **kwargs):
        resp = http_send(client, req, *args, **kwargs)
        history.append(Transaction(req, resp))
        return resp

    with mock.patch("httpx.Client.send", side_effect=wrap_send, autospec=True):
        yield history


@pytest.fixture(autouse=True)
def assert_all_responses_are_closed(response_history: ResponseHistory):
    yield
    for transaction in response_history:
        assert transaction.response.is_closed