aci/tests/unit/editor/test_file_cache.py

import os
import tempfile

import pytest

from hanzo_aci.editor import FileCache


@pytest.fixture
def file_cache():
    with tempfile.TemporaryDirectory() as temp_dir:
        cache = FileCache(temp_dir)
        yield cache
        cache.clear()


def test_init(file_cache):
    assert isinstance(file_cache, FileCache)
    assert file_cache.directory.exists()
    assert file_cache.directory.is_dir()


def test_set_and_get(file_cache):
    file_cache.set('test_key', 'test_value')
    assert file_cache.get('test_key') == 'test_value'


def test_get_nonexistent_key(file_cache):
    assert file_cache.get('nonexistent_key') is None
    assert file_cache.get('nonexistent_key', 'default') == 'default'


def test_set_nested_key(file_cache):
    file_cache.set('folder/nested/key', 'nested_value')
    assert file_cache.get('folder/nested/key') == 'nested_value'


def test_set_overwrite(file_cache):
    file_cache.set('test_key', 'initial_value')
    file_cache.set('test_key', 'new_value')
    assert file_cache.get('test_key') == 'new_value'


def test_delete(file_cache):
    file_cache.set('test_key', 'test_value')
    file_cache.delete('test_key')
    assert file_cache.get('test_key') is None


def test_delete_nonexistent_key(file_cache):
    file_cache.delete('nonexistent_key')  # Should not raise an exception


def test_delete_nested_key(file_cache):
    file_cache.set('folder/nested/key', 'nested_value')
    file_cache.delete('folder/nested/key')
    assert file_cache.get('folder/nested/key') is None


def test_clear(file_cache):
    file_cache.set('key1', 'value1')
    file_cache.set('key2', 'value2')
    file_cache.set('folder/key3', 'value3')
    file_cache.clear()
    assert len(file_cache) == 0
    assert file_cache.get('key1') is None
    assert file_cache.get('key2') is None
    assert file_cache.get('folder/key3') is None


def test_contains(file_cache):
    file_cache.set('test_key', 'test_value')
    assert 'test_key' in file_cache
    assert 'nonexistent_key' not in file_cache


def test_len(file_cache):
    assert len(file_cache) == 0
    file_cache.set('key1', 'value1')
    file_cache.set('key2', 'value2')
    assert len(file_cache) == 2
    file_cache.set('folder/key3', 'value3')
    assert len(file_cache) == 3


def test_iter(file_cache):
    file_cache.set('key1', 'value1')
    file_cache.set('key2', 'value2')
    file_cache.set('folder/key3', 'value3')
    keys = set(file_cache)
    assert keys == {'key1', 'key2', 'folder/key3'}


def test_large_value(file_cache):
    large_value = 'x' * 1024 * 1024  # 1 MB string
    file_cache.set('large_key', large_value)
    assert file_cache.get('large_key') == large_value


def test_many_items(file_cache):
    for i in range(1000):
        file_cache.set(f'key_{i}', f'value_{i}')

    assert len(file_cache) == 1000
    for i in range(1000):
        assert file_cache.get(f'key_{i}') == f'value_{i}'


def test_nested_structure(file_cache):
    file_cache.set('folder1/file1', 'content1')
    file_cache.set('folder1/file2', 'content2')
    file_cache.set('folder2/subfolder/file3', 'content3')

    assert file_cache.get('folder1/file1') == 'content1'
    assert file_cache.get('folder1/file2') == 'content2'
    assert file_cache.get('folder2/subfolder/file3') == 'content3'
    assert len(file_cache) == 3


def test_clear_nested_structure(file_cache):
    file_cache.set('folder1/file1', 'content1')
    file_cache.set('folder1/file2', 'content2')
    file_cache.set('folder2/subfolder/file3', 'content3')
    file_cache.clear()

    assert len(file_cache) == 0
    assert list(file_cache) == []
    assert not any(file_cache.directory.iterdir())


def test_delete_removes_empty_directories(file_cache):
    file_cache.set('folder1/subfolder/file1', 'content1')
    file_cache.delete('folder1/subfolder/file1')

    assert not (file_cache.directory / 'folder1' / 'subfolder').exists()
    assert not (file_cache.directory / 'folder1').exists()


def test_size_limit():
    with tempfile.TemporaryDirectory() as temp_dir:
        cache = FileCache(temp_dir, size_limit=100)
        val1 = 'x' * 50
        val2 = 'y' * 60
        cache.set('key1', val1)
        cache.set('key2', val2)

        assert len(val1.encode('utf-8')) <= 100
        assert len(val1.encode('utf-8') + val2.encode('utf-8')) > 100

        val3 = 'z' * 40
        # This should cause key1 to be evicted
        cache.set('key3', val3)  # 40 bytes

        assert 'key1' not in cache
        assert 'key2' in cache
        assert 'key3' in cache


def test_file_permissions(file_cache):
    file_cache.set('test_key', 'test_value')
    file_path = file_cache._get_file_path('test_key')
    assert os.access(file_path, os.R_OK)
    assert os.access(file_path, os.W_OK)
    assert not os.access(file_path, os.X_OK)


def test_unicode_keys_and_values(file_cache):
    unicode_key = 'üñîçødé_këy'
    unicode_value = 'üñîçødé_vålüé'
    file_cache.set(unicode_key, unicode_value)
    assert file_cache.get(unicode_key) == unicode_value


def test_empty_string_as_key_and_value(file_cache):
    file_cache.set('', '')
    assert file_cache.get('') == ''


def test_none_as_value(file_cache):
    file_cache.set('none_key', None)
    assert file_cache.get('none_key') is None


def test_special_characters_in_key(file_cache):
    special_key = '!@#$%^&*()_+{}[]|\\:;"\'<>,.?/~`'
    file_cache.set(special_key, 'special_value')
    assert file_cache.get(special_key) == 'special_value'


def test_size_limit_with_empty_key():
    with tempfile.TemporaryDirectory() as temp_dir:
        cache = FileCache(temp_dir, size_limit=100)  # 100 bytes limit
        cache.set('', 'x' * 50)  # 50 bytes with empty key
        cache.set('key2', 'y' * 60)  # 60 bytes

        # This should cause the empty key to be evicted
        cache.set('key3', 'z' * 40)  # 40 bytes

        assert '' not in cache
        assert 'key2' in cache
        assert 'key3' in cache
        assert cache.get('key2') == 'y' * 60
        assert cache.get('key3') == 'z' * 40


# Add more tests as needed