import pytest
from pathlib import Path
from unittest.mock import MagicMock, patch
from setup.components.mcp_docs import MCPDocsComponent


class TestMCPDocsComponent:
    @patch(
        "setup.components.mcp_docs.MCPDocsComponent._post_install", return_value=True
    )
    def test_install_calls_post_install_even_if_no_docs(self, mock_post_install):
        component = MCPDocsComponent(install_dir=Path("/fake/dir"))

        # Simulate no servers selected
        config = {"selected_mcp_servers": []}

        success = component._install(config)

        assert success is True
        mock_post_install.assert_called_once()

    @patch(
        "setup.components.mcp_docs.MCPDocsComponent._post_install", return_value=True
    )
    @patch(
        "setup.components.mcp_docs.MCPDocsComponent.get_files_to_install",
        return_value=[],
    )
    @patch("setup.core.base.Component.validate_prerequisites", return_value=(True, []))
    def test_install_calls_post_install_if_docs_not_found(
        self, mock_validate_prereqs, mock_get_files, mock_post_install
    ):
        component = MCPDocsComponent(install_dir=Path("/tmp/fake_dir"))

        # Simulate a server was selected, but the doc file doesn't exist
        config = {"selected_mcp_servers": ["some_server_with_no_doc_file"]}

        success = component._install(config)

        assert success is True
        mock_post_install.assert_called_once()