@arizeai/phoenix-mcp

import os
from typing import Optional
from unittest.mock import patch
from urllib.parse import urlparse

import pytest

from phoenix.otel.otel import _construct_http_endpoint
from phoenix.otel.settings import get_env_collector_endpoint


@pytest.mark.parametrize(
    "env,expected",
    [
        ({"PHOENIX_COLLECTOR_ENDPOINT": "http://localhost:6006"}, "http://localhost:6006"),
        ({"OTEL_EXPORTER_OTLP_ENDPOINT": "http://localhost:6006"}, "http://localhost:6006"),
        (
            {
                "PHOENIX_COLLECTOR_ENDPOINT": "http://localhost:6006",
                "OTEL_EXPORTER_OTLP_ENDPOINT": "http://localhost:4318",
            },
            "http://localhost:6006",
        ),
        ({}, None),
    ],
)
def test_get_env_collector_endpoint(env: dict[str, str], expected: Optional[str]) -> None:
    with patch.dict(os.environ, env, clear=True):
        assert get_env_collector_endpoint() == expected


@pytest.mark.parametrize(
    "endpoint, expected",
    [
        ("http://localhost:6006", "http://localhost:6006/v1/traces"),
        ("http://localhost:6006/v1/traces", "http://localhost:6006/v1/traces"),
        ("http://localhost:6006/prefix/v1/traces", "http://localhost:6006/prefix/v1/traces"),
    ],
)
def test_construct_http_endpoint(endpoint: str, expected: str) -> None:
    parsed = urlparse(endpoint)
    result = _construct_http_endpoint(parsed)
    assert result.geturl() == expected