Function TestManagerForceRunOperations

internal/task/manager_test.go:6726–7240

(t *testing.T)

}

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

	t.Run(
		"Should force release claimed run and invalidate queued session input",
		func(t *testing.T) {
			t.Parallel()

			store := newForceInputStore()
			store.pendingCancelCounts["sess-force"] = 2
			manager := newTaskManagerForTest(t, store)
			actor := validActorContext()

			taskRecord, err := manager.CreateTask(context.Background(), CreateTask{
				Scope: ScopeGlobal,
				Title: "Force release",
			}, actor)
			if err != nil {
				t.Fatalf("CreateTask() error = %v", err)
			}
			run, err := manager.EnqueueRun(
				context.Background(),
				EnqueueRun{TaskID: taskRecord.ID},
				actor,
			)
			if err != nil {
				t.Fatalf("EnqueueRun() error = %v", err)
			}
			run, err = manager.ClaimRun(context.Background(), run.ID, ClaimRun{}, actor)
			if err != nil {
				t.Fatalf("ClaimRun() error = %v", err)
			}
			storedRun := store.runs[run.ID]
			storedRun.SessionID = "sess-force"
			store.runs[run.ID] = storedRun

			released, err := manager.ForceReleaseRun(context.Background(), run.ID, ForceReleaseRun{
				Reason: "handoff",
			}, actor)
			if err != nil {
				t.Fatalf("ForceReleaseRun() error = %v", err)
			}
			if got, want := released.Status, TaskRunStatusQueued; got != want {
				t.Fatalf("ForceReleaseRun().Status = %q, want %q", got, want)
			}
			if released.SessionID != "" || released.ClaimTokenHash != "" ||
				!released.LeaseUntil.IsZero() {
				t.Fatalf("released run retained claim fields: %#v", released)
			}
			if len(store.advanceCalls) != 1 || store.advanceCalls[0] != "sess-force" {
				t.Fatalf("advanceCalls = %#v, want sess-force", store.advanceCalls)
			}
			if len(store.cancelCalls) != 1 ||
				store.cancelCalls[0].SessionID != "sess-force" ||
				store.cancelCalls[0].Generation != 1 ||
				store.cancelCalls[0].CanceledInputs != 2 {
				t.Fatalf(
					"cancelCalls = %#v, want session generation cancellation",