feat(tools): stream shell_run output live into the TUI

internal/tools/shell_test.go

@@ -0,0 +1,71 @@
+package tools
+
+import (
+	"context"
+	"encoding/json"
+	"strings"
+	"sync"
+	"testing"
+)
+
+func TestShellRunStreamsOutputWhileRunning(t *testing.T) {
+	tool := NewShellRunTool(t.TempDir())
+	var mu sync.Mutex
+	var chunks []string
+	output, err := tool.ExecuteStream(context.Background(), json.RawMessage(`{"command":"printf 'a\\n'; sleep 0.05; printf 'b\\n'"}`), func(chunk string) error {
+		mu.Lock()
+		chunks = append(chunks, chunk)
+		mu.Unlock()
+		return nil
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !strings.Contains(output, "a") || !strings.Contains(output, "b") {
+		t.Fatalf("output = %q", output)
+	}
+	mu.Lock()
+	joined := strings.Join(chunks, "")
+	mu.Unlock()
+	if len(chunks) == 0 {
+		t.Fatal("expected at least one streamed chunk")
+	}
+	if !strings.Contains(joined, "a") || !strings.Contains(joined, "b") {
+		t.Fatalf("streamed chunks = %q", joined)
+	}
+}
+
+func TestShellRunExecuteCollectsFullOutput(t *testing.T) {
+	tool := NewShellRunTool(t.TempDir())
+	output, err := tool.Execute(context.Background(), json.RawMessage(`{"command":"printf 'hello world'"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if strings.TrimSpace(output) != "hello world" {
+		t.Fatalf("output = %q", output)
+	}
+}
+
+func TestShellRunStreamsStderrAndStdout(t *testing.T) {
+	tool := NewShellRunTool(t.TempDir())
+	var mu sync.Mutex
+	var chunks []string
+	output, err := tool.ExecuteStream(context.Background(), json.RawMessage(`{"command":"echo out; echo err >&2"}`), func(chunk string) error {
+		mu.Lock()
+		chunks = append(chunks, chunk)
+		mu.Unlock()
+		return nil
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !strings.Contains(output, "out") || !strings.Contains(output, "err") {
+		t.Fatalf("output = %q", output)
+	}
+	mu.Lock()
+	joined := strings.Join(chunks, "")
+	mu.Unlock()
+	if !strings.Contains(joined, "out") || !strings.Contains(joined, "err") {
+		t.Fatalf("streamed chunks = %q", joined)
+	}
+}