Skip to content

Commit 800d409

Browse files
DevStrategistclaude
andcommitted
Fix test_get_providers mock to match actual API response format
The OpenRouter endpoints API returns { data: { endpoints: [...] } } with provider_name fields, not the flat array the test was mocking. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent b6de01d commit 800d409

1 file changed

Lines changed: 12 additions & 9 deletions

File tree

llm-benchmark/tests/test_clients.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -51,21 +51,24 @@ async def test_get_models(self, client):
5151
async def test_get_providers(self, client):
5252
mock_response = MagicMock()
5353
mock_response.json.return_value = {
54-
"data": [
55-
{"id": "together", "name": "Together", "order": 0, "is_moderated": False},
56-
{"id": "openai", "name": "OpenAI", "order": 1, "is_moderated": True}
57-
]
54+
"data": {
55+
"endpoints": [
56+
{"provider_name": "Together"},
57+
{"provider_name": "OpenAI"}
58+
]
59+
}
5860
}
5961
mock_response.raise_for_status = MagicMock()
60-
62+
6163
with patch.object(client.client, 'get', return_value=mock_response) as mock_get:
6264
providers = await client.get_providers("openai/gpt-4")
63-
65+
6466
assert len(providers) == 2
65-
assert providers[0].id == "together"
67+
assert providers[0].name == "Together"
6668
assert providers[0].order == 0
67-
assert providers[1].is_moderated == True
68-
69+
assert providers[1].name == "OpenAI"
70+
assert providers[1].order == 1
71+
6972
mock_get.assert_called_once()
7073
assert "openai/gpt-4/endpoints" in mock_get.call_args[0][0]
7174

0 commit comments

Comments
 (0)