import asyncio
from unittest.mock import AsyncMock, MagicMock, patch, call

import pytest

from whisper_local.inserter import Inserter


class TestInserter:
    @pytest.mark.asyncio
    @patch("whisper_local.inserter.asyncio.sleep", new_callable=AsyncMock)
    @patch("whisper_local.inserter.asyncio.create_subprocess_exec")
    async def test_insert_text_calls_tools_in_order(self, mock_exec, mock_sleep):
        # Mock für alle subprocess-Aufrufe
        mock_proc = AsyncMock()
        mock_proc.communicate.return_value = (b"alter clipboard", b"")
        mock_proc.returncode = 0
        mock_exec.return_value = mock_proc

        inserter = Inserter()
        await inserter.insert("Hallo Welt")

        # Prüfe dass wl-paste, wl-copy, wtype, wl-copy aufgerufen werden
        calls = mock_exec.call_args_list
        assert len(calls) == 4
        assert calls[0][0][0] == "wl-paste"                    # Clipboard sichern
        assert calls[1][0] == ("wl-copy", "Hallo Welt")        # Text in Clipboard
        assert calls[2][0][0] == "wtype"                        # Ctrl+V simulieren
        assert calls[3][0] == ("wl-copy", "alter clipboard")   # Clipboard wiederherstellen

    @pytest.mark.asyncio
    @patch("whisper_local.inserter.asyncio.create_subprocess_exec")
    async def test_insert_empty_text_does_nothing(self, mock_exec):
        inserter = Inserter()
        await inserter.insert("")

        mock_exec.assert_not_called()

    @pytest.mark.asyncio
    @patch("whisper_local.inserter.asyncio.sleep", new_callable=AsyncMock)
    @patch("whisper_local.inserter.asyncio.create_subprocess_exec")
    async def test_clipboard_restore_on_paste_failure(self, mock_exec, mock_sleep):
        call_count = 0

        async def mock_create_proc(*args, **kwargs):
            nonlocal call_count
            call_count += 1
            mock_proc = AsyncMock()
            if call_count == 1:  # wl-paste
                mock_proc.communicate.return_value = (b"original", b"")
                mock_proc.returncode = 0
            elif call_count == 2:  # wl-copy
                mock_proc.communicate.return_value = (b"", b"")
                mock_proc.returncode = 0
            elif call_count == 3:  # wtype — fails
                mock_proc.communicate.return_value = (b"", b"error")
                mock_proc.returncode = 1
            else:  # wl-copy restore
                mock_proc.communicate.return_value = (b"", b"")
                mock_proc.returncode = 0
            return mock_proc

        mock_exec.side_effect = mock_create_proc

        inserter = Inserter()
        await inserter.insert("Test")

        # Clipboard muss trotzdem wiederhergestellt werden
        assert call_count == 4