DeepSeek-Reasonix/desktop/reload_runtime_test.go

package main

import (
	"context"
	"io"
	"os"
	"path/filepath"
	"slices"
	"sync"
	"testing"
	"time"

	"reasonix/internal/agent"
	"reasonix/internal/config"
	"reasonix/internal/control"
	"reasonix/internal/event"
	"reasonix/internal/jobs"
)

// reloadRuntimeFixture writes the config the ReloadRuntime tests share (one
// configured provider) and returns the isolated session dir.
func reloadRuntimeFixture(t *testing.T) string {
	t.Helper()
	isolateDesktopUserDirs(t)
	setDesktopTestCredential(t, "OLD_MODEL_KEY", "sk-test")

	cfg := config.Default()
	cfg.DefaultModel = "old/old-model"
	cfg.Desktop.ProviderAccess = []string{"old"}
	cfg.Providers = []config.ProviderEntry{
		{Name: "old", Kind: "openai", BaseURL: "https://example.invalid/v1", Model: "old-model", APIKeyEnv: "OLD_MODEL_KEY"},
	}
	if err := cfg.SaveTo(config.UserConfigPath()); err != nil {
		t.Fatalf("save config: %v", err)
	}

	dir := config.SessionDir()
	if err := os.MkdirAll(dir, 0o755); err != nil {
		t.Fatalf("mkdir session dir: %v", err)
	}
	return dir
}

func reloadRuntimeTab(t *testing.T, app *App, dir string, oldCtrl *control.Controller) *WorkspaceTab {
	t.Helper()
	tab := &WorkspaceTab{
		ID:          "tab_a",
		Scope:       "global",
		Ready:       true,
		model:       "old/old-model",
		Ctrl:        oldCtrl,
		sink:        &tabEventSink{tabID: "tab_a", app: app},
		disabledMCP: map[string]ServerView{},
	}
	app.tabs = map[string]*WorkspaceTab{tab.ID: tab}
	app.tabOrder = []string{tab.ID}
	app.activeTabID = tab.ID
	t.Cleanup(func() {
		tab.releaseSessionLease()
		if tab.Ctrl != nil {
			tab.Ctrl.Close()
		}
	})
	return tab
}

// TestReloadRuntimeSwapsAndClosesOldAfterSwap covers the success path through
// boot.Rebuild: the tab's controller is replaced, the session (grants, file)
// migrates, the outgoing controller is closed only after the swap published
// the replacement, and the frontend fence is emitted.
func TestReloadRuntimeSwapsAndClosesOldAfterSwap(t *testing.T) {
	dir := reloadRuntimeFixture(t)

	oldPath := filepath.Join(dir, "old.jsonl")
	oldExec := agent.New(nil, nil, agent.NewSession("old system prompt"), agent.Options{}, event.Discard)
	app := NewApp()
	app.ctx = context.Background()
	app.readyHook = func() {}
	var fenceMu sync.Mutex
	var fenceNames []string
	app.runtimeEvents.emit = func(_ context.Context, name string, _ ...any) {
		fenceMu.Lock()
		fenceNames = append(fenceNames, name)
		fenceMu.Unlock()
	}

	closed := false
	var ctrlAtClose control.SessionAPI
	oldCtrl := control.New(control.Options{
		Executor:    oldExec,
		SessionDir:  dir,
		SessionPath: oldPath,
		Label:       "old",
		Sink:        event.Discard,
		Cleanup: func() {
			closed = true
			app.mu.RLock()
			ctrlAtClose = app.tabs["tab_a"].Ctrl
			app.mu.RUnlock()
		},
	})
	oldCtrl.RestoreSessionAuthorizations(control.SessionAuthorizations{
		Grants:                   []string{"bash|go test ./..."},
		PlanModeReadOnlyCommands: []string{"go test ./..."},
	})
	tab := reloadRuntimeTab(t, app, dir, oldCtrl)

	if err := app.ReloadRuntime(tab.ID); err != nil {
		t.Fatalf("ReloadRuntime: %v", err)
	}

	newCtrl, ok := tab.Ctrl.(*control.Controller)
	if !ok {
		t.Fatalf("tab.Ctrl = %T, want *control.Controller", tab.Ctrl)
	}
	if newCtrl == oldCtrl {
		t.Fatal("ReloadRuntime kept the outgoing controller installed")
	}
	if got := newCtrl.SessionPath(); got != oldPath {
		t.Fatalf("session path = %q, want the carried session file %q", got, oldPath)
	}
	got := newCtrl.SessionAuthorizations()
	if len(got.Grants) != 1 || got.Grants[0] != "bash|go test ./..." {
		t.Fatalf("migrated grants = %+v, want [\"bash|go test ./...\"]", got.Grants)
	}
	if !closed {
		t.Fatal("outgoing controller was not closed")
	}
	if ctrlAtClose != newCtrl {
		t.Fatal("outgoing controller closed before the swap published the replacement")
	}
	deadline := time.Now().Add(2 * time.Second)
	for {
		fenceMu.Lock()
		found := slices.Contains(fenceNames, "runtime:rebuilt")
		fenceMu.Unlock()
		if found {
			break
		}
		if time.Now().After(deadline) {
			t.Fatal("no runtime:rebuilt fence emitted")
		}
		time.Sleep(5 * time.Millisecond)
	}
}

// TestReloadRuntimeFailureKeepsOldController: a failed build leaves the tab on
// the outgoing controller, which stays open.
func TestReloadRuntimeFailureKeepsOldController(t *testing.T) {
	isolateDesktopUserDirs(t)

	// No providers at all: the build cannot resolve the tab's model.
	cfg := config.Default()
	cfg.DefaultModel = ""
	cfg.Providers = []config.ProviderEntry{}
	if err := cfg.SaveTo(config.UserConfigPath()); err != nil {
		t.Fatalf("save config: %v", err)
	}
	dir := config.SessionDir()
	if err := os.MkdirAll(dir, 0o755); err != nil {
		t.Fatalf("mkdir session dir: %v", err)
	}

	closed := false
	oldCtrl := control.New(control.Options{
		SessionDir:  dir,
		SessionPath: filepath.Join(dir, "old.jsonl"),
		Label:       "old",
		Sink:        event.Discard,
		Cleanup:     func() { closed = true },
	})
	app := NewApp()
	app.ctx = context.Background()
	app.readyHook = func() {}
	tab := reloadRuntimeTab(t, app, dir, oldCtrl)

	if err := app.ReloadRuntime(tab.ID); err == nil {
		t.Fatal("ReloadRuntime with an unresolvable model returned nil error")
	}
	if tab.Ctrl != oldCtrl {
		t.Fatal("failed reload replaced the tab controller")
	}
	if closed {
		t.Fatal("failed reload closed the outgoing controller")
	}
	if app.deferredRebuildPending(tab.ID) {
		t.Fatal("hard failure was queued for retry")
	}
}

// TestReloadRuntimeBusyQueuesDeferred covers the busy contract: active work
// queues exactly one reload on the deferred-rebuild loop (coalesced), and the
// retry runs the boot.Rebuild reload once the tab is idle.
func TestReloadRuntimeBusyQueuesDeferred(t *testing.T) {
	dir := reloadRuntimeFixture(t)

	jm := jobs.NewManager(event.Discard)
	releaseJob := make(chan struct{})
	jm.Start("test", "blocking job", func(ctx context.Context, _ io.Writer) (string, error) {
		select {
		case <-releaseJob:
			return "", nil
		case <-ctx.Done():
			return "", ctx.Err()
		}
	})
	oldCtrl := control.New(control.Options{
		SessionDir:  dir,
		SessionPath: filepath.Join(dir, "old.jsonl"),
		Label:       "old",
		Sink:        event.Discard,
		Jobs:        jm,
	})
	app := NewApp()
	app.ctx = context.Background()
	app.readyHook = func() {}
	tab := reloadRuntimeTab(t, app, dir, oldCtrl)

	if err := app.ReloadRuntime(tab.ID); err != nil {
		t.Fatalf("busy ReloadRuntime returned %v, want nil (queued)", err)
	}
	if tab.Ctrl != oldCtrl {
		t.Fatal("busy reload swapped the controller")
	}
	if !app.deferredRebuildPending(tab.ID) {
		t.Fatal("busy reload was not queued on the deferred loop")
	}
	// A second request while busy coalesces into the same queued reload.
	if err := app.ReloadRuntime(tab.ID); err != nil {
		t.Fatalf("second busy ReloadRuntime returned %v, want nil", err)
	}
	app.deferredRebuild.mu.Lock()
	label, ok := app.deferredRebuild.pending[tab.ID]
	count := len(app.deferredRebuild.pending)
	app.deferredRebuild.mu.Unlock()
	if !ok || label != deferredRuntimeReloadLabel {
		t.Fatalf("pending label = %q (present=%t), want %q", label, ok, deferredRuntimeReloadLabel)
	}
	if count != 1 {
		t.Fatalf("pending entries = %d, want exactly 1 (coalesced)", count)
	}

	// The work finishes; the retry path reloads the now-idle tab.
	close(releaseJob)
	deadline := time.Now().Add(2 * time.Second)
	for oldCtrl.RuntimeStatus().BackgroundJobs > 0 {
		if time.Now().After(deadline) {
			t.Fatal("background job did not finish")
		}
		time.Sleep(5 * time.Millisecond)
	}
	app.retryDeferredRuntimeReload(tab.ID, tab)
	if tab.Ctrl == oldCtrl {
		t.Fatal("deferred retry did not reload the idle tab")
	}
	if app.deferredRebuildPending(tab.ID) {
		t.Fatal("deferred entry survived a successful retry")
	}
}