mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-29 18:46:59 +00:00
a61edf952f
Adds cleanup_audio_cache() to gateway/platforms/base.py and wires it into _start_gateway_housekeeping() in gateway/run.py, matching the existing image/document cache cleanup pattern. Re-implements the fix proposed in #16473, which targeted the now-deprecated _start_cron_ticker() and has since stalled with a merge conflict against main. _start_gateway_housekeeping()'s docstring already claimed audio cleanup ("prunes the image/audio/document cache"); this makes the code match it. Co-authored-by: nftpoetrist <264138787+nftpoetrist@users.noreply.github.com>
129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
"""
|
|
Tests for audio cache utilities in gateway/platforms/base.py.
|
|
|
|
Covers: get_audio_cache_dir, cache_audio_from_bytes, cleanup_audio_cache.
|
|
"""
|
|
|
|
import os
|
|
import time
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from gateway.platforms.base import (
|
|
cache_audio_from_bytes,
|
|
cleanup_audio_cache,
|
|
get_audio_cache_dir,
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Fixture: redirect AUDIO_CACHE_DIR to a temp directory for every test
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _redirect_cache(tmp_path, monkeypatch):
|
|
"""Point the module-level AUDIO_CACHE_DIR to a fresh tmp_path."""
|
|
monkeypatch.setattr(
|
|
"gateway.platforms.base.AUDIO_CACHE_DIR", tmp_path / "audio_cache"
|
|
)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# TestGetAudioCacheDir
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestGetAudioCacheDir:
|
|
def test_creates_directory(self):
|
|
cache_dir = get_audio_cache_dir()
|
|
assert cache_dir.exists()
|
|
assert cache_dir.is_dir()
|
|
|
|
def test_returns_existing_directory(self):
|
|
first = get_audio_cache_dir()
|
|
second = get_audio_cache_dir()
|
|
assert first == second
|
|
assert first.exists()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# TestCacheAudioFromBytes
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestCacheAudioFromBytes:
|
|
def test_basic_caching(self):
|
|
data = b"fake-ogg-bytes"
|
|
path = cache_audio_from_bytes(data)
|
|
assert os.path.exists(path)
|
|
assert Path(path).read_bytes() == data
|
|
|
|
def test_default_extension(self):
|
|
path = cache_audio_from_bytes(b"data")
|
|
assert path.endswith(".ogg")
|
|
|
|
def test_custom_extension(self):
|
|
path = cache_audio_from_bytes(b"data", ext=".mp3")
|
|
assert path.endswith(".mp3")
|
|
|
|
def test_unique_filenames(self):
|
|
p1 = cache_audio_from_bytes(b"a")
|
|
p2 = cache_audio_from_bytes(b"b")
|
|
assert p1 != p2
|
|
|
|
def test_file_written_inside_cache_dir(self):
|
|
path = cache_audio_from_bytes(b"data")
|
|
cache_dir = get_audio_cache_dir()
|
|
assert Path(path).resolve().is_relative_to(cache_dir.resolve())
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# TestCleanupAudioCache
|
|
# ---------------------------------------------------------------------------
|
|
|
|
class TestCleanupAudioCache:
|
|
def test_removes_old_files(self):
|
|
cache_dir = get_audio_cache_dir()
|
|
old_file = cache_dir / "old.ogg"
|
|
old_file.write_text("old")
|
|
# Set modification time to 48 hours ago
|
|
old_mtime = time.time() - 48 * 3600
|
|
os.utime(old_file, (old_mtime, old_mtime))
|
|
|
|
removed = cleanup_audio_cache(max_age_hours=24)
|
|
assert removed == 1
|
|
assert not old_file.exists()
|
|
|
|
def test_keeps_recent_files(self):
|
|
cache_dir = get_audio_cache_dir()
|
|
recent = cache_dir / "recent.ogg"
|
|
recent.write_text("fresh")
|
|
|
|
removed = cleanup_audio_cache(max_age_hours=24)
|
|
assert removed == 0
|
|
assert recent.exists()
|
|
|
|
def test_returns_removed_count(self):
|
|
cache_dir = get_audio_cache_dir()
|
|
old_time = time.time() - 48 * 3600
|
|
for i in range(3):
|
|
f = cache_dir / f"old_{i}.ogg"
|
|
f.write_text("old")
|
|
os.utime(f, (old_time, old_time))
|
|
|
|
removed = cleanup_audio_cache(max_age_hours=24)
|
|
assert removed == 3
|
|
|
|
def test_ignores_subdirectories(self):
|
|
cache_dir = get_audio_cache_dir()
|
|
subdir = cache_dir / "subdir"
|
|
subdir.mkdir()
|
|
old_time = time.time() - 48 * 3600
|
|
os.utime(subdir, (old_time, old_time))
|
|
|
|
# Should not raise or attempt to remove the directory as a file.
|
|
removed = cleanup_audio_cache(max_age_hours=24)
|
|
assert removed == 0
|
|
assert subdir.exists()
|
|
|
|
def test_empty_cache_returns_zero(self):
|
|
removed = cleanup_audio_cache(max_age_hours=24)
|
|
assert removed == 0
|