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
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from dev_aci.editor.editor import OHEditor
|
|
from dev_aci.editor.exceptions import FileValidationError
|
|
|
|
|
|
def test_validate_large_file(tmp_path):
|
|
"""Test that large files are rejected."""
|
|
editor = OHEditor()
|
|
large_file = tmp_path / "large.txt"
|
|
|
|
# Create a file just over 10MB
|
|
file_size = 10 * 1024 * 1024 + 1024 # 10MB + 1KB
|
|
with open(large_file, "wb") as f:
|
|
f.write(b"0" * file_size)
|
|
|
|
with pytest.raises(FileValidationError) as exc_info:
|
|
editor.validate_file(large_file)
|
|
assert "File is too large" in str(exc_info.value)
|
|
assert "10.0MB" in str(exc_info.value)
|
|
|
|
|
|
def test_validate_binary_file(tmp_path):
|
|
"""Test that binary files are rejected."""
|
|
editor = OHEditor()
|
|
binary_file = tmp_path / "binary.bin"
|
|
|
|
# Create a binary file with null bytes
|
|
with open(binary_file, "wb") as f:
|
|
f.write(b"Some text\x00with binary\x00content")
|
|
|
|
with pytest.raises(FileValidationError) as exc_info:
|
|
editor.validate_file(binary_file)
|
|
assert "binary" in str(exc_info.value).lower()
|
|
|
|
|
|
def test_validate_text_file(tmp_path):
|
|
"""Test that valid text files are accepted."""
|
|
editor = OHEditor()
|
|
text_file = tmp_path / "valid.txt"
|
|
|
|
# Create a valid text file
|
|
with open(text_file, "w") as f:
|
|
f.write("This is a valid text file\nwith multiple lines\n")
|
|
|
|
# Should not raise any exception
|
|
editor.validate_file(text_file)
|
|
|
|
|
|
def test_validate_directory():
|
|
"""Test that directories are skipped in validation."""
|
|
editor = OHEditor()
|
|
# Should not raise any exception for directories
|
|
editor.validate_file(Path("/tmp"))
|
|
|
|
|
|
def test_validate_nonexistent_file():
|
|
"""Test validation of nonexistent file."""
|
|
editor = OHEditor()
|
|
nonexistent = Path("/nonexistent/file.txt")
|
|
# Should not raise FileValidationError since validate_path will handle this case
|
|
editor.validate_file(nonexistent)
|