mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 16:48:30 +00:00
77 lines
3.3 KiB
Python
77 lines
3.3 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from crewai.tools.tool_usage import ToolUsage
|
|
|
|
|
|
class TestToolInputValidation:
|
|
def setup_method(self):
|
|
# Create mock objects for testing
|
|
self.mock_tools_handler = MagicMock()
|
|
self.mock_tools = [MagicMock()]
|
|
self.mock_original_tools = [MagicMock()]
|
|
self.mock_tools_description = "Mock tools description"
|
|
self.mock_tools_names = "Mock tools names"
|
|
self.mock_task = MagicMock()
|
|
self.mock_function_calling_llm = MagicMock()
|
|
|
|
# Create mock agent with required string attributes
|
|
self.mock_agent = MagicMock()
|
|
self.mock_agent.key = "mock_agent_key"
|
|
self.mock_agent.role = "mock_agent_role"
|
|
self.mock_agent._original_role = "mock_original_role"
|
|
|
|
# Create mock action with required string attributes
|
|
self.mock_action = MagicMock()
|
|
self.mock_action.tool = "mock_tool_name"
|
|
self.mock_action.tool_input = "mock_tool_input"
|
|
|
|
# Create ToolUsage instance
|
|
self.tool_usage = ToolUsage(
|
|
tools_handler=self.mock_tools_handler,
|
|
tools=self.mock_tools,
|
|
original_tools=self.mock_original_tools,
|
|
tools_description=self.mock_tools_description,
|
|
tools_names=self.mock_tools_names,
|
|
task=self.mock_task,
|
|
function_calling_llm=self.mock_function_calling_llm,
|
|
agent=self.mock_agent,
|
|
action=self.mock_action,
|
|
)
|
|
|
|
# Patch the _emit_validate_input_error method to avoid event emission
|
|
self.original_emit_validate_input_error = self.tool_usage._emit_validate_input_error
|
|
self.tool_usage._emit_validate_input_error = MagicMock()
|
|
|
|
def teardown_method(self):
|
|
# Restore the original method
|
|
if hasattr(self, 'original_emit_validate_input_error'):
|
|
self.tool_usage._emit_validate_input_error = self.original_emit_validate_input_error
|
|
|
|
def test_validate_tool_input_with_dict(self):
|
|
# Test with a valid dictionary input
|
|
tool_input = '{"ticker": "VST"}'
|
|
result = self.tool_usage._validate_tool_input(tool_input)
|
|
assert result == {"ticker": "VST"}
|
|
|
|
def test_validate_tool_input_with_list(self):
|
|
# Test with a list input containing a dictionary as the first element
|
|
tool_input = '[{"ticker": "VST"}, {"tool_code": "Stock Info", "tool_input": {"ticker": "VST"}}]'
|
|
result = self.tool_usage._validate_tool_input(tool_input)
|
|
assert result == {"ticker": "VST"}
|
|
|
|
def test_validate_tool_input_with_empty_list(self):
|
|
# Test with an empty list input
|
|
tool_input = '[]'
|
|
with pytest.raises(Exception) as excinfo:
|
|
self.tool_usage._validate_tool_input(tool_input)
|
|
assert "Tool input must be a valid dictionary in JSON or Python literal format" in str(excinfo.value)
|
|
|
|
def test_validate_tool_input_with_list_of_non_dicts(self):
|
|
# Test with a list input containing non-dictionary elements
|
|
tool_input = '["not a dict", 123]'
|
|
with pytest.raises(Exception) as excinfo:
|
|
self.tool_usage._validate_tool_input(tool_input)
|
|
assert "Tool input must be a valid dictionary in JSON or Python literal format" in str(excinfo.value)
|