import pytest
from unittest.mock import patch, MagicMock
import ai_client

def test_list_models_gemini_cli() -> None:
 """
    Verifies that 'ai_client.list_models' correctly returns a list of models
    for the 'gemini_cli' provider.
    """
 models = ai_client.list_models("gemini_cli")
 assert "gemini-3.1-pro-preview" in models
 assert "gemini-3-flash-preview" in models
 assert "gemini-2.5-pro" in models
 assert "gemini-2.5-flash" in models
 assert "gemini-2.5-flash-lite" in models
 assert len(models) == 5