import pytest
import asyncio
from src import ai_client
from src import mcp_client
from src import models
from src.models import ToolPreset
from unittest.mock import MagicMock, patch

@pytest.mark.asyncio
async def test_tool_auto_approval():
 # Setup a preset with read_file as auto
 preset = ToolPreset(name="AutoTest", categories={
  "General": {"read_file": "auto"}
 })
 
 with patch("src.tool_presets.ToolPresetManager.load_all", return_value={"AutoTest": preset}):
  ai_client.set_tool_preset("AutoTest")
  
  # Mock mcp_client.async_dispatch to avoid actual file reads
  with patch("src.mcp_client.async_dispatch", return_value="File Content") as mock_dispatch:
   # pre_tool_callback should NOT be called
   mock_cb = MagicMock()
   
   name, call_id, out, _ = await ai_client._execute_single_tool_call_async(
    name="read_file",
    args={"path": "test.txt"},
    call_id="call_1",
    base_dir=".",
    pre_tool_callback=mock_cb,
    qa_callback=None,
    r_idx=0
   )
   
   assert out == "File Content"
   mock_cb.assert_not_called()
   mock_dispatch.assert_called_once()

@pytest.mark.asyncio
async def test_tool_ask_approval():
 # Setup a preset with run_powershell as ask
 preset = ToolPreset(name="AskTest", categories={
  "General": {"run_powershell": "ask"}
 })
 
 with patch("src.tool_presets.ToolPresetManager.load_all", return_value={"AskTest": preset}):
  ai_client.set_tool_preset("AskTest")
  
  # pre_tool_callback SHOULD be called
  mock_cb = MagicMock(return_value="Success")
  
  name, call_id, out, _ = await ai_client._execute_single_tool_call_async(
   name="run_powershell",
   args={"script": "dir"},
   call_id="call_2",
   base_dir=".",
   pre_tool_callback=mock_cb,
   qa_callback=None,
   r_idx=0
  )
  
  assert out == "Success"
  mock_cb.assert_called_once()

@pytest.mark.asyncio
async def test_tool_rejection():
 # Setup a preset with run_powershell as ask
 preset = ToolPreset(name="AskTest", categories={
  "General": {"run_powershell": "ask"}
 })
 
 with patch("src.tool_presets.ToolPresetManager.load_all", return_value={"AskTest": preset}):
  ai_client.set_tool_preset("AskTest")
  
  # mock_cb returns None (rejected)
  mock_cb = MagicMock(return_value=None)
  
  name, call_id, out, _ = await ai_client._execute_single_tool_call_async(
   name="run_powershell",
   args={"script": "dir"},
   call_id="call_3",
   base_dir=".",
   pre_tool_callback=mock_cb,
   qa_callback=None,
   r_idx=0
  )
  
  assert "USER REJECTED" in out
  mock_cb.assert_called_once()