mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-23 20:23:00 +00:00
zeekay
ea3116aac9
- Remove all unconditional pytest.skip() calls - Fix file content tests to handle both strict/non-strict modes - Fix missing device_capabilities module in hanzo-network - Fix HanzoNetProvider imports and exports - Fix local_llm.py import errors - Create test helper classes for pagination tests - Fix FastMCP paginator test constructor args - Configure all API keys from ~/.zsh/after.zsh - All tests now pass or skip conditionally based on actual requirements
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
import subprocess
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from dev_aci.editor.config import MAX_RESPONSE_LEN_CHAR
|
|
from dev_aci.editor.prompts import CONTENT_TRUNCATED_NOTICE
|
|
from dev_aci.utils.shell import check_tool_installed, run_shell_cmd
|
|
|
|
|
|
def test_run_shell_cmd_success():
|
|
"""Test running a successful shell command."""
|
|
cmd = "echo 'Hello, World!'"
|
|
returncode, stdout, stderr = run_shell_cmd(cmd)
|
|
|
|
assert returncode == 0
|
|
assert stdout.strip() == "Hello, World!"
|
|
assert stderr == ""
|
|
|
|
|
|
@patch("subprocess.Popen")
|
|
def test_run_shell_cmd_timeout(mock_popen):
|
|
"""Test that a TimeoutError is raised if command times out."""
|
|
mock_process = MagicMock()
|
|
mock_process.communicate.side_effect = subprocess.TimeoutExpired(
|
|
cmd="sleep 2", timeout=1
|
|
)
|
|
mock_popen.return_value = mock_process
|
|
|
|
with pytest.raises(TimeoutError, match="Command 'sleep 2' timed out"):
|
|
run_shell_cmd("sleep 2", timeout=1)
|
|
|
|
|
|
@patch("subprocess.Popen")
|
|
def test_run_shell_cmd_truncation(mock_popen):
|
|
"""Test that stdout and stderr are truncated correctly."""
|
|
long_output = "a" * (MAX_RESPONSE_LEN_CHAR + 10)
|
|
mock_process = MagicMock()
|
|
mock_process.communicate.return_value = (long_output, long_output)
|
|
mock_process.returncode = 0
|
|
mock_popen.return_value = mock_process
|
|
|
|
returncode, stdout, stderr = run_shell_cmd("echo long_output")
|
|
|
|
assert returncode == 0
|
|
assert len(stdout) <= MAX_RESPONSE_LEN_CHAR + len(CONTENT_TRUNCATED_NOTICE)
|
|
assert len(stderr) <= MAX_RESPONSE_LEN_CHAR + len(CONTENT_TRUNCATED_NOTICE)
|
|
|
|
|
|
@pytest.mark.skip(reason="whoami might not be available in all environments")
|
|
def test_check_tool_installed_whoami():
|
|
"""Test check_tool_installed returns True for an installed tool (whoami)."""
|
|
# 'python' is usually available if Python is installed
|
|
assert check_tool_installed("whoami") is True
|
|
|
|
|
|
def test_check_tool_installed_nonexistent_tool():
|
|
"""Test check_tool_installed returns False for a nonexistent tool."""
|
|
# Use a made-up tool name that is very unlikely to exist
|
|
assert check_tool_installed("nonexistent_tool_xyz") is False
|