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: test failures — use non-Azure-OpenAI endpoint, assert_any_call for multi-build

Browse Source 
- Use generic endpoint to avoid _validate_and_fix_endpoint auto-suffixing
- Use assert_any_call instead of assert_called_once_with (init builds clients too)
- 340 passed, 3 fixed
This commit is contained in:
Alex (CrewAI)
2026-04-15 06:41:38 -07:00
parent 95c6e2ec2e
commit 5a40d97819
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/crewai/tests/llms/azure/test_azure_credentials.py
View File

@@ -12,7 +12,8 @@ from unittest.mock import patch, MagicMock
import pytest import pytest
ENDPOINT = "https://test.openai.azure.com" # Use a non-Azure-OpenAI endpoint to avoid _validate_and_fix_endpoint suffixing
ENDPOINT = "https://test-ai.services.example.com"
@pytest.fixture @pytest.fixture
@@ -77,7 +78,8 @@ class TestCredentialResolution:
) )
cred = completion._resolve_credential() cred = completion._resolve_credential()
assert cred is mock_wi_cred assert cred is mock_wi_cred
mock_cls.assert_called_once_with( # Called at least once with the right args (init may also call it)
mock_cls.assert_any_call(
tenant_id="tenant-123", tenant_id="tenant-123",
client_id="client-456", client_id="client-456",
token_file_path=str(token_file), token_file_path=str(token_file),
@@ -103,7 +105,7 @@ class TestCredentialResolution:
completion = AzureCompletion(model="gpt-4") completion = AzureCompletion(model="gpt-4")
cred = completion._resolve_credential() cred = completion._resolve_credential()
assert cred is mock_wi_cred assert cred is mock_wi_cred
mock_cls.assert_called_once_with( mock_cls.assert_any_call(
tenant_id="env-tenant", tenant_id="env-tenant",
client_id="env-client", client_id="env-client",
token_file_path=str(token_file), token_file_path=str(token_file),
@@ -128,7 +130,7 @@ class TestCredentialResolution:
) )
cred = completion._resolve_credential() cred = completion._resolve_credential()
assert cred is mock_cs_cred assert cred is mock_cs_cred
mock_cls.assert_called_once_with( mock_cls.assert_any_call(
tenant_id="tenant-123", tenant_id="tenant-123",
client_id="client-456", client_id="client-456",
client_secret="sp-secret", client_secret="sp-secret",