tests/test_invalid_model.py

import os
os.environ["MOCK_INFERENCE"] = "true"

from fastapi.testclient import TestClient
from app import app

client = TestClient(app)

def test_invalid_model_load():
    response = client.post("/v1/models/load", json={"model": "invalid-model-id"})
    assert response.status_code == 400
    assert "detail" in response.json()
    assert "Invalid model ID" in response.json()["detail"]

def test_invalid_model_chat():
    payload = {
        "model": "invalid-model-id",
        "messages": [{"role": "user", "content": "hello"}]
    }
    response = client.post("/v1/chat/completions", json=payload)
    assert response.status_code == 400
    assert "Invalid model ID" in response.json()["detail"]