Class SerialStudioClient

tests/utils/api_client.py:26–804

High-level client for Serial Studio API. Provides clean interface for testing without worrying about low-level socket communication details.

class SerialStudioClient:
    """
    High-level client for Serial Studio API.

    Provides clean interface for testing without worrying about
    low-level socket communication details.
    """

    DEFAULT_HOST = "127.0.0.1"
    DEFAULT_PORT = 7777
    TIMEOUT = 5.0

    def __init__(
        self,
        host: str = DEFAULT_HOST,
        port: int = DEFAULT_PORT,
        timeout: float = TIMEOUT,
    ):
        self.host = host
        self.port = port
        self.timeout = timeout
        self._socket: Optional[socket.socket] = None
        self._buffer = b""

    def __enter__(self):
        self.connect()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.disconnect()

    def connect(self) -> None:
        """Establish connection to Serial Studio API."""
        if self._socket:
            return

        self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self._socket.settimeout(self.timeout)
        try:
            self._socket.connect((self.host, self.port))
        except (ConnectionRefusedError, socket.timeout) as e:
            raise ConnectionError(
                f"Could not connect to Serial Studio at {self.host}:{self.port}. "
                f"Ensure Serial Studio is running with API Server enabled."
            ) from e

    def disconnect(self) -> None:
        """Close the connection."""
        if self._socket:
            try:
                self._socket.close()
            except Exception:
                pass
            self._socket = None
        self._buffer = b""

    def _recv_message(self, timeout: Optional[float] = None) -> dict:
        """Receive a single JSON message from the socket."""