Class InMemoryTokenStorage

tests/interaction/auth/_harness.py:104–126

A `TokenStorage` that holds tokens and client info as instance attributes. Tests pre-seed `client_info` (via the constructor or by assignment) to drive the pre-registered path, and read both attributes after the flow to assert what the SDK persisted.

class InMemoryTokenStorage:
    """A `TokenStorage` that holds tokens and client info as instance attributes.

    Tests pre-seed `client_info` (via the constructor or by assignment) to drive the
    pre-registered path, and read both attributes after the flow to assert what the SDK
    persisted.
    """

    def __init__(self, *, client_info: OAuthClientInformationFull | None = None) -> None:
        self.tokens: OAuthToken | None = None
        self.client_info: OAuthClientInformationFull | None = client_info

    async def get_tokens(self) -> OAuthToken | None:
        return self.tokens

    async def set_tokens(self, tokens: OAuthToken) -> None:
        self.tokens = tokens

    async def get_client_info(self) -> OAuthClientInformationFull | None:
        return self.client_info

    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
        self.client_info = client_info


class HeadlessOAuth: