andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-07-06 15:39:24 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Apply automatic linting fixes to tests directory

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-12 13:31:07 +00:00
parent ad1ea46bbb
commit 46621113af
62 changed files with 1738 additions and 1821 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
tests/cli/authentication/test_auth_main.py
View File

@@ -1,17 +1,18 @@
import unittest
from unittest.mock import MagicMock, patch
import pytest
import requests
from crewai.cli.authentication.main import AuthenticationCommand
class TestAuthenticationCommand(unittest.TestCase):
def setUp(self):
def setUp(self) -> None:
self.auth_command = AuthenticationCommand()
@patch("crewai.cli.authentication.main.requests.post")
def test_get_device_code(self, mock_post):
def test_get_device_code(self, mock_post) -> None:
mock_response = MagicMock()
mock_response.json.return_value = {
"device_code": "123456",
@@ -23,16 +24,14 @@ class TestAuthenticationCommand(unittest.TestCase):
device_code_data = self.auth_command._get_device_code()
self.assertEqual(device_code_data["device_code"], "123456")
self.assertEqual(device_code_data["user_code"], "ABCDEF")
self.assertEqual(
device_code_data["verification_uri_complete"], "https://example.com"
)
self.assertEqual(device_code_data["interval"], 5)
assert device_code_data["device_code"] == "123456"
assert device_code_data["user_code"] == "ABCDEF"
assert device_code_data["verification_uri_complete"] == "https://example.com"
assert device_code_data["interval"] == 5
@patch("crewai.cli.authentication.main.console.print")
@patch("crewai.cli.authentication.main.webbrowser.open")
def test_display_auth_instructions(self, mock_open, mock_print):
def test_display_auth_instructions(self, mock_open, mock_print) -> None:
device_code_data = {
"verification_uri_complete": "https://example.com",
"user_code": "ABCDEF",
@@ -49,8 +48,8 @@ class TestAuthenticationCommand(unittest.TestCase):
@patch("crewai.cli.authentication.main.validate_token")
@patch("crewai.cli.authentication.main.console.print")
def test_poll_for_token_success(
self, mock_print, mock_validate_token, mock_post, mock_tool
):
self, mock_print, mock_validate_token, mock_post, mock_tool,
) -> None:
mock_response = MagicMock()
mock_response.status_code = 200
mock_response.json.return_value = {
@@ -66,12 +65,12 @@ class TestAuthenticationCommand(unittest.TestCase):
mock_validate_token.assert_called_once_with("TOKEN")
mock_print.assert_called_once_with(
"\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n"
"\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n",
)
@patch("crewai.cli.authentication.main.requests.post")
@patch("crewai.cli.authentication.main.console.print")
def test_poll_for_token_error(self, mock_print, mock_post):
def test_poll_for_token_error(self, mock_print, mock_post) -> None:
mock_response = MagicMock()
mock_response.status_code = 400
mock_response.json.return_value = {
@@ -80,14 +79,14 @@ class TestAuthenticationCommand(unittest.TestCase):
}
mock_post.return_value = mock_response
with self.assertRaises(requests.HTTPError):
with pytest.raises(requests.HTTPError):
self.auth_command._poll_for_token({"device_code": "123456"})
mock_print.assert_not_called()
@patch("crewai.cli.authentication.main.requests.post")
@patch("crewai.cli.authentication.main.console.print")
def test_poll_for_token_timeout(self, mock_print, mock_post):
def test_poll_for_token_timeout(self, mock_print, mock_post) -> None:
mock_response = MagicMock()
mock_response.status_code = 400
mock_response.json.return_value = {
@@ -99,5 +98,5 @@ class TestAuthenticationCommand(unittest.TestCase):
self.auth_command._poll_for_token({"device_code": "123456", "interval": 0.01})
mock_print.assert_called_once_with(
"Timeout: Failed to get the token. Please try again.", style="bold red"
"Timeout: Failed to get the token. Please try again.", style="bold red",
)

40
tests/cli/authentication/test_utils.py
View File

@@ -11,14 +11,14 @@ from crewai.cli.authentication.utils import TokenManager, validate_token
class TestValidateToken(unittest.TestCase):
@patch("crewai.cli.authentication.utils.AsymmetricSignatureVerifier")
@patch("crewai.cli.authentication.utils.TokenVerifier")
def test_validate_token(self, mock_token_verifier, mock_asymmetric_verifier):
def test_validate_token(self, mock_token_verifier, mock_asymmetric_verifier) -> None:
mock_verifier_instance = mock_token_verifier.return_value
mock_id_token = "mock_id_token"
validate_token(mock_id_token)
mock_asymmetric_verifier.assert_called_once_with(
"https://crewai.us.auth0.com/.well-known/jwks.json"
"https://crewai.us.auth0.com/.well-known/jwks.json",
)
mock_token_verifier.assert_called_once_with(
signature_verifier=mock_asymmetric_verifier.return_value,
@@ -29,38 +29,38 @@ class TestValidateToken(unittest.TestCase):
class TestTokenManager(unittest.TestCase):
def setUp(self):
def setUp(self) -> None:
self.token_manager = TokenManager()
@patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
@patch("crewai.cli.authentication.utils.TokenManager.save_secure_file")
@patch("crewai.cli.authentication.utils.TokenManager._get_or_create_key")
def test_get_or_create_key_existing(self, mock_get_or_create, mock_save, mock_read):
def test_get_or_create_key_existing(self, mock_get_or_create, mock_save, mock_read) -> None:
mock_key = Fernet.generate_key()
mock_get_or_create.return_value = mock_key
token_manager = TokenManager()
result = token_manager.key
self.assertEqual(result, mock_key)
assert result == mock_key
@patch("crewai.cli.authentication.utils.Fernet.generate_key")
@patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
@patch("crewai.cli.authentication.utils.TokenManager.save_secure_file")
def test_get_or_create_key_new(self, mock_save, mock_read, mock_generate):
def test_get_or_create_key_new(self, mock_save, mock_read, mock_generate) -> None:
mock_key = b"new_key"
mock_read.return_value = None
mock_generate.return_value = mock_key
result = self.token_manager._get_or_create_key()
self.assertEqual(result, mock_key)
assert result == mock_key
mock_read.assert_called_once_with("secret.key")
mock_generate.assert_called_once()
mock_save.assert_called_once_with("secret.key", mock_key)
@patch("crewai.cli.authentication.utils.TokenManager.save_secure_file")
def test_save_tokens(self, mock_save):
def test_save_tokens(self, mock_save) -> None:
access_token = "test_token"
expires_in = 3600
@@ -68,10 +68,10 @@ class TestTokenManager(unittest.TestCase):
mock_save.assert_called_once()
args = mock_save.call_args[0]
self.assertEqual(args[0], "tokens.enc")
assert args[0] == "tokens.enc"
decrypted_data = self.token_manager.fernet.decrypt(args[1])
data = json.loads(decrypted_data)
self.assertEqual(data["access_token"], access_token)
assert data["access_token"] == access_token
expiration = datetime.fromisoformat(data["expiration"])
self.assertAlmostEqual(
expiration,
@@ -80,7 +80,7 @@ class TestTokenManager(unittest.TestCase):
)
@patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
def test_get_token_valid(self, mock_read):
def test_get_token_valid(self, mock_read) -> None:
access_token = "test_token"
expiration = (datetime.now() + timedelta(hours=1)).isoformat()
data = {"access_token": access_token, "expiration": expiration}
@@ -89,10 +89,10 @@ class TestTokenManager(unittest.TestCase):
result = self.token_manager.get_token()
self.assertEqual(result, access_token)
assert result == access_token
@patch("crewai.cli.authentication.utils.TokenManager.read_secure_file")
def test_get_token_expired(self, mock_read):
def test_get_token_expired(self, mock_read) -> None:
access_token = "test_token"
expiration = (datetime.now() - timedelta(hours=1)).isoformat()
data = {"access_token": access_token, "expiration": expiration}
@@ -101,12 +101,12 @@ class TestTokenManager(unittest.TestCase):
result = self.token_manager.get_token()
self.assertIsNone(result)
assert result is None
@patch("crewai.cli.authentication.utils.TokenManager.get_secure_storage_path")
@patch("builtins.open", new_callable=unittest.mock.mock_open)
@patch("crewai.cli.authentication.utils.os.chmod")
def test_save_secure_file(self, mock_chmod, mock_open, mock_get_path):
def test_save_secure_file(self, mock_chmod, mock_open, mock_get_path) -> None:
mock_path = MagicMock()
mock_get_path.return_value = mock_path
filename = "test_file.txt"
@@ -121,9 +121,9 @@ class TestTokenManager(unittest.TestCase):
@patch("crewai.cli.authentication.utils.TokenManager.get_secure_storage_path")
@patch(
"builtins.open", new_callable=unittest.mock.mock_open, read_data=b"test_content"
"builtins.open", new_callable=unittest.mock.mock_open, read_data=b"test_content",
)
def test_read_secure_file_exists(self, mock_open, mock_get_path):
def test_read_secure_file_exists(self, mock_open, mock_get_path) -> None:
mock_path = MagicMock()
mock_get_path.return_value = mock_path
mock_path.__truediv__.return_value.exists.return_value = True
@@ -131,12 +131,12 @@ class TestTokenManager(unittest.TestCase):
result = self.token_manager.read_secure_file(filename)
self.assertEqual(result, b"test_content")
assert result == b"test_content"
mock_path.__truediv__.assert_called_once_with(filename)
mock_open.assert_called_once_with(mock_path.__truediv__.return_value, "rb")
@patch("crewai.cli.authentication.utils.TokenManager.get_secure_storage_path")
def test_read_secure_file_not_exists(self, mock_get_path):
def test_read_secure_file_not_exists(self, mock_get_path) -> None:
mock_path = MagicMock()
mock_get_path.return_value = mock_path
mock_path.__truediv__.return_value.exists.return_value = False
@@ -144,5 +144,5 @@ class TestTokenManager(unittest.TestCase):
result = self.token_manager.read_secure_file(filename)
self.assertIsNone(result)
assert result is None
mock_path.__truediv__.assert_called_once_with(filename)