120 lines
4.6 KiB
Python
120 lines
4.6 KiB
Python
import pytest
|
|
import os
|
|
import shutil
|
|
from datetime import datetime, timedelta
|
|
from src.log_registry import LogRegistry
|
|
from src.log_pruner import LogPruner
|
|
|
|
class TestLogPruningHeuristic:
|
|
@pytest.fixture
|
|
def temp_logs_dir(self, tmp_path):
|
|
# Create a structure: project_root/logs/sessions
|
|
project_root = tmp_path
|
|
logs_dir = project_root / "logs" / "sessions"
|
|
logs_dir.mkdir(parents=True)
|
|
return logs_dir
|
|
|
|
@pytest.fixture
|
|
def registry_path(self, temp_logs_dir):
|
|
return temp_logs_dir / "log_registry.toml"
|
|
|
|
def test_get_old_non_whitelisted_sessions_includes_empty_sessions(self, registry_path):
|
|
registry = LogRegistry(str(registry_path))
|
|
|
|
# 1. Old non-whitelisted session
|
|
old_time = datetime.now() - timedelta(days=2)
|
|
registry.register_session("old_session", "logs/old_session", old_time)
|
|
|
|
# 2. New session with 0 messages (empty)
|
|
new_time = datetime.now()
|
|
registry.register_session("empty_session", "logs/empty_session", new_time)
|
|
registry.update_session_metadata("empty_session", message_count=0, errors=0, size_kb=0, whitelisted=False, reason="")
|
|
|
|
# 3. New session with messages (not empty)
|
|
registry.register_session("active_session", "logs/active_session", new_time)
|
|
registry.update_session_metadata("active_session", message_count=5, errors=0, size_kb=10, whitelisted=False, reason="")
|
|
|
|
# 4. Whitelisted session
|
|
registry.register_session("starred_session", "logs/starred_session", old_time)
|
|
registry.update_session_metadata("starred_session", message_count=0, errors=0, size_kb=0, whitelisted=True, reason="starred")
|
|
|
|
cutoff = datetime.now() - timedelta(days=1)
|
|
sessions = registry.get_old_non_whitelisted_sessions(cutoff)
|
|
|
|
session_ids = [s['session_id'] for s in sessions]
|
|
assert "old_session" in session_ids
|
|
assert "empty_session" in session_ids
|
|
assert "active_session" not in session_ids
|
|
assert "starred_session" not in session_ids
|
|
|
|
def test_get_old_non_whitelisted_sessions_includes_sessions_without_metadata(self, registry_path):
|
|
registry = LogRegistry(str(registry_path))
|
|
new_time = datetime.now()
|
|
# Session registered but update_auto_whitelist_status not called yet
|
|
registry.register_session("no_meta_session", "logs/no_meta_session", new_time)
|
|
|
|
cutoff = datetime.now() - timedelta(days=1)
|
|
sessions = registry.get_old_non_whitelisted_sessions(cutoff)
|
|
|
|
session_ids = [s['session_id'] for s in sessions]
|
|
assert "no_meta_session" in session_ids
|
|
|
|
def test_prune_handles_relative_paths_starting_with_logs(self, temp_logs_dir, registry_path):
|
|
# Project structure: project_root/logs/sessions
|
|
project_root = temp_logs_dir.parent.parent
|
|
|
|
# Create a session dir in project_root/logs/session_1
|
|
session_id = "session_1"
|
|
session_rel_path = os.path.join("logs", session_id)
|
|
session_abs_path = project_root / session_rel_path
|
|
session_abs_path.mkdir(parents=True)
|
|
(session_abs_path / "comms.log").write_text("empty")
|
|
|
|
registry = LogRegistry(str(registry_path))
|
|
registry.register_session(session_id, session_rel_path, datetime.now())
|
|
# No metadata = empty
|
|
|
|
pruner = LogPruner(registry, str(temp_logs_dir))
|
|
pruner.prune(max_age_days=1, min_size_kb=10)
|
|
|
|
assert not session_abs_path.exists()
|
|
assert session_id not in registry.data
|
|
|
|
def test_prune_removes_empty_sessions_regardless_of_age(self, temp_logs_dir, registry_path):
|
|
# Mock project root resolution in LogPruner
|
|
project_root = temp_logs_dir.parent.parent
|
|
|
|
session_id = "new_empty"
|
|
session_rel_path = os.path.join("logs", session_id)
|
|
session_abs_path = project_root / session_rel_path
|
|
session_abs_path.mkdir(parents=True)
|
|
|
|
registry = LogRegistry(str(registry_path))
|
|
registry.register_session(session_id, session_rel_path, datetime.now())
|
|
registry.update_session_metadata(session_id, message_count=0, errors=0, size_kb=0, whitelisted=False, reason="")
|
|
|
|
pruner = LogPruner(registry, str(temp_logs_dir))
|
|
# max_age_days=10, but should still prune because it's empty
|
|
pruner.prune(max_age_days=10, min_size_kb=10)
|
|
|
|
assert not session_abs_path.exists()
|
|
assert session_id not in registry.data
|
|
|
|
def test_prune_removes_sessions_without_metadata_regardless_of_age(self, temp_logs_dir, registry_path):
|
|
project_root = temp_logs_dir.parent.parent
|
|
|
|
session_id = "new_no_meta"
|
|
session_rel_path = os.path.join("logs", session_id)
|
|
session_abs_path = project_root / session_rel_path
|
|
session_abs_path.mkdir(parents=True)
|
|
|
|
registry = LogRegistry(str(registry_path))
|
|
registry.register_session(session_id, session_rel_path, datetime.now())
|
|
# No metadata update
|
|
|
|
pruner = LogPruner(registry, str(temp_logs_dir))
|
|
pruner.prune(max_age_days=10, min_size_kb=10)
|
|
|
|
assert not session_abs_path.exists()
|
|
assert session_id not in registry.data
|