test_settings.py•842 B
import os
from typing import Optional
from unittest.mock import patch
import pytest
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