whisper-local/tests/test_inserter.py

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

import pytest

from whisper_local.inserter._wayland import WaylandInserter


class TestWaylandInserter:
    @pytest.mark.asyncio
    @patch("whisper_local.inserter._wayland.asyncio.sleep", new_callable=AsyncMock)
    @patch("whisper_local.inserter._wayland.asyncio.create_subprocess_exec")
    async def test_insert_text_calls_tools_in_order(self, mock_exec, mock_sleep):
        mock_proc = AsyncMock()
        mock_proc.communicate.return_value = (b"alter clipboard", b"")
        mock_proc.returncode = 0
        mock_proc.wait = AsyncMock()
        mock_exec.return_value = mock_proc

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

        calls = mock_exec.call_args_list
        assert len(calls) == 4
        assert calls[0][0] == ("wl-paste", "--no-newline")
        assert calls[1][0] == ("wl-copy", "--", "Hallo Welt")
        assert calls[2][0][0] == "ydotool"
        assert calls[3][0] == ("wl-copy", "--", "alter clipboard")

    @pytest.mark.asyncio
    async def test_insert_empty_text_does_nothing(self):
        inserter = WaylandInserter()
        await inserter.insert("")

    @pytest.mark.asyncio
    @patch("whisper_local.inserter._wayland.asyncio.sleep", new_callable=AsyncMock)
    @patch("whisper_local.inserter._wayland.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()
            mock_proc.wait = AsyncMock()
            if call_count == 1:  # wl-paste
                mock_proc.communicate.return_value = (b"original", b"")
                mock_proc.returncode = 0
            elif call_count == 3:  # ydotool — fails
                mock_proc.wait.side_effect = OSError("ydotool not found")
            else:  # wl-copy calls
                mock_proc.returncode = 0
            return mock_proc

        mock_exec.side_effect = mock_create_proc

        inserter = WaylandInserter()
        with pytest.raises(OSError):
            await inserter.insert("Test")

        assert call_count == 4