andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 07:13:00 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix tests: Import SSLError from requests.exceptions

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-13 12:07:29 +00:00
parent 71eebe6b1f
commit 22b81c4dc8
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/cli/provider_test.py
View File

@@ -4,6 +4,7 @@ from unittest import mock
import pytest import pytest
import requests import requests
from requests.exceptions import RequestException, SSLError, Timeout
from crewai.cli.provider import ( from crewai.cli.provider import (
fetch_provider_data, fetch_provider_data,
@@ -53,7 +54,7 @@ class TestProviderFunctions:
@mock.patch("crewai.cli.provider.requests.get") @mock.patch("crewai.cli.provider.requests.get")
def test_fetch_provider_data_handles_request_exception(self, mock_get, mock_cache_file): def test_fetch_provider_data_handles_request_exception(self, mock_get, mock_cache_file):
"""Test that fetch_provider_data handles RequestException properly.""" """Test that fetch_provider_data handles RequestException properly."""
mock_get.side_effect = requests.RequestException("Test error") mock_get.side_effect = RequestException("Test error")
result = fetch_provider_data(mock_cache_file) result = fetch_provider_data(mock_cache_file)
@@ -63,7 +64,7 @@ class TestProviderFunctions:
@mock.patch("crewai.cli.provider.requests.get") @mock.patch("crewai.cli.provider.requests.get")
def test_fetch_provider_data_handles_timeout(self, mock_get, mock_cache_file): def test_fetch_provider_data_handles_timeout(self, mock_get, mock_cache_file):
"""Test that fetch_provider_data handles Timeout exception properly.""" """Test that fetch_provider_data handles Timeout exception properly."""
mock_get.side_effect = requests.Timeout("Connection timed out") mock_get.side_effect = Timeout("Connection timed out")
result = fetch_provider_data(mock_cache_file) result = fetch_provider_data(mock_cache_file)
@@ -73,7 +74,7 @@ class TestProviderFunctions:
@mock.patch("crewai.cli.provider.requests.get") @mock.patch("crewai.cli.provider.requests.get")
def test_fetch_provider_data_handles_ssl_error(self, mock_get, mock_cache_file): def test_fetch_provider_data_handles_ssl_error(self, mock_get, mock_cache_file):
"""Test that fetch_provider_data handles SSLError exception properly.""" """Test that fetch_provider_data handles SSLError exception properly."""
mock_get.side_effect = requests.SSLError("SSL Certificate verification failed") mock_get.side_effect = SSLError("SSL Certificate verification failed")
result = fetch_provider_data(mock_cache_file) result = fetch_provider_data(mock_cache_file)