Class TestChatContext

src/mcp_cli/chat/testing.py:20–129

Test-specific ChatContext that works with stream_manager instead of ToolManager. This is used for: 1. The --test-mode flag in chat handler 2. Unit tests that need a mock context

class TestChatContext(ChatContext):
    """
    Test-specific ChatContext that works with stream_manager instead of ToolManager.

    This is used for:
    1. The --test-mode flag in chat handler
    2. Unit tests that need a mock context
    """

    def __init__(self, stream_manager: Any, model_manager: ModelManager):
        """Create test context with stream_manager."""
        # Initialize base attributes without calling super().__init__
        self.tool_manager = None  # type: ignore[assignment]  # Tests don't use ToolManager
        self.stream_manager = stream_manager
        self.model_manager = model_manager

        # Conversation state
        self.exit_requested = False
        self.conversation_history: list = []

        # Context management notices
        self._pending_context_notices: list[str] = []

        # ToolProcessor back-reference
        self.tool_processor: Any = None

        # Tool state
        self.tools: list = []
        self.internal_tools: list = []
        self.server_info: list = []
        self.tool_to_server_map: dict = {}
        self.openai_tools: list = []
        self.tool_name_mapping: dict = {}

        logger.debug(f"TestChatContext created with {self.provider}/{self.model}")

    @classmethod
    def create_for_testing(
        cls,
        stream_manager: Any,
        provider: str | None = None,
        model: str | None = None,
    ) -> "TestChatContext":
        """Factory for test contexts."""
        model_manager = ModelManager()

        if provider and model:
            model_manager.switch_model(provider, model)
        elif provider:
            model_manager.switch_provider(provider)
        elif model:
            # Switch model in current provider
            current_provider = model_manager.get_active_provider()
            model_manager.switch_model(current_provider, model)

        return cls(stream_manager, model_manager)

    async def _initialize_tools(self, on_progress=None) -> None: