tests/test_space.py

"""Tests for the Gradio space helpers (without launching the UI)."""

from __future__ import annotations

import numpy as np

from signbridge.space import (
    _capture_sign,
    _clear,
    _format_history,
    _new_session,
    _speak,
)


class TestSession:
    def test_new_session_is_empty(self) -> None:
        s = _new_session()
        assert s.sign_history == []
        assert s.last_sentence == ""
        assert s.last_audio_path is None


class TestFormatHistory:
    def test_empty_returns_hint(self) -> None:
        out = _format_history([])
        assert "no signs" in out.lower()

    def test_renders_signs_as_code(self) -> None:
        out = _format_history(["A", "B", "hello"])
        assert "`A`" in out
        assert "`B`" in out
        assert "`hello`" in out


class TestCaptureSign:
    def test_no_frame(self) -> None:
        s = _new_session()
        msg, history, new_state = _capture_sign(None, s)
        assert "no webcam" in msg.lower()
        assert new_state.sign_history == []

    def test_no_provider_says_unrecognized(self) -> None:
        # Without API keys, recognizer returns ("", 0.0) → UI shows hint.
        s = _new_session()
        frame = np.zeros((32, 32, 3), dtype=np.uint8)
        msg, history, new_state = _capture_sign(frame, s)
        assert "couldn't recognise" in msg.lower() or "couldn't recognize" in msg.lower()
        assert new_state.sign_history == []


class TestSpeak:
    def test_empty_history(self) -> None:
        s = _new_session()
        sentence, audio, new_state = _speak(s)
        assert "no signs" in sentence.lower()
        assert audio is None

    def test_with_history_invokes_composer_and_tts(self) -> None:
        s = _new_session()
        s.sign_history = ["L", "U", "C", "A", "S"]
        sentence, audio, new_state = _speak(s)
        # Naive joiner path
        assert "Lucas" in sentence
        # TTS produces a silent-stub WAV path
        assert audio is not None and audio != ""


class TestClear:
    def test_resets_state(self) -> None:
        s = _new_session()
        s.sign_history = ["A", "B"]
        s.last_sentence = "AB."
        s.last_audio_path = "/tmp/foo.wav"

        latest, history, sentence, audio, new_state = _clear(s)
        assert latest == ""
        assert "no signs" in history.lower()
        assert sentence == ""
        assert audio is None
        assert new_state.sign_history == []
        assert new_state.last_sentence == ""
        assert new_state.last_audio_path is None


class TestSampleFramesFromVideo:
    def test_returns_n_frames(self, tmp_path):
        import cv2
        import numpy as np

        from signbridge.space import _sample_frames_from_video

        video_path = str(tmp_path / "synthetic.mp4")
        h, w, fps = 64, 64, 30
        fourcc = cv2.VideoWriter_fourcc(*"mp4v")
        writer = cv2.VideoWriter(video_path, fourcc, fps, (w, h))
        try:
            for i in range(10):
                writer.write(np.full((h, w, 3), 25 * i, dtype=np.uint8))
        finally:
            writer.release()

        frames = _sample_frames_from_video(video_path, n_frames=4)
        assert len(frames) == 4
        for fr in frames:
            assert fr.shape == (h, w, 3)
            assert fr.dtype == np.uint8

    def test_zero_frames_for_missing_file(self, tmp_path):
        from signbridge.space import _sample_frames_from_video

        non_existent = str(tmp_path / "nope.mp4")
        # Must NOT raise; must return [] so the UI can fall back gracefully.
        frames = _sample_frames_from_video(non_existent, n_frames=4)
        assert frames == []

    def test_none_path_returns_empty(self):
        from signbridge.space import _sample_frames_from_video

        assert _sample_frames_from_video(None, n_frames=4) == []
        assert _sample_frames_from_video("", n_frames=4) == []