stdin_context_test.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"errors"
6	"os"
7	"testing"
8)
9
10// An orchestrator waiting for a piped prompt must stop when its signal
11// context ends instead of blocking on an open pipe.
12func TestReadAllStdinContextReturnsOnCancel(t *testing.T) {
13	r, w, err := os.Pipe()
14	if err != nil {
15		t.Fatal(err)
16	}
17	t.Cleanup(func() { _ = w.Close(); _ = r.Close() })
18	old := os.Stdin
19	os.Stdin = r
20	t.Cleanup(func() { os.Stdin = old })
21
22	ctx, cancel := context.WithCancel(context.Background())
23	cancel()
24	if _, err := readAllStdinContext(ctx); !errors.Is(err, context.Canceled) {
25		t.Fatalf("err = %v, want context.Canceled", err)
26	}
27}