summaryrefslogtreecommitdiff
path: root/internal/tmuxedit/capture_test.go
blob: 40d0e98c490510d45ef3b97da73e94354bda80d3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
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)
	}
}