from __future__ import annotations

import unittest
from pathlib import Path

from owner_voice_pet.assets import REQUIRED_ASSET_STATES, validate_pet_assets
from owner_voice_pet.models import PipelineState
from owner_voice_pet.ui import ConsolePetWindow, PetStateController, PetVisualState


class UiAssetTests(unittest.TestCase):
    def test_pipeline_state_maps_to_visual_state(self) -> None:
        controller = PetStateController()
        self.assertEqual(controller.apply_pipeline_state(PipelineState.SPEECH_DETECTING), PetVisualState.LISTENING)
        self.assertEqual(controller.apply_pipeline_state(PipelineState.THINKING), PetVisualState.THINKING)
        self.assertEqual(controller.apply_pipeline_state(PipelineState.ERROR_RECOVERING), PetVisualState.ERROR)

    def test_console_window_fallback_records_history(self) -> None:
        window = ConsolePetWindow()
        self.assertEqual(window.update(PipelineState.SPEAKING), PetVisualState.SPEAKING)
        self.assertEqual(window.history, [PetVisualState.SPEAKING])

    def test_project_pet_assets_are_rgba_and_complete(self) -> None:
        infos = validate_pet_assets(Path("assets/pet"))
        self.assertEqual(len(infos), len(REQUIRED_ASSET_STATES))
        self.assertTrue(all(info.width == 256 and info.height == 256 for info in infos))
        self.assertTrue(all(info.has_transparency for info in infos))


if __name__ == "__main__":
    unittest.main()