feat: execute durable client commands through supervisor

2026-09-06 12:18:15 +00:00
parent 3f84d3b2f1
commit 56b15c7f4f
41 changed files with 4272 additions and 66 deletions
@@ -0,0 +1,93 @@
+//go:build !windows
+
+package windows
+
+import (
+	"context"
+	"crypto/sha256"
+	"errors"
+	"io"
+	"strings"
+	"testing"
+	"time"
+
+	rvboxv1 "github.com/rvbox/rvbox/gen/go/rvbox/v1"
+	"github.com/rvbox/rvbox/internal/client/supervisor"
+	"github.com/rvbox/rvbox/internal/domain"
+)
+
+func TestPortableSupervisorCapturesOutputAndSupportsStdin_HP_SUPERVISOR_03(t *testing.T) {
+	t.Parallel()
+	manager, err := NewSupervisor(NativeOptions{MaxOutputChunk: 8, WindowsTermGrace: 10 * time.Millisecond})
+	if err != nil {
+		t.Fatal(err)
+	}
+	issue, err := domain.ParseUUIDv7("019c46f1-1d02-7000-8000-0000000000c1")
+	if err != nil {
+		t.Fatal(err)
+	}
+	process, err := manager.Start(context.Background(), supervisor.StartSpec{IssueUUID: issue, CommandRevision: 1, WorkingDirectory: t.TempDir(), Execution: &rvboxv1.ExecutionSpec{ShellType: rvboxv1.ShellType_SHELL_SH, Source: &rvboxv1.ExecutionSpec_CommandText{CommandText: "read line; printf 'out:%s\\n' \"$line\"; printf 'err\\n' >&2"}}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := process.WriteStdin(context.Background(), []byte("hello"), true); err != nil {
+		t.Fatal(err)
+	}
+	if err := process.CloseStdin(context.Background()); err != nil {
+		t.Fatal(err)
+	}
+	status, err := process.Wait(context.Background())
+	if err != nil || status.Code != 0 {
+		t.Fatalf("wait = %+v, %v", status, err)
+	}
+	var output strings.Builder
+	streams := map[rvboxv1.StreamKind]bool{}
+	for {
+		chunk, err := process.ReadOutput(context.Background())
+		if errors.Is(err, io.EOF) {
+			break
+		}
+		if err != nil {
+			t.Fatal(err)
+		}
+		output.Write(chunk.Data)
+		streams[chunk.Stream] = true
+	}
+	if !strings.Contains(output.String(), "out:hello\n") || !strings.Contains(output.String(), "err\n") || !streams[rvboxv1.StreamKind_STREAM_STDOUT] || !streams[rvboxv1.StreamKind_STREAM_STDERR] {
+		t.Fatalf("captured output = %q streams=%v", output.String(), streams)
+	}
+	second, err := manager.Start(context.Background(), supervisor.StartSpec{IssueUUID: issue, CommandRevision: 1, WorkingDirectory: t.TempDir(), Execution: &rvboxv1.ExecutionSpec{ShellType: rvboxv1.ShellType_SHELL_SH, Source: &rvboxv1.ExecutionSpec_CommandText{CommandText: "true"}}})
+	if err != nil {
+		t.Fatalf("restart after terminal = %v", err)
+	}
+	_, _ = second.Wait(context.Background())
+}
+
+func TestPortableSupervisorScriptMaterializationAndCleanup_HP_SUPERVISOR_04(t *testing.T) {
+	t.Parallel()
+	manager, err := NewSupervisor(NativeOptions{MaxWrapperBytes: 1024})
+	if err != nil {
+		t.Fatal(err)
+	}
+	issue, _ := domain.ParseUUIDv7("019c46f1-1d02-7000-8000-0000000000c2")
+	body := []byte("printf script-ok")
+	process, err := manager.Start(context.Background(), supervisor.StartSpec{IssueUUID: issue, CommandRevision: 1, WorkingDirectory: t.TempDir(), ScriptBody: body, Execution: &rvboxv1.ExecutionSpec{ShellType: rvboxv1.ShellType_SHELL_SH, Source: &rvboxv1.ExecutionSpec_Script{Script: &rvboxv1.ScriptDescriptor{SizeBytes: uint64(len(body)), Sha256: digest(body)}}}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := process.Wait(context.Background()); err != nil {
+		t.Fatal(err)
+	}
+	chunk, err := process.ReadOutput(context.Background())
+	if err != nil || string(chunk.Data) != "script-ok" {
+		t.Fatalf("script output = %+v, %v", chunk, err)
+	}
+	if _, err := process.ReadOutput(context.Background()); !errors.Is(err, io.EOF) {
+		t.Fatalf("script output close = %v", err)
+	}
+}
+
+func digest(value []byte) []byte {
+	result := sha256.Sum256(value)
+	return result[:]
+}