chore(conductor): Mark track 'Saved Tool Presets' as complete

This commit is contained in:
2026-03-10 01:23:57 -04:00
parent 5f208684db
commit dcc13efaf7
24 changed files with 899 additions and 121 deletions

View File

@@ -0,0 +1,88 @@
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()