hud-python 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

hud/tools/tests/test_bash.py

@@ -1,158 +1,158 @@
-"""Tests for bash tool."""
-
-from __future__ import annotations
-
-from unittest.mock import AsyncMock, MagicMock, patch
-
-import pytest
-
-from hud.tools.bash import BashTool, ContentResult, ToolError, _BashSession
-from hud.tools.types import TextContent
-
-
-class TestBashSession:
-    """Tests for _BashSession."""
-
-    @pytest.mark.asyncio
-    async def test_session_start(self):
-        """Test starting a bash session."""
-        session = _BashSession()
-        assert session._started is False
-
-        with patch("asyncio.create_subprocess_shell") as mock_create:
-            mock_process = MagicMock()
-            mock_create.return_value = mock_process
-
-            await session.start()
-
-            assert session._started is True
-            assert session._process == mock_process
-            mock_create.assert_called_once()
-
-    def test_session_stop_not_started(self):
-        """Test stopping a session that hasn't started."""
-        session = _BashSession()
-
-        with pytest.raises(ToolError) as exc_info:
-            session.stop()
-
-        assert "Session has not started" in str(exc_info.value)
-
-    @pytest.mark.asyncio
-    async def test_session_run_not_started(self):
-        """Test running command on a session that hasn't started."""
-        session = _BashSession()
-
-        with pytest.raises(ToolError) as exc_info:
-            await session.run("echo test")
-
-        assert "Session has not started" in str(exc_info.value)
-
-    @pytest.mark.asyncio
-    async def test_session_run_success(self):
-        """Test successful command execution."""
-        session = _BashSession()
-        session._started = True
-
-        # Mock process
-        mock_process = MagicMock()
-        mock_process.returncode = None
-        mock_process.stdin = MagicMock()
-        mock_process.stdin.write = MagicMock()
-        mock_process.stdin.drain = AsyncMock()
-        mock_process.stdout = MagicMock()
-        mock_process.stdout.readuntil = AsyncMock(return_value=b"Hello World\n<<exit>>\n")
-        mock_process.stderr = MagicMock()
-        mock_process.stderr.read = AsyncMock(return_value=b"")
-
-        session._process = mock_process
-
-        result = await session.run("echo Hello World")
-
-        assert result.output == "Hello World\n"
-        assert result.error == ""
-
-
-class TestBashTool:
-    """Tests for BashTool."""
-
-    def test_bash_tool_init(self):
-        """Test BashTool initialization."""
-        tool = BashTool()
-        assert tool.session is None
-
-    @pytest.mark.asyncio
-    async def test_call_with_command(self):
-        """Test calling tool with a command."""
-        tool = BashTool()
-
-        # Mock session
-        mock_session = MagicMock()
-        mock_session.run = AsyncMock(return_value=ContentResult(output="test output"))
-
-        # Mock _BashSession creation
-        with patch("hud.tools.bash._BashSession") as mock_session_class:
-            mock_session_class.return_value = mock_session
-            mock_session.start = AsyncMock()
-
-            result = await tool(command="echo test")
-
-            assert isinstance(result, list)
-            assert len(result) == 1
-            assert isinstance(result[0], TextContent)
-            assert result[0].text == "test output"
-            mock_session.start.assert_called_once()
-            mock_session.run.assert_called_once_with("echo test")
-
-    @pytest.mark.asyncio
-    async def test_call_restart(self):
-        """Test restarting the tool."""
-        tool = BashTool()
-
-        # Set up existing session
-        old_session = MagicMock()
-        old_session.stop = MagicMock()
-        tool.session = old_session
-
-        # Mock new session
-        new_session = MagicMock()
-        new_session.start = AsyncMock()
-
-        with patch("hud.tools.bash._BashSession", return_value=new_session):
-            result = await tool(restart=True)
-
-            assert isinstance(result, list)
-            assert len(result) == 1
-            assert isinstance(result[0], TextContent)
-            assert result[0].text == "Bash session restarted."
-            old_session.stop.assert_called_once()
-            new_session.start.assert_called_once()
-            assert tool.session == new_session
-
-    @pytest.mark.asyncio
-    async def test_call_no_command_error(self):
-        """Test calling without command raises error."""
-        tool = BashTool()
-
-        with pytest.raises(ToolError) as exc_info:
-            await tool()
-
-        assert str(exc_info.value) == "No command provided."
-
-    @pytest.mark.asyncio
-    async def test_call_with_existing_session(self):
-        """Test calling with an existing session."""
-        tool = BashTool()
-
-        # Set up existing session
-        existing_session = MagicMock()
-        existing_session.run = AsyncMock(return_value=ContentResult(output="result"))
-        tool.session = existing_session
-
-        result = await tool(command="ls")
-
-        assert isinstance(result, list)
-        assert len(result) == 1
-        assert isinstance(result[0], TextContent)
-        assert result[0].text == "result"
-        existing_session.run.assert_called_once_with("ls")
+"""Tests for bash tool."""
+
+from __future__ import annotations
+
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from hud.tools.bash import BashTool, ContentResult, ToolError, _BashSession
+from hud.tools.types import TextContent
+
+
+class TestBashSession:
+    """Tests for _BashSession."""
+
+    @pytest.mark.asyncio
+    async def test_session_start(self):
+        """Test starting a bash session."""
+        session = _BashSession()
+        assert session._started is False
+
+        with patch("asyncio.create_subprocess_shell") as mock_create:
+            mock_process = MagicMock()
+            mock_create.return_value = mock_process
+
+            await session.start()
+
+            assert session._started is True
+            assert session._process == mock_process
+            mock_create.assert_called_once()
+
+    def test_session_stop_not_started(self):
+        """Test stopping a session that hasn't started."""
+        session = _BashSession()
+
+        with pytest.raises(ToolError) as exc_info:
+            session.stop()
+
+        assert "Session has not started" in str(exc_info.value)
+
+    @pytest.mark.asyncio
+    async def test_session_run_not_started(self):
+        """Test running command on a session that hasn't started."""
+        session = _BashSession()
+
+        with pytest.raises(ToolError) as exc_info:
+            await session.run("echo test")
+
+        assert "Session has not started" in str(exc_info.value)
+
+    @pytest.mark.asyncio
+    async def test_session_run_success(self):
+        """Test successful command execution."""
+        session = _BashSession()
+        session._started = True
+
+        # Mock process
+        mock_process = MagicMock()
+        mock_process.returncode = None
+        mock_process.stdin = MagicMock()
+        mock_process.stdin.write = MagicMock()
+        mock_process.stdin.drain = AsyncMock()
+        mock_process.stdout = MagicMock()
+        mock_process.stdout.readuntil = AsyncMock(return_value=b"Hello World\n<<exit>>\n")
+        mock_process.stderr = MagicMock()
+        mock_process.stderr.read = AsyncMock(return_value=b"")
+
+        session._process = mock_process
+
+        result = await session.run("echo Hello World")
+
+        assert result.output == "Hello World\n"
+        assert result.error == ""
+
+
+class TestBashTool:
+    """Tests for BashTool."""
+
+    def test_bash_tool_init(self):
+        """Test BashTool initialization."""
+        tool = BashTool()
+        assert tool.session is None
+
+    @pytest.mark.asyncio
+    async def test_call_with_command(self):
+        """Test calling tool with a command."""
+        tool = BashTool()
+
+        # Mock session
+        mock_session = MagicMock()
+        mock_session.run = AsyncMock(return_value=ContentResult(output="test output"))
+
+        # Mock _BashSession creation
+        with patch("hud.tools.bash._BashSession") as mock_session_class:
+            mock_session_class.return_value = mock_session
+            mock_session.start = AsyncMock()
+
+            result = await tool(command="echo test")
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            assert isinstance(result[0], TextContent)
+            assert result[0].text == "test output"
+            mock_session.start.assert_called_once()
+            mock_session.run.assert_called_once_with("echo test")
+
+    @pytest.mark.asyncio
+    async def test_call_restart(self):
+        """Test restarting the tool."""
+        tool = BashTool()
+
+        # Set up existing session
+        old_session = MagicMock()
+        old_session.stop = MagicMock()
+        tool.session = old_session
+
+        # Mock new session
+        new_session = MagicMock()
+        new_session.start = AsyncMock()
+
+        with patch("hud.tools.bash._BashSession", return_value=new_session):
+            result = await tool(restart=True)
+
+            assert isinstance(result, list)
+            assert len(result) == 1
+            assert isinstance(result[0], TextContent)
+            assert result[0].text == "Bash session restarted."
+            old_session.stop.assert_called_once()
+            new_session.start.assert_called_once()
+            assert tool.session == new_session
+
+    @pytest.mark.asyncio
+    async def test_call_no_command_error(self):
+        """Test calling without command raises error."""
+        tool = BashTool()
+
+        with pytest.raises(ToolError) as exc_info:
+            await tool()
+
+        assert str(exc_info.value) == "No command provided."
+
+    @pytest.mark.asyncio
+    async def test_call_with_existing_session(self):
+        """Test calling with an existing session."""
+        tool = BashTool()
+
+        # Set up existing session
+        existing_session = MagicMock()
+        existing_session.run = AsyncMock(return_value=ContentResult(output="result"))
+        tool.session = existing_session
+
+        result = await tool(command="ls")
+
+        assert isinstance(result, list)
+        assert len(result) == 1
+        assert isinstance(result[0], TextContent)
+        assert result[0].text == "result"
+        existing_session.run.assert_called_once_with("ls")