mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
import requests
|
|
|
|
import pytest
|
|
from unittest.mock import Mock, patch
|
|
|
|
from crewai.cli.constants import JSON_URL, MODELS, PROVIDERS
|
|
from crewai.cli.provider import fetch_provider_data, get_provider_data
|
|
|
|
def test_fetch_provider_data_timeout():
|
|
with patch('requests.get') as mock_get:
|
|
mock_get.side_effect = requests.exceptions.Timeout
|
|
result = fetch_provider_data('/tmp/cache.json')
|
|
assert result is None
|
|
|
|
def test_fetch_provider_data_wrong_content_type():
|
|
with patch('requests.get') as mock_get:
|
|
mock_response = Mock()
|
|
mock_response.headers = {'content-type': 'text/plain'}
|
|
mock_get.return_value = mock_response
|
|
result = fetch_provider_data('/tmp/cache.json')
|
|
assert result is None
|
|
|
|
def test_get_provider_data_fallback():
|
|
with patch('crewai.cli.provider.load_provider_data') as mock_load:
|
|
mock_load.return_value = None
|
|
result = get_provider_data()
|
|
assert result is not None
|
|
assert all(provider.lower() in result for provider in PROVIDERS)
|
|
# Verify that each provider has its models from MODELS
|
|
for provider in PROVIDERS:
|
|
assert result[provider.lower()] == MODELS.get(provider.lower(), [])
|