import pytest
from ai_client import set_agent_tools, _build_anthropic_tools

def test_agent_tools_wiring():
    # Only enable read_file and run_powershell
    agent_tools = {
        "run_powershell": True,
        "read_file": True,
        "list_directory": False,
        "search_files": False,
        "get_file_summary": False,
        "web_search": False,
        "fetch_url": False
    }
    set_agent_tools(agent_tools)
    
    anth_tools = _build_anthropic_tools()
    tool_names = [t["name"] for t in anth_tools]
    
    assert "read_file" in tool_names
    assert "run_powershell" in tool_names
    assert "list_directory" not in tool_names
    assert "web_search" not in tool_names