Method create_for_testing

src/mcp_cli/chat/testing.py:57–75

Factory for test contexts.

(
        cls,
        stream_manager: Any,
        provider: str | None = None,
        model: str | None = None,
    )

    @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:
        """Test-specific tool initialization."""