fix(test): mock environment variables for callback validation test

[jquinter]

Summary

Fixes test_proxy_config_state_post_init_callback_call failure when running with pytest-xdist parallel execution (--dist=loadscope).

Problem

The test was failing with Pydantic validation error:

ValidationError: 2 validation errors for TeamCallbackMetadata
callback_vars.langfuse_public_key
  Input should be a valid string [type=string_type, input_value=None, input_type=NoneType]
callback_vars.langfuse_secret
  Input should be a valid string [type=string_type, input_value=None, input_type=NoneType]

Root cause: The test configures callback settings with environment variable references:

"langfuse_public_key": "os.environ/LANGFUSE_PUBLIC_KEY",
"langfuse_secret": "os.environ/LANGFUSE_SECRET_KEY",

These get resolved at runtime. In parallel execution:

• Different worker processes may have different environment variables
• Required env vars (LANGFUSE_PUBLIC_KEY, LANGFUSE_SECRET_KEY) may not be set
• Resolution returns None instead of a string
• Pydantic validation fails expecting string type

Solution

Use monkeypatch to set the required environment variables before test execution:

def test_proxy_config_state_post_init_callback_call(monkeypatch):
    monkeypatch.setenv("LANGFUSE_PUBLIC_KEY", "test_public_key")
    monkeypatch.setenv("LANGFUSE_SECRET_KEY", "test_secret_key")
    # ... rest of test

This ensures:

• Consistent behavior across all execution environments (local, CI, parallel workers)
• Environment variable resolution always produces valid strings
• Pydantic validation passes
• No pollution between tests (monkeypatch auto-cleans up)

Testing

• Tested locally: pytest tests/proxy_unit_tests/test_proxy_utils.py::test_proxy_config_state_post_init_callback_call -v
• Verified Pydantic validation passes with mocked env vars

Related

• Fixes test failure exposed by PR improve(ci): enhance test stability with better isolation and distribution #21277 (CI improvements with better test isolation)
• Not caused by PR improve(ci): enhance test stability with better isolation and distribution #21277, but exposed by parallel execution environment differences

🤖 Generated with Claude Code

[vercel]

The latest updates on your projects. Learn more about Vercel for GitHub.

Project	Deployment	Actions	Updated (UTC)
litellm	Ready	Preview, Comment	Feb 15, 2026 11:46pm

[greptile-apps]

Greptile Summary

This PR fixes a test isolation issue in test_proxy_config_state_post_init_callback_call that surfaces during parallel test execution (e.g., pytest-xdist). The test configures team callback settings with os.environ/LANGFUSE_PUBLIC_KEY and os.environ/LANGFUSE_SECRET_KEY references, which get resolved by ProxyConfig._get_team_config() via get_secret(). When these environment variables are not set, get_secret() returns None, and the subsequent TeamCallbackMetadata Pydantic model validation fails because callback_vars expects Dict[str, str].

• Adds monkeypatch.setenv() calls for LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY to ensure consistent test behavior across all execution environments
• Uses pytest's monkeypatch fixture for proper cleanup, avoiding env var leakage between tests

Confidence Score: 5/5

• This PR is safe to merge — it only modifies a test file by adding environment variable mocking for test isolation.
• The change is minimal (adding 2 monkeypatch.setenv calls and updating the function signature), correctly addresses a real test isolation problem, and follows pytest best practices. No production code is modified.
• No files require special attention.

Important Files Changed

Filename	Overview
tests/proxy_unit_tests/test_proxy_utils.py	Adds monkeypatch fixture to mock LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY env vars, preventing Pydantic validation errors when running in parallel execution environments.

Sequence Diagram

sequenceDiagram
    participant Test as Test Function
    participant MP as monkeypatch
    participant PC as ProxyConfig
    participant GTC as _get_team_config
    participant GS as get_secret()
    participant ENV as os.environ
    participant TCM as TeamCallbackMetadata

    Test->>MP: setenv("LANGFUSE_PUBLIC_KEY", "test_public_key")
    Test->>MP: setenv("LANGFUSE_SECRET_KEY", "test_secret_key")
    MP->>ENV: Set env vars
    Test->>PC: update_config_state(config)
    Test->>PC: load_team_config("test")
    PC->>GTC: _get_team_config("test", all_teams_config)
    GTC->>GS: get_secret("os.environ/LANGFUSE_PUBLIC_KEY")
    GS->>ENV: os.environ.get("LANGFUSE_PUBLIC_KEY")
    ENV-->>GS: "test_public_key"
    GS-->>GTC: "test_public_key"
    GTC->>GS: get_secret("os.environ/LANGFUSE_SECRET_KEY")
    GS->>ENV: os.environ.get("LANGFUSE_SECRET_KEY")
    ENV-->>GS: "test_secret_key"
    GS-->>GTC: "test_secret_key"
    GTC-->>PC: team_config (with resolved values)
    PC-->>Test: team_config
    Test->>TCM: TeamCallbackMetadata(callback_vars=...)
    Note over TCM: Pydantic validates Dict[str, str] ✓

Loading

_{Last reviewed commit: 2d41b03}

[greptile-apps]

_{1 file reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}