orcs-oss/orcs-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(router): scope FAKE_KEY via pytest monkeypatch fixture

Browse Source
This commit is contained in:
Md.Nahin Alam
2026-04-03 04:18:20 +06:00
parent 43deb49c2c
commit 184ec250fd
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
test_smart_router.py
View File

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