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

test: fix get_auth_token test

Browse Source
This commit is contained in:
Lucas Gomide
2025-05-12 16:52:40 -03:00
parent 9bdf0fd84e
commit a5f42cab62
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/agent_test.py
View File

@@ -2026,8 +2026,16 @@ def test_get_knowledge_search_query():
)
@pytest.fixture
def mock_get_auth_token():
with patch(
"crewai.cli.authentication.token.get_auth_token", return_value="test_token"
):
yield
@patch("crewai.cli.plus_api.PlusAPI.get_agent")
def test_agent_from_repository(mock_get_agent):
def test_agent_from_repository(mock_get_agent, mock_get_auth_token):
from crewai_tools import SerperDevTool
mock_get_response = MagicMock()
@@ -2048,12 +2056,6 @@ def test_agent_from_repository(mock_get_agent):
assert isinstance(agent.tools[0], SerperDevTool)
@pytest.fixture
def mock_get_auth_token():
with patch("crewai.cli.command.get_auth_token", return_value="test_token"):
yield
@patch("crewai.cli.plus_api.PlusAPI.get_agent")
def test_agent_from_repository_override_attributes(mock_get_agent, mock_get_auth_token):
from crewai_tools import SerperDevTool