from pathlib import Path
from unittest import mock

import pytest
from click.testing import CliRunner

from mlflow.assistant.skill_installer import BundledSkill
from mlflow.cli.skills import commands


@pytest.fixture
def runner():
    return CliRunner()


@pytest.fixture
def mock_bundled_skills():
    with mock.patch("mlflow.cli.skills.list_bundled_skills", return_value=[]) as m:
        yield m


def test_list_command_renders_skills(runner, mock_bundled_skills):
    skills = [
        BundledSkill(name="alpha-skill", description="Does alpha things.", path=Path("/s/alpha")),
        BundledSkill(name="beta-skill", description="", path=Path("/s/beta")),
    ]
    mock_bundled_skills.return_value = skills

    result = runner.invoke(commands, ["list"])

    assert result.exit_code == 0
    assert "alpha-skill" in result.output
    assert "Does alpha things." in result.output
    assert "beta-skill" in result.output
    assert str(Path("/s/alpha")) in result.output
    assert str(Path("/s/beta")) in result.output


def test_list_command_handles_no_skills(runner, mock_bundled_skills):
    result = runner.invoke(commands, ["list"])

    assert result.exit_code == 0
    assert "No MLflow skills found" in result.output
    assert "git submodule update --init" in result.output


def test_view_command_renders_skill(runner, mock_bundled_skills):
    skills = [
        BundledSkill(name="alpha-skill", description="Does alpha things.", path=Path("/s/alpha")),
        BundledSkill(name="beta-skill", description="Does beta things.", path=Path("/s/beta")),
    ]
    mock_bundled_skills.return_value = skills

    result = runner.invoke(commands, ["view", "alpha-skill"])

    assert result.exit_code == 0
    assert "alpha-skill" in result.output
    assert "Does alpha things." in result.output
    assert str(Path("/s/alpha")) in result.output
    assert "beta-skill" not in result.output


def test_view_command_skill_not_found(runner, mock_bundled_skills):
    result = runner.invoke(commands, ["view", "missing-skill"])

    assert result.exit_code != 0
    assert "Skill missing-skill not found." in result.output


def test_view_command_skill_without_description(runner, mock_bundled_skills):
    skills = [BundledSkill(name="alpha-skill", description="", path=Path("/s/alpha"))]
    mock_bundled_skills.return_value = skills

    result = runner.invoke(commands, ["view", "alpha-skill"])

    assert result.exit_code == 0
    assert "alpha-skill" in result.output
    assert str(Path("/s/alpha")) in result.output
    assert "\n  \n" not in result.output