import unittest.mock as mock

import pytest

from mistralai.client import MistralClient

from mistralai.models.models import ModelList

from .utils import mock_list_models_response_payload, mock_response

@pytest.fixture()

def client():

client = MistralClient()

client._client = mock.MagicMock()

return client

class TestListModels:

def test_list_models(self, client):

client._client.request.return_value = mock_response(

200,

mock_list_models_response_payload(),

)

result = client.list_models()

client._client.request.assert_called_once_with(

"get",

"https://api.mistral.ai/v1/models",

headers={

"User-Agent": f"mistral-client-python/{client._version}",

"Accept": "application/json",

"Authorization": "Bearer None",

"Content-Type": "application/json",

},

json={},

)

assert isinstance(result, ModelList), "Should return an ModelList"

assert len(result.data) == 4

assert result.object == "list"