tests/test_ingest.py

"""Tests for IngestAgent."""
import pytest
import numpy as np

from formscout.agents.ingest import IngestAgent
from formscout.types import IngestResult


def _make_test_video(path, fps=30, n_frames=30, w=640, h=480):
    """Create a minimal test video using OpenCV."""
    import cv2
    out = cv2.VideoWriter(str(path), cv2.VideoWriter_fourcc(*"mp4v"), fps, (w, h))
    for _ in range(n_frames):
        out.write(np.zeros((h, w, 3), dtype=np.uint8))
    out.release()


class TestIngestAgent:
    def test_returns_typed_result(self, tmp_path):
        p = tmp_path / "test.mp4"
        _make_test_video(p)
        agent = IngestAgent()
        result = agent.run(str(p))
        assert isinstance(result, IngestResult)
        assert result.fps == pytest.approx(30.0, abs=2.0)
        assert len(result.frames) > 0
        assert result.width == 640
        assert result.height == 480
        assert result.confidence == 1.0

    def test_rejects_missing_file(self):
        agent = IngestAgent()
        result = agent.run("/nonexistent/path.mp4")
        assert result.confidence == 0.0
        assert "not found" in result.notes.lower()

    def test_result_is_frozen(self, tmp_path):
        p = tmp_path / "test.mp4"
        _make_test_video(p, n_frames=10, w=64, h=64)
        agent = IngestAgent()
        result = agent.run(str(p))
        with pytest.raises(Exception):
            result.fps = 999.0

    def test_caps_frames(self, tmp_path):
        """Verify frame cap works on long videos."""
        p = tmp_path / "long.mp4"
        _make_test_video(p, n_frames=600)
        agent = IngestAgent()
        result = agent.run(str(p))
        assert len(result.frames) <= 300