import pytest
from unittest.mock import patch, MagicMock
from pyserve.cli import main


def test_cli_version():
    with patch('sys.argv', ['pyserve', '--version']):
        with pytest.raises(SystemExit) as exc_info:
            main()
        assert exc_info.value.code == 0


def test_cli_help():
    with patch('sys.argv', ['pyserve', '--help']):
        with pytest.raises(SystemExit) as exc_info:
            main()
        assert exc_info.value.code == 0


@patch('pyserve.cli.PyServeServer')
@patch('pyserve.cli.Config')
def test_cli_default_config(mock_config, mock_server):
    mock_config_instance = MagicMock()
    mock_config.return_value = mock_config_instance
    mock_server_instance = MagicMock()
    mock_server.return_value = mock_server_instance
    
    with patch('sys.argv', ['pyserve']):
        with patch('pathlib.Path.exists', return_value=False):
            main()
    
    mock_config.assert_called_once()
    mock_server.assert_called_once_with(mock_config_instance)
    mock_server_instance.run.assert_called_once()


@patch('pyserve.cli.PyServeServer')
@patch('pyserve.cli.Config')
def test_cli_custom_host_port(mock_config, mock_server):
    mock_config_instance = MagicMock()
    mock_config.return_value = mock_config_instance
    mock_server_instance = MagicMock()
    mock_server.return_value = mock_server_instance
    
    with patch('sys.argv', ['pyserve', '--host', '127.0.0.1', '--port', '9000']):
        with patch('pathlib.Path.exists', return_value=False):
            main()
    
    assert mock_config_instance.server.host == '127.0.0.1'
    assert mock_config_instance.server.port == 9000