bash_unix_test.go RawHistoryFolder

1//go:build !windows
2
3package tools
4
5import (
6	"context"
7	"errors"
8	"fmt"
9	"os"
10	"path/filepath"
11	"strconv"
12	"syscall"
13	"testing"
14	"time"
15)
16
17func TestBashTimeoutKillsDescendant(t *testing.T) {
18	pidPath := filepath.Join(t.TempDir(), "child.pid")
19	command := fmt.Sprintf("sleep 30 & child=$!; printf '%%s' \"$child\" > %s; wait", shellQuoteForTest(pidPath))
20	tool := &BashTool{CWD: t.TempDir()}
21
22	res, err := tool.Execute(context.Background(), mustJSON(t, map[string]any{
23		"command": command,
24		"timeout": 1,
25	}), nil)
26	if err != nil {
27		t.Fatal(err)
28	}
29	if !res.IsError {
30		t.Fatal("timed-out command should be an error")
31	}
32
33	data, err := os.ReadFile(pidPath)
34	if err != nil {
35		t.Fatalf("read child pid: %v", err)
36	}
37	childPID, err := strconv.Atoi(string(data))
38	if err != nil || childPID <= 0 {
39		t.Fatalf("child pid = %q: %v", data, err)
40	}
41
42	deadline := time.Now().Add(2 * time.Second)
43	for time.Now().Before(deadline) && bashProcessExists(childPID) {
44		time.Sleep(10 * time.Millisecond)
45	}
46	if bashProcessExists(childPID) {
47		t.Fatalf("bash descendant pid %d survived timeout cancellation", childPID)
48	}
49}
50
51func bashProcessExists(pid int) bool {
52	err := syscall.Kill(pid, 0)
53	return err == nil || errors.Is(err, syscall.EPERM)
54}