"""PtyManager 테스트."""

from __future__ import annotations

from collections.abc import Sequence
from dataclasses import dataclass

import pytest

from lazy_enter.pty_manager import PtyManager


@dataclass
class Result:
    returncode: int
    stdout: str = ""


def test_initial_state():
    pty = PtyManager("claude")
    assert not pty.is_alive


def test_start_raises_when_tmux_session_missing(monkeypatch: pytest.MonkeyPatch):
    def fake_run(
        cmd: Sequence[str],
        check: bool,
        capture_output: bool,
        text: bool,
    ):
        assert list(cmd) == ["tmux", "has-session", "-t", "claude"]
        assert check is False
        assert capture_output is True
        assert text is True

        return Result(returncode=1)

    monkeypatch.setattr("lazy_enter.pty_manager.subprocess.run", fake_run)

    pty = PtyManager("claude")
    with pytest.raises(RuntimeError):
        pty.start()


def test_start_send_read_and_stop(monkeypatch: pytest.MonkeyPatch):
    calls: list[list[str]] = []

    def fake_run(
        cmd: Sequence[str],
        check: bool,
        capture_output: bool,
        text: bool,
    ):
        calls.append(list(cmd))
        assert check is False
        assert capture_output is True
        assert text is True

        if cmd[:4] == ["tmux", "has-session", "-t", "claude"]:
            return Result(returncode=0)
        if cmd[:7] == ["tmux", "capture-pane", "-p", "-t", "claude", "-S", "-200"]:
            # 첫 캡처는 baseline, 두 번째 캡처에서 한 줄 증가.
            if calls.count(list(cmd)) == 1:
                return Result(returncode=0, stdout="line1\nline2\n")
            return Result(returncode=0, stdout="line1\nline2\nline3\n")
        if cmd[:6] == ["tmux", "send-keys", "-t", "claude", "hello", "C-m"]:
            return Result(returncode=0)
        return Result(returncode=1)

    monkeypatch.setattr("lazy_enter.pty_manager.subprocess.run", fake_run)

    pty = PtyManager("claude")
    pty.start()
    assert pty.is_alive

    pty.send("hello")
    assert pty.read_output() == "line3"

    pty.stop()
    assert not pty.is_alive


def test_read_output_returns_empty_when_unchanged(monkeypatch: pytest.MonkeyPatch):
    def fake_run(
        cmd: Sequence[str],
        check: bool,
        capture_output: bool,
        text: bool,
    ):
        assert check is False
        assert capture_output is True
        assert text is True
        if cmd[:4] == ["tmux", "has-session", "-t", "claude"]:
            return Result(returncode=0)
        if cmd[:7] == ["tmux", "capture-pane", "-p", "-t", "claude", "-S", "-200"]:
            return Result(returncode=0, stdout="same\n")
        return Result(returncode=1)

    monkeypatch.setattr("lazy_enter.pty_manager.subprocess.run", fake_run)

    pty = PtyManager("claude")
    pty.start()
    assert pty.read_output() == ""