Function TestManagerTerminalRunStopsBackingSession

internal/task/manager_test.go:6079–6219

(t *testing.T)

}

func TestManagerTerminalRunStopsBackingSession(t *testing.T) {
	t.Parallel()

	t.Run("Should stop completed run session", func(t *testing.T) {
		t.Parallel()

		store := newInMemoryManagerStore()
		executor := &recordingSessionExecutor{}
		manager := newTaskManagerForTestWithOptions(
			t,
			store,
			WithSessionExecutor(executor),
			WithCancelGracePeriod(0),
		)
		actor := validActorContext()
		runningRun := createRunningRunForTest(t, manager, actor)

		if _, err := manager.CompleteRun(context.Background(), runningRun.ID, RunResult{
			Value: json.RawMessage(`{"ok":true}`),
		}, actor); err != nil {
			t.Fatalf("CompleteRun() error = %v", err)
		}

		assertSessionStopCalls(t, executor, runningRun.SessionID, StopReasonCompleted)
	})

	t.Run("Should stop failed run session", func(t *testing.T) {
		t.Parallel()

		store := newInMemoryManagerStore()
		executor := &recordingSessionExecutor{}
		manager := newTaskManagerForTestWithOptions(
			t,
			store,
			WithSessionExecutor(executor),
			WithCancelGracePeriod(0),
		)
		actor := validActorContext()
		runningRun := createRunningRunForTest(t, manager, actor)

		if _, err := manager.FailRun(context.Background(), runningRun.ID, RunFailure{
			Error: "release validation failed",
		}, actor); err != nil {
			t.Fatalf("FailRun() error = %v", err)
		}

		assertSessionStopCalls(t, executor, runningRun.SessionID, StopReasonFailed)
	})

	t.Run("Should keep run non-terminal when completed stop request fails", func(t *testing.T) {
		t.Parallel()

		assertTerminalStopFailureLeavesRunActive(t, errors.New("stop failed"), nil, func(
			ctx context.Context,
			manager *Service,
			runID string,
			actor ActorContext,
		) error {