47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
from src.summary_cache import SummaryCache, get_file_hash
|
|
|
|
def test_get_file_hash():
|
|
content = "hello world"
|
|
# sha256 of "hello world"
|
|
expected = "b94d27b9934d3e08a52e52d7da7dabfac484efe37a5380ee9088f7ace2efcde9"
|
|
assert get_file_hash(content) == expected
|
|
|
|
def test_summary_cache():
|
|
cache_dir = Path(".test_cache")
|
|
if cache_dir.exists():
|
|
shutil.rmtree(cache_dir)
|
|
cache_file = cache_dir / "cache.json"
|
|
|
|
cache = SummaryCache(str(cache_file))
|
|
|
|
file_path = "test.py"
|
|
content = "print('hello')"
|
|
content_hash = get_file_hash(content)
|
|
summary = "**Python** - 1 lines"
|
|
|
|
# Test empty cache
|
|
assert cache.get_summary(file_path, content_hash) is None
|
|
|
|
# Test set and get
|
|
cache.set_summary(file_path, content_hash, summary)
|
|
assert cache.get_summary(file_path, content_hash) == summary
|
|
|
|
# Test cache invalidation
|
|
assert cache.get_summary(file_path, "different_hash") is None
|
|
|
|
# Test persistence
|
|
cache2 = SummaryCache(str(cache_file))
|
|
assert cache2.get_summary(file_path, content_hash) == summary
|
|
|
|
# Cleanup
|
|
if cache_dir.exists():
|
|
shutil.rmtree(cache_dir)
|
|
|
|
if __name__ == "__main__":
|
|
test_get_file_hash()
|
|
test_summary_cache()
|
|
print("Tests passed!")
|