Class ResponseMocker

tests/common.py:425–444

class ResponseMocker:
    calls: list[dict[str, Any]] = []
    responses: dict[str, MockedResponse] = {}

    def add(self, url: str, response: MockedResponse) -> None:
        self.responses[url] = response

    def get(self, url: str, *args, **kwargs) -> MockedResponse:
        data = {"url": url, "args": list(args), "kwargs": kwargs}
        if (request := REQUEST_CONTEXT.get()) is not None:
            data["_test_caller"] = f"{
                request.node.location[0]}::{request.node.name}"
            data["_uses_setup_integration"] = request.node.name != "test_integration_setup" and (
                "setup_integration" in request.fixturenames or "hacs" in request.fixturenames
            )
        self.calls.append(data)
        response = self.responses.get(url, None)
        if response is not None and response.keep:
            return response
        return self.responses.pop(url, None)


class ProxyClientSession(ClientSession):