package tmuxedit

import (
	"fmt"
	"testing"
)

func TestCapturePane_Success(t *testing.T) {
	old := runCommand
	defer func() { runCommand = old }()
	runCommand = func(name string, args ...string) ([]byte, error) {
		if name == "tmux" && len(args) >= 3 && args[0] == "capture-pane" {
			return []byte("Claude Code v1.0\n> hello world\n"), nil
		}
		return nil, fmt.Errorf("unexpected: %s %v", name, args)
	}
	got, err := capturePane("%5")
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if got != "Claude Code v1.0\n> hello world" {
		t.Errorf("got %q, want trimmed content", got)
	}
}

func TestCapturePane_Error(t *testing.T) {
	old := runCommand
	defer func() { runCommand = old }()
	runCommand = func(string, ...string) ([]byte, error) {
		return nil, fmt.Errorf("pane not found")
	}
	_, err := capturePane("%999")
	if err == nil {
		t.Fatal("expected error for failed capture")
	}
}

func TestCapturePane_EmptyContent(t *testing.T) {
	old := runCommand
	defer func() { runCommand = old }()
	runCommand = func(string, ...string) ([]byte, error) {
		return []byte("\n\n"), nil
	}
	got, err := capturePane("%1")
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if got != "" {
		t.Errorf("got %q, want empty string", got)
	}
}