Function fake_transform_request

tests/test_upstream_dual_pool.py:126–152

(
        self,
        request,
        excluded_tokens=None,
        excluded_guest_user_ids=None,
    )

def _bind_minimal_request_flow(client: UpstreamClient, captures: list[dict]):
    async def fake_transform_request(
        self,
        request,
        excluded_tokens=None,
        excluded_guest_user_ids=None,
    ):
        auth_info = await self.get_auth_info(
            excluded_tokens=excluded_tokens,
            excluded_guest_user_ids=excluded_guest_user_ids,
        )
        captures.append(dict(auth_info))
        return {
            "url": "https://upstream.test/chat",
            "headers": {
                "x-token": str(auth_info["token"]),
                "x-token-source": str(auth_info["token_source"]),
                "x-guest-user-id": str(auth_info.get("guest_user_id") or ""),
            },
            "body": {"model": request.model},
            "token": auth_info["token"],
            "chat_id": "chat-id",
            "model": request.model,
            "user_id": auth_info["user_id"],
            "auth_mode": auth_info["auth_mode"],
            "token_source": auth_info["token_source"],
            "guest_user_id": auth_info["guest_user_id"],
        }

    async def fake_transform_response(self, response, request, transformed):
        return {