tests/test_tool_execution.py

import pytest
from agent.nodes import execute_tool
from agent.state import AgentState

def test_tool_execution_missing_fields(mocker):
    mock_tool = mocker.patch('agent.nodes.mock_lead_capture')

    state = AgentState(
        conversation_history=[],
        current_message="",
        detected_intent="HIGH_INTENT_LEAD",
        retrieved_documents=[],
        user_name="Alex",
        user_email="alex@email.com",
        creator_platform=None,
        lead_ready=True,
        response=""
    )

    result = execute_tool(state)


    mock_tool.assert_not_called()
    assert "Error" in result["response"]

def test_tool_execution_all_fields(mocker):
    mock_tool = mocker.patch('agent.nodes.mock_lead_capture')

    state = AgentState(
        conversation_history=[],
        current_message="",
        detected_intent="HIGH_INTENT_LEAD",
        retrieved_documents=[],
        user_name="Alex",
        user_email="alex@email.com",
        creator_platform="YouTube",
        lead_ready=True,
        response=""
    )

    result = execute_tool(state)


    mock_tool.assert_called_once_with("Alex", "alex@email.com", "YouTube")
    assert "Thanks Alex" in result["response"]