wehub-resource-sync
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
from fastapi import HTTPException
|
|
|
|
from api.v1.ppt.endpoints import anthropic as anthropic_endpoint
|
|
from api.v1.ppt.endpoints import google as google_endpoint
|
|
from api.v1.ppt.endpoints import openai as openai_endpoint
|
|
from utils.available_models import ModelAvailabilityError
|
|
from utils.provider_error_messages import INVALID_API_KEY_MESSAGE
|
|
|
|
|
|
def test_model_availability_error_maps_provider_client_errors_to_bad_request():
|
|
error = ModelAvailabilityError(
|
|
"OpenAI-compatible provider",
|
|
"Incorrect API key provided.",
|
|
provider_status_code=401,
|
|
)
|
|
|
|
assert error.status_code == 400
|
|
assert str(error) == INVALID_API_KEY_MESSAGE
|
|
assert error.raw_message == "Incorrect API key provided."
|
|
|
|
|
|
def test_model_availability_error_keeps_provider_server_errors_internal():
|
|
error = ModelAvailabilityError(
|
|
"OpenAI-compatible provider",
|
|
"Provider unavailable.",
|
|
provider_status_code=503,
|
|
)
|
|
|
|
assert error.status_code == 500
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("endpoint_module", "list_function_name", "args"),
|
|
[
|
|
(
|
|
openai_endpoint,
|
|
"list_available_openai_compatible_models",
|
|
("https://api.openai.com/v1", "bad-key"),
|
|
),
|
|
(google_endpoint, "list_available_google_models", ("bad-key",)),
|
|
(anthropic_endpoint, "list_available_anthropic_models", ("bad-key",)),
|
|
],
|
|
)
|
|
def test_model_availability_endpoints_return_400_for_provider_validation_errors(
|
|
monkeypatch, endpoint_module, list_function_name, args
|
|
):
|
|
async def fail_model_validation(*_args):
|
|
raise ModelAvailabilityError(
|
|
"Provider",
|
|
"Invalid API key.",
|
|
provider_status_code=401,
|
|
)
|
|
|
|
monkeypatch.setattr(endpoint_module, list_function_name, fail_model_validation)
|
|
|
|
with pytest.raises(HTTPException) as exc_info:
|
|
asyncio.run(endpoint_module.get_available_models(*args))
|
|
|
|
assert exc_info.value.status_code == 400
|
|
assert exc_info.value.detail == INVALID_API_KEY_MESSAGE
|