Method batch

tests/utils/api_client.py:169–227

Send batch of commands. Args: commands: List of dicts with "command" and optional "params" timeout: Optional timeout override Returns: List of results (may include errors) on success, or error dict if batch rejected Raises:

(
        self, commands: list[dict], timeout: Optional[float] = None
    )

                    )

    def batch(
        self, commands: list[dict], timeout: Optional[float] = None
    ) -> list[dict] | dict:
        """
        Send batch of commands.

        Args:
            commands: List of dicts with "command" and optional "params"
            timeout: Optional timeout override

        Returns:
            List of results (may include errors) on success, or error dict if batch rejected

        Raises:
            TimeoutError: If response not received in time
        """
        request_id = str(uuid.uuid4())

        batch_commands = []
        for i, cmd in enumerate(commands):
            batch_cmd = {
                "id": f"{request_id}-{i}",
                "command": cmd["command"],
            }
            if "params" in cmd:
                batch_cmd["params"] = cmd["params"]
            batch_commands.append(batch_cmd)

        message = {"type": "batch", "id": request_id, "commands": batch_commands}

        self._send_message(message)

        deadline = time.time() + (timeout or self.timeout)

        while True:
            remaining = deadline - time.time()
            if remaining <= 0:
                raise TimeoutError("Timeout waiting for response")

            response = self._recv_message(remaining)

            if response.get("type") == "response" and response.get("id") == request_id:
                # Batch was processed: return individual results even if some failed.
                # A "results" key means the batch ran and each entry has its own
                # success/failure status (partial failure is normal).
                if "results" in response:
                    return response.get("results", [])

                # No "results" key means the batch was rejected at the protocol
                # level before any commands were executed (e.g. empty batch, size
                # limit exceeded).
                return {
                    "error": True,
                    "success": False,
                    "message": response.get("error", {}).get(
                        "message", "Batch rejected"
                    ),
                    "code": response.get("error", {}).get("code", "UNKNOWN"),