mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-27 04:20:09 +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
143 lines
5.0 KiB
Python
143 lines
5.0 KiB
Python
"""Tests for file history management."""
|
|
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
from dev_aci.editor.history import FileHistoryManager
|
|
|
|
|
|
def test_default_history_limit():
|
|
"""Test that default history limit is 5 entries."""
|
|
with tempfile.NamedTemporaryFile() as temp_file:
|
|
path = Path(temp_file.name)
|
|
manager = FileHistoryManager()
|
|
|
|
# Add 6 entries - this should trigger removal of the first entry
|
|
for i in range(6):
|
|
manager.add_history(path, f"content{i}")
|
|
|
|
# Get the metadata
|
|
metadata = manager.get_metadata(path)
|
|
assert len(metadata["entries"]) == 5 # Should only keep last 5 entries
|
|
# First entry should be content1, last should be content5
|
|
assert manager.get_all_history(path)[0].startswith("content1")
|
|
assert manager.get_all_history(path)[-1].startswith("content5")
|
|
|
|
|
|
def test_history_keys_are_unique():
|
|
"""Test that history keys remain unique even after removing old entries."""
|
|
with tempfile.NamedTemporaryFile() as temp_file:
|
|
path = Path(temp_file.name)
|
|
manager = FileHistoryManager(max_history_per_file=2)
|
|
|
|
# Add 3 entries - this should trigger removal of the first entry
|
|
manager.add_history(path, "content1")
|
|
manager.add_history(path, "content2")
|
|
manager.add_history(path, "content3")
|
|
|
|
# Get the metadata
|
|
metadata = manager.get_metadata(path)
|
|
assert len(metadata["entries"]) == 2 # Should only keep last 2 entries
|
|
|
|
# Keys should be unique and sequential
|
|
keys = metadata["entries"]
|
|
assert len(set(keys)) == len(keys) # All keys should be unique
|
|
assert sorted(keys) == keys # Keys should be sequential
|
|
|
|
# Add another entry
|
|
manager.add_history(path, "content4")
|
|
new_metadata = manager.get_metadata(path)
|
|
new_keys = new_metadata["entries"]
|
|
|
|
# New key should be greater than all previous keys
|
|
assert min(new_keys) > min(keys)
|
|
assert len(set(new_keys)) == len(new_keys) # All keys should still be unique
|
|
|
|
|
|
def test_history_counter_persists():
|
|
"""Test that history counter persists across manager instances."""
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
path = Path(temp_dir) / "test.txt"
|
|
path.write_text("initial")
|
|
|
|
# First manager instance
|
|
manager1 = FileHistoryManager(history_dir=Path(temp_dir))
|
|
manager1.add_history(path, "content1")
|
|
manager1.add_history(path, "content2")
|
|
|
|
# Second manager instance using same directory
|
|
manager2 = FileHistoryManager(history_dir=Path(temp_dir))
|
|
manager2.add_history(path, "content3")
|
|
|
|
# Get metadata
|
|
metadata = manager2.get_metadata(path)
|
|
keys = metadata["entries"]
|
|
|
|
# Keys should be sequential even across instances
|
|
assert len(set(keys)) == len(keys) # All keys should be unique
|
|
assert sorted(keys) == keys # Keys should be sequential
|
|
|
|
|
|
def test_clear_history_resets_counter():
|
|
"""Test that clearing history resets the counter."""
|
|
with tempfile.NamedTemporaryFile() as temp_file:
|
|
path = Path(temp_file.name)
|
|
manager = FileHistoryManager()
|
|
|
|
# Add some entries
|
|
manager.add_history(path, "content1")
|
|
manager.add_history(path, "content2")
|
|
|
|
# Clear history
|
|
manager.clear_history(path)
|
|
|
|
# Counter should be reset
|
|
metadata = manager.get_metadata(path)
|
|
assert metadata["counter"] == 0
|
|
|
|
# Adding new entries should start from 0
|
|
manager.add_history(path, "new_content")
|
|
metadata = manager.get_metadata(path)
|
|
assert len(metadata["entries"]) == 1
|
|
assert metadata["entries"][0] == 0 # First key should be 0
|
|
|
|
|
|
def test_pop_last_history_removes_entry():
|
|
"""Test that pop_last_history removes the latest entry."""
|
|
with tempfile.NamedTemporaryFile() as temp_file:
|
|
path = Path(temp_file.name)
|
|
manager = FileHistoryManager()
|
|
|
|
# Add some entries
|
|
manager.add_history(path, "content1")
|
|
manager.add_history(path, "content2")
|
|
manager.add_history(path, "content3")
|
|
|
|
# Pop the last history entry
|
|
last_entry = manager.pop_last_history(path)
|
|
assert last_entry == "content3"
|
|
|
|
# Check that the entry has been removed
|
|
metadata = manager.get_metadata(path)
|
|
assert len(metadata["entries"]) == 2
|
|
|
|
# Pop the last history entry again
|
|
last_entry = manager.pop_last_history(path)
|
|
assert last_entry == "content2"
|
|
|
|
# Check that the entry has been removed
|
|
metadata = manager.get_metadata(path)
|
|
assert len(metadata["entries"]) == 1
|
|
|
|
# Pop the last history entry one more time
|
|
last_entry = manager.pop_last_history(path)
|
|
assert last_entry == "content1"
|
|
|
|
# Check that all entries have been removed
|
|
metadata = manager.get_metadata(path)
|
|
assert len(metadata["entries"]) == 0
|
|
|
|
# Try to pop last history when there are no entries
|
|
last_entry = manager.pop_last_history(path)
|
|
assert last_entry is None
|