Function _make_ui

tests/chat/test_chat_handler_coverage.py:14–34

Build a mock ChatUIManager.

(
    is_streaming=False,
    tools_running=False,
)

def _make_ui(
    is_streaming=False,
    tools_running=False,
):
    """Build a mock ChatUIManager."""
    ui = MagicMock()
    ui.is_streaming_response = is_streaming
    ui.tools_running = tools_running
    ui.verbose_mode = False
    ui.streaming_handler = MagicMock()
    ui.streaming_handler.interrupt_streaming = MagicMock()

    ui.get_user_input = AsyncMock(return_value="exit")
    ui.handle_command = AsyncMock(return_value=True)
    ui.print_user_message = MagicMock()
    ui.interrupt_streaming = MagicMock()
    ui._interrupt_now = MagicMock()
    ui.stop_streaming_response_sync = MagicMock()
    ui.stop_tool_calls = MagicMock()
    ui.cleanup = MagicMock()
    return ui


def _make_ctx(exit_requested=False):