diff --git a/tests/agent_test.py b/tests/agent_test.py index 968761ce9..8cd2c075c 100644 --- a/tests/agent_test.py +++ b/tests/agent_test.py @@ -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