import pytest
from pathlib import Path
from unittest.mock import MagicMock, patch
from setup.components.mcp import MCPComponent


class TestMCPComponent:
    @patch("setup.components.mcp.MCPComponent._post_install", return_value=True)
    @patch(
        "setup.components.mcp.MCPComponent.validate_prerequisites",
        return_value=(True, []),
    )
    @patch("setup.components.mcp.MCPComponent._install_mcp_server")
    def test_install_selected_servers_only(
        self, mock_install_mcp_server, mock_validate_prereqs, mock_post_install
    ):
        mock_install_mcp_server.return_value = True

        component = MCPComponent(install_dir=Path("/fake/dir"))
        component.installed_servers_in_session = []

        # Simulate selecting only the 'magic' server
        config = {"selected_mcp_servers": ["magic"]}

        success = component._install(config)

        assert success is True
        assert component.installed_servers_in_session == ["magic"]

        # Assert that _install_mcp_server was called exactly once
        assert mock_install_mcp_server.call_count == 1

        # Assert that it was called with the correct server info
        called_args, _ = mock_install_mcp_server.call_args
        server_info_arg = called_args[0]

        assert server_info_arg["name"] == "magic"
        assert server_info_arg["npm_package"] == "@21st-dev/magic"

    @patch("subprocess.run")
    def test_validate_installation_success(self, mock_subprocess_run):
        component = MCPComponent(install_dir=Path("/fake/dir"))

        # Mock settings manager
        component.settings_manager = MagicMock()
        component.settings_manager.is_component_installed.return_value = True
        component.settings_manager.get_component_version.return_value = (
            component.get_metadata()["version"]
        )
        component.settings_manager.get_metadata_setting.return_value = [
            "magic",
            "playwright",
        ]

        # Mock `claude mcp list` output
        mock_subprocess_run.return_value.returncode = 0
        mock_subprocess_run.return_value.stdout = "magic\nplaywright\n"

        success, errors = component.validate_installation()

        assert success is True
        assert not errors

    @patch("subprocess.run")
    def test_validate_installation_failure(self, mock_subprocess_run):
        component = MCPComponent(install_dir=Path("/fake/dir"))

        # Mock settings manager
        component.settings_manager = MagicMock()
        component.settings_manager.is_component_installed.return_value = True
        component.settings_manager.get_component_version.return_value = (
            component.get_metadata()["version"]
        )
        component.settings_manager.get_metadata_setting.return_value = [
            "magic",
            "playwright",
        ]

        # Mock `claude mcp list` output - 'playwright' is missing
        mock_subprocess_run.return_value.returncode = 0
        mock_subprocess_run.return_value.stdout = "magic\n"

        success, errors = component.validate_installation()

        assert success is False
        assert len(errors) == 1
        assert "playwright" in errors[0]