supervisor_launch_test.go

package aicliobserve

import (
	"context"
	"errors"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"
)

func TestLaunchUnsupportedAdapter(t *testing.T) {
	s := NewSupervisor().(*supervisor)
	_, err := s.Launch(context.Background(), LaunchRequest{Adapter: "unknown"})
	if !errors.Is(err, ErrUnsupportedCLI) {
		t.Fatalf("expected unsupported error, got %v", err)
	}
}

func TestLaunchAndReceiveEvents(t *testing.T) {
	binDir := t.TempDir()
	script := filepath.Join(binDir, "codex")
	content := "#!/bin/sh\necho permission required\nsleep 1\necho agent-turn-complete\nsleep 1\n"
	if err := os.WriteFile(script, []byte(content), 0o755); err != nil {
		t.Fatal(err)
	}
	t.Setenv("PATH", binDir+string(os.PathListSeparator)+os.Getenv("PATH"))

	s := NewSupervisor().(*supervisor)
	ctx, cancel := context.WithTimeout(context.Background(), 8*time.Second)
	defer cancel()

	subCtx, subCancel := context.WithCancel(context.Background())
	defer subCancel()
	ch, err := s.Subscribe(subCtx)
	if err != nil {
		t.Fatal(err)
	}

	resp, err := s.Launch(ctx, LaunchRequest{Adapter: AdapterCodex})
	if err != nil {
		t.Fatal(err)
	}
	if resp.Mode != ModePTYFallback {
		t.Fatalf("unexpected mode: %s", resp.Mode)
	}

	seenStart := false
	seenTerminal := false
	deadline := time.After(5 * time.Second)
	for !(seenStart && seenTerminal) {
		select {
		case e, ok := <-ch:
			if !ok {
				t.Fatalf("subscriber channel closed before expected events")
			}
			if e.SessionID != resp.SessionID {
				continue
			}
			switch e.State {
			case StateStarting:
				seenStart = true
			case StateExited, StateError:
				seenTerminal = true
			}
		case <-deadline:
			t.Fatalf("missing deterministic states start=%v terminal=%v", seenStart, seenTerminal)
		}
	}
}

func TestStopSessionKillsProcess(t *testing.T) {
	binDir := t.TempDir()
	script := filepath.Join(binDir, "codex")
	content := "#!/bin/sh\necho started\nsleep 20\n"
	if err := os.WriteFile(script, []byte(content), 0o755); err != nil {
		t.Fatal(err)
	}
	t.Setenv("PATH", strings.Join([]string{binDir, os.Getenv("PATH")}, string(os.PathListSeparator)))

	s := NewSupervisor().(*supervisor)
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	resp, err := s.Launch(ctx, LaunchRequest{Adapter: AdapterCodex})
	if err != nil {
		t.Fatal(err)
	}
	if err := s.StopSession(context.Background(), resp.SessionID); err != nil {
		t.Fatal(err)
	}
}