# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from unittest.mock import patch

import data_designer.cli.runtime as runtime_mod


def test_ensure_cli_default_model_settings_attempts_default_setup() -> None:
    """CLI bootstrap delegates to default setup when the CLI starts."""
    with (
        patch("data_designer.cli.runtime.print_warning") as mock_print_warning,
        patch("data_designer.cli.runtime.resolve_seed_default_model_settings") as mock_resolve,
    ):
        runtime_mod.ensure_cli_default_model_settings()

    mock_resolve.assert_called_once_with()
    mock_print_warning.assert_not_called()


def test_ensure_cli_default_model_settings_warns_and_continues() -> None:
    """CLI bootstrap prints an actionable warning when setup fails."""
    with (
        patch("data_designer.cli.runtime.print_warning") as mock_print_warning,
        patch(
            "data_designer.cli.runtime.resolve_seed_default_model_settings",
            side_effect=RuntimeError("boom"),
        ) as mock_resolve,
    ):
        runtime_mod.ensure_cli_default_model_settings()

    mock_resolve.assert_called_once_with()
    mock_print_warning.assert_called_once()
    warning = mock_print_warning.call_args[0][0]
    assert "Could not initialize default model providers and model configs automatically." in warning
    assert "The command will continue." in warning
    assert "boom" in warning
    assert "data-designer config providers" in warning
    assert "data-designer config models" in warning