96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
import pytest
|
|
from pathlib import Path
|
|
from typing import Generator
|
|
from src import session_logger
|
|
from src import paths
|
|
|
|
@pytest.fixture
|
|
def temp_session_setup(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> Generator[tuple[Path, Path], None, None]:
|
|
# Ensure session is closed and state is reset
|
|
session_logger.close_session()
|
|
monkeypatch.setattr(session_logger, "_comms_fh", None)
|
|
monkeypatch.setattr(session_logger, "_session_dir", None)
|
|
monkeypatch.setattr(session_logger, "_seq", 0)
|
|
monkeypatch.setattr(session_logger, "_output_seq", 0)
|
|
|
|
log_dir = tmp_path / "logs"
|
|
scripts_dir = tmp_path / "scripts" / "generated"
|
|
log_dir.mkdir(parents=True, exist_ok=True)
|
|
scripts_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
monkeypatch.setattr(paths, "get_logs_dir", lambda: log_dir)
|
|
monkeypatch.setattr(paths, "get_scripts_dir", lambda: scripts_dir)
|
|
|
|
yield log_dir, scripts_dir
|
|
|
|
# Cleanup
|
|
session_logger.close_session()
|
|
|
|
def test_session_directory_and_subdirectories_creation(temp_session_setup: tuple[Path, Path]) -> None:
|
|
log_dir, _ = temp_session_setup
|
|
session_logger.open_session(label="opt-test")
|
|
|
|
# Find the session directory
|
|
session_dirs = [d for d in log_dir.iterdir() if d.is_dir()]
|
|
assert len(session_dirs) == 1
|
|
session_dir = session_dirs[0]
|
|
|
|
assert (session_dir / "scripts").exists()
|
|
assert (session_dir / "outputs").exists()
|
|
assert (session_dir / "comms.log").exists()
|
|
assert (session_dir / "toolcalls.log").exists()
|
|
|
|
def test_log_tool_call_saves_in_session_scripts(temp_session_setup: tuple[Path, Path]) -> None:
|
|
log_dir, _ = temp_session_setup
|
|
session_logger.open_session(label="tool-call-test")
|
|
|
|
# Find the session directory
|
|
session_dir = next(d for d in log_dir.iterdir() if d.is_dir())
|
|
scripts_subdir = session_dir / "scripts"
|
|
|
|
script_content = "Write-Host 'Hello from test'"
|
|
result_content = "Success"
|
|
|
|
# Call log_tool_call with script_path=None
|
|
ps1_path_str = session_logger.log_tool_call(script_content, result_content, None)
|
|
assert ps1_path_str is not None
|
|
|
|
ps1_path = Path(ps1_path_str)
|
|
assert ps1_path.parent == scripts_subdir
|
|
assert ps1_path.name == "script_0001.ps1"
|
|
assert ps1_path.read_text(encoding="utf-8") == script_content
|
|
|
|
# Verify second call increments sequence
|
|
ps1_path_str_2 = session_logger.log_tool_call("Get-Date", "2026-03-08", None)
|
|
assert ps1_path_str_2 is not None
|
|
assert Path(ps1_path_str_2).name == "script_0002.ps1"
|
|
|
|
def test_log_tool_output_saves_in_session_outputs(temp_session_setup: tuple[Path, Path]) -> None:
|
|
log_dir, _ = temp_session_setup
|
|
session_logger.open_session(label="output-test")
|
|
|
|
# Find the session directory
|
|
session_dir = next(d for d in log_dir.iterdir() if d.is_dir())
|
|
outputs_subdir = session_dir / "outputs"
|
|
|
|
output_content = "This is some tool output content."
|
|
|
|
# Call log_tool_output
|
|
output_path_str = session_logger.log_tool_output(output_content)
|
|
assert output_path_str is not None
|
|
|
|
output_path = Path(output_path_str)
|
|
assert output_path.parent == outputs_subdir
|
|
assert output_path.name == "output_0001.txt"
|
|
assert output_path.read_text(encoding="utf-8") == output_content
|
|
|
|
# Verify second call increments sequence
|
|
output_path_str_2 = session_logger.log_tool_output("More content")
|
|
assert output_path_str_2 is not None
|
|
assert Path(output_path_str_2).name == "output_0002.txt"
|
|
|
|
def test_log_tool_output_returns_none_if_no_session(temp_session_setup: tuple[Path, Path]) -> None:
|
|
# We don't call open_session here
|
|
output_path_str = session_logger.log_tool_output("Should not save")
|
|
assert output_path_str is None
|