Function TestManagerTaskBlockAgentSessionScope

internal/task/manager_test.go:4684–4801

(t *testing.T)

}

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

	t.Run(
		"Should constrain agent-session block lifecycle to the active leased task",
		func(t *testing.T) {
			t.Parallel()

			ctx := context.Background()
			store := newInMemoryManagerStore()
			manager := newTaskManagerForTest(t, store)
			operator := validActorContext()
			agent := agentSessionActorContext("sess-block-scope")

			leasedTask, err := manager.CreateTask(
				ctx,
				CreateTask{Scope: ScopeGlobal, Title: "leased task"},
				operator,
			)
			if err != nil {
				t.Fatalf("CreateTask(leased) error = %v", err)
			}
			foreignTask, err := manager.CreateTask(
				ctx,
				CreateTask{Scope: ScopeGlobal, Title: "foreign task"},
				operator,
			)
			if err != nil {
				t.Fatalf("CreateTask(foreign) error = %v", err)
			}
			if _, err := manager.EnqueueRun(ctx, EnqueueRun{TaskID: leasedTask.ID}, operator); err != nil {
				t.Fatalf("EnqueueRun(leased) error = %v", err)
			}
			claim, err := manager.ClaimNextRun(ctx, ClaimCriteria{
				Scope:            ScopeGlobal,
				ClaimerSessionID: "sess-block-scope",
				LeaseDuration:    time.Minute,
			}, agent)
			if err != nil {
				t.Fatalf("ClaimNextRun() error = %v", err)
			}

			leasedBlock, err := manager.BlockTask(ctx, BlockRequest{
				TaskID: leasedTask.ID,
				Kind:   BlockKindNeedsInput,
				Reason: "missing input",
			}, agent)
			if err != nil {
				t.Fatalf("BlockTask(leased agent) error = %v", err)
			}
			blocks, err := manager.ListTaskBlocks(ctx, leasedTask.ID, true, agent)
			if err != nil {
				t.Fatalf("ListTaskBlocks(leased agent) error = %v", err)
			}
			if got, want := len(blocks), 1; got != want {
				t.Fatalf("len(leased blocks) = %d, want %d", got, want)
			}
			if blocks[0].ID != leasedBlock.ID {