aci/tests/integration/editor/test_file_validation.py

"""Tests for file validation in file editor."""

import os
from pathlib import Path

from hanzo_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()