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
95 lines
3.5 KiB
Python
95 lines
3.5 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from dev_aci.editor.editor import OHEditor
|
|
from dev_aci.editor.exceptions import EditorToolParameterInvalidError
|
|
|
|
|
|
def test_workspace_root_as_cwd(tmp_path):
|
|
"""Test that workspace_root is used as the current working directory for path suggestions."""
|
|
# Create a workspace root
|
|
workspace_root = tmp_path / "workspace"
|
|
workspace_root.mkdir()
|
|
|
|
# Create a file inside the workspace root
|
|
test_file = workspace_root / "test.txt"
|
|
test_file.write_text("This is a test file")
|
|
|
|
# Initialize editor with workspace_root
|
|
editor = OHEditor(workspace_root=str(workspace_root))
|
|
|
|
# Test that a relative path suggestion uses the workspace_root
|
|
relative_path = "test.txt"
|
|
with pytest.raises(EditorToolParameterInvalidError) as exc_info:
|
|
editor(command="view", path=relative_path)
|
|
|
|
error_message = str(exc_info.value.message)
|
|
assert "The path should be an absolute path" in error_message
|
|
assert "Maybe you meant" in error_message
|
|
|
|
# Extract the suggested path from the error message
|
|
suggested_path = error_message.split("Maybe you meant ")[1].strip("?")
|
|
assert Path(suggested_path).is_absolute()
|
|
assert str(workspace_root) in suggested_path
|
|
|
|
# Test with a non-existent file
|
|
non_existent_path = "non_existent.txt"
|
|
with pytest.raises(EditorToolParameterInvalidError) as exc_info:
|
|
editor(command="view", path=non_existent_path)
|
|
|
|
error_message = str(exc_info.value.message)
|
|
assert "The path should be an absolute path" in error_message
|
|
assert "Maybe you meant" not in error_message
|
|
|
|
|
|
def test_relative_workspace_root_raises_error(tmp_path, monkeypatch):
|
|
"""Test that a relative workspace_root raises a ValueError."""
|
|
# Set up a directory structure
|
|
current_dir = tmp_path / "current_dir"
|
|
current_dir.mkdir()
|
|
|
|
# Change to the current directory
|
|
monkeypatch.chdir(current_dir)
|
|
|
|
# Initialize editor with a relative workspace_root should raise ValueError
|
|
with pytest.raises(ValueError) as exc_info:
|
|
OHEditor(workspace_root="workspace")
|
|
|
|
# Check error message
|
|
error_message = str(exc_info.value)
|
|
assert "workspace_root must be an absolute path" in error_message
|
|
|
|
|
|
def test_no_suggestion_when_no_workspace_root(tmp_path, monkeypatch):
|
|
"""Test that no path suggestion is made when workspace_root is not provided."""
|
|
# Create a temporary file in the current directory
|
|
current_dir = tmp_path / "current_dir"
|
|
current_dir.mkdir()
|
|
test_file = current_dir / "test.txt"
|
|
test_file.write_text("This is a test file")
|
|
|
|
# Set the current directory to our temporary directory
|
|
monkeypatch.chdir(current_dir)
|
|
|
|
# Initialize editor without workspace_root
|
|
editor = OHEditor()
|
|
|
|
# Test that no path suggestion is made, even for existing files
|
|
relative_path = "test.txt"
|
|
with pytest.raises(EditorToolParameterInvalidError) as exc_info:
|
|
editor(command="view", path=relative_path)
|
|
|
|
error_message = str(exc_info.value.message)
|
|
assert "The path should be an absolute path" in error_message
|
|
assert "Maybe you meant" not in error_message
|
|
|
|
# Test with a non-existent file (should also have no suggestion)
|
|
non_existent_path = "non_existent.txt"
|
|
with pytest.raises(EditorToolParameterInvalidError) as exc_info:
|
|
editor(command="view", path=non_existent_path)
|
|
|
|
error_message = str(exc_info.value.message)
|
|
assert "The path should be an absolute path" in error_message
|
|
assert "Maybe you meant" not in error_message
|