mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-26 11:49:36 +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
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
"""Tests for file validation in file editor."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from dev_aci.editor import file_editor
|
|
|
|
from .conftest import parse_result
|
|
|
|
|
|
def test_file_validation(temp_file):
|
|
"""Test file validation for various file types."""
|
|
# Ensure temp_file has .sql suffix
|
|
temp_file_sql = Path(temp_file).with_suffix(".sql")
|
|
os.rename(temp_file, temp_file_sql)
|
|
|
|
# Test binary file
|
|
with open(temp_file_sql, "wb") as f:
|
|
f.write(b"Some text\x00with binary\x00content")
|
|
|
|
result = file_editor(
|
|
command="view",
|
|
path=str(temp_file_sql),
|
|
enable_linting=False,
|
|
)
|
|
result_json = parse_result(result)
|
|
assert "binary" in result_json["formatted_output_and_error"].lower()
|
|
|
|
# Test large file
|
|
large_size = 11 * 1024 * 1024 # 11MB
|
|
with open(temp_file_sql, "w") as f:
|
|
f.write("x" * large_size)
|
|
|
|
result = file_editor(
|
|
command="view",
|
|
path=str(temp_file_sql),
|
|
enable_linting=False,
|
|
)
|
|
result_json = parse_result(result)
|
|
assert "too large" in result_json["formatted_output_and_error"]
|
|
assert "10MB" in result_json["formatted_output_and_error"]
|
|
|
|
# Test SQL file
|
|
sql_content = """
|
|
SELECT *
|
|
FROM users
|
|
WHERE id = 1;
|
|
"""
|
|
with open(temp_file_sql, "w") as f:
|
|
f.write(sql_content)
|
|
|
|
result = file_editor(
|
|
command="view",
|
|
path=str(temp_file_sql),
|
|
enable_linting=False,
|
|
)
|
|
result_json = parse_result(result)
|
|
assert "SELECT *" in result_json["formatted_output_and_error"]
|
|
assert "binary" not in result_json["formatted_output_and_error"].lower()
|