diff --git a/tests/test_tools.py b/tests/test_tools.py index bfd8bb7..89aad6e 100644 --- a/tests/test_tools.py +++ b/tests/test_tools.py @@ -1,21 +1,19 @@ """Tests for tool execution engine and built-in tools.""" -import pytest -import asyncio from pathlib import Path -from unittest.mock import patch, MagicMock -from mcp_server_cli.tools.base import ToolBase, ToolResult, ToolRegistry +import pytest + +from mcp_server_cli.models import ToolParameter, ToolSchema +from mcp_server_cli.tools.base import ToolBase, ToolRegistry, ToolResult from mcp_server_cli.tools.file_tools import ( - FileTools, + GlobFilesTool, + ListDirectoryTool, ReadFileTool, WriteFileTool, - ListDirectoryTool, - GlobFilesTool, ) -from mcp_server_cli.tools.shell_tools import ShellTools, ExecuteCommandTool from mcp_server_cli.tools.git_tools import GitTools -from mcp_server_cli.models import ToolSchema, ToolParameter +from mcp_server_cli.tools.shell_tools import ExecuteCommandTool class TestToolBase: