test(router): scope FAKE_KEY via pytest monkeypatch fixture

This commit is contained in:
Md.Nahin Alam
2026-04-03 04:18:20 +06:00
parent 43deb49c2c
commit 184ec250fd

View File

@@ -7,13 +7,17 @@ Run: pytest test_smart_router.py -v
import pytest
import asyncio
import os
from unittest.mock import AsyncMock, MagicMock, patch
from smart_router import SmartRouter, Provider
# ── Fixtures ──────────────────────────────────────────────────────────────────
@pytest.fixture(autouse=True)
def fake_api_key(monkeypatch):
monkeypatch.setenv("FAKE_KEY", "test-key")
def make_provider(name, healthy=True, configured=True,
latency=100.0, cost=0.002, errors=0, requests=0):
p = Provider(
@@ -28,9 +32,7 @@ def make_provider(name, healthy=True, configured=True,
p.avg_latency_ms = latency
p.error_count = errors
p.request_count = requests
if configured:
os.environ.setdefault("FAKE_KEY", "test-key")
else:
if not configured:
p.api_key_env = "" # makes is_configured False for non-ollama
return p