chore(conductor): Mark track 'Saved Tool Presets' as complete
This commit is contained in:
88
tests/test_tool_presets_execution.py
Normal file
88
tests/test_tool_presets_execution.py
Normal 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()
|
||||
Reference in New Issue
Block a user