Function assertTerminalStopFailureLeavesRunActive

internal/task/manager_test.go:6281–6333

(
	t *testing.T,
	requestStopErr error,
	forceStopErr error,
	transition func(context.Context, *Service, string, ActorContext) error,
)

}

func assertTerminalStopFailureLeavesRunActive(
	t *testing.T,
	requestStopErr error,
	forceStopErr error,
	transition func(context.Context, *Service, string, ActorContext) error,
) {
	t.Helper()

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

	err := transition(context.Background(), manager, runningRun.ID, actor)
	if err == nil {
		t.Fatal("transition() error = nil, want stop failure")
	}

	storedRun, getRunErr := store.GetTaskRun(context.Background(), runningRun.ID)
	if getRunErr != nil {
		t.Fatalf("GetTaskRun() error = %v", getRunErr)
	}
	if got, want := storedRun.Status, TaskRunStatusRunning; got != want {
		t.Fatalf("storedRun.Status = %q, want %q", got, want)
	}
	if !storedRun.EndedAt.IsZero() {
		t.Fatalf("storedRun.EndedAt = %v, want zero", storedRun.EndedAt)
	}
	if got, want := store.tasks[storedRun.TaskID].Status, TaskStatusInProgress; got != want {
		t.Fatalf("task.Status = %q, want %q", got, want)
	}
	if len(executor.requestStopCalls) != 1 {
		t.Fatalf("len(requestStopCalls) = %d, want 1", len(executor.requestStopCalls))
	}
	if requestStopErr != nil {
		if len(executor.forceStopCalls) != 0 {
			t.Fatalf("len(forceStopCalls) = %d, want 0", len(executor.forceStopCalls))
		}
		return
	}
	if len(executor.forceStopCalls) != 1 {
		t.Fatalf("len(forceStopCalls) = %d, want 1", len(executor.forceStopCalls))
	}
}

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