import pytest
from unittest.mock import patch, MagicMock
import argparse
from setup.cli.commands.install import get_components_to_install

class TestGetComponents:
    @patch('setup.cli.commands.install.select_mcp_servers')
    def test_get_components_to_install_interactive_mcp(self, mock_select_mcp):
        # Arrange
        mock_registry = MagicMock()
        mock_config_manager = MagicMock()
        mock_config_manager._installation_context = {}
        mock_select_mcp.return_value = ['magic']

        args = argparse.Namespace(components=['mcp'])

        # Act
        components = get_components_to_install(args, mock_registry, mock_config_manager)

        # Assert
        mock_select_mcp.assert_called_once()
        assert 'mcp' in components
        assert 'mcp_docs' in components # Should be added automatically
        assert hasattr(mock_config_manager, '_installation_context')
        assert mock_config_manager._installation_context['selected_mcp_servers'] == ['magic']