Function test_process_options_with_servers

tests/config/test_cli_options.py:155–196

(mock_discovery, dummy_config_file, monkeypatch)

@patch("mcp_cli.config.cli_options.trigger_discovery_after_setup")
def test_process_options_with_servers(mock_discovery, dummy_config_file, monkeypatch):
    # Mock discovery to avoid actual network calls
    mock_discovery.return_value = 0

    # Prepare inputs.
    # server: a comma-separated string.
    server_input = "Server1, Server2"
    disable_filesystem = False
    provider = "openai"
    model = "custom-model"

    # Clear any preexisting environment variables for a clean test.
    monkeypatch.delenv("LLM_PROVIDER", raising=False)
    monkeypatch.delenv("LLM_MODEL", raising=False)
    monkeypatch.delenv("SOURCE_FILESYSTEMS", raising=False)

    servers_list, user_specified, server_names = process_options(
        server=server_input,
        disable_filesystem=disable_filesystem,
        provider=provider,
        model=model,
        config_file=dummy_config_file,
    )

    # Check that server list and user_specified were parsed correctly.
    assert servers_list == ["Server1", "Server2"]
    assert user_specified == ["Server1", "Server2"]

    # In the dummy config, the keys are "Server1" and "Server2". Because the user specified
    # these names, extract_server_names should only include those that match.
    # Mapping is based on order as encountered from the specified servers.
    expected_mapping = {0: "Server1", 1: "Server2"}
    assert server_names == expected_mapping

    # Check environment variables.
    assert os.environ["LLM_PROVIDER"] == provider
    assert os.environ["LLM_MODEL"] == model

    # Since disable_filesystem is False, SOURCE_FILESYSTEMS should be set.
    source_fs = json.loads(os.environ["SOURCE_FILESYSTEMS"])
    # For testing, we expect at least the current working directory.
    assert os.getcwd() in source_fs


@patch("mcp_cli.config.cli_options.trigger_discovery_after_setup")