Function TestAgentTaskCommandsMapLeaseRequests

internal/cli/task_test.go:1090–1288

(t *testing.T)

}

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

	t.Run("Should map task next lease request", func(t *testing.T) {
		t.Parallel()

		var gotRequest AgentTaskClaimNextRequest
		deps := newAgentCommandTestDeps(t, &stubClient{
			agentTaskClaimNextFn: func(
				_ context.Context,
				request AgentTaskClaimNextRequest,
				credentials agentidentity.Credentials,
			) (AgentTaskNextRecord, error) {
				assertAgentCredentials(t, credentials)
				gotRequest = request
				return AgentTaskNextRecord{
					Claimed: true,
					Claim:   new(agentTaskClaimRecord()),
				}, nil
			},
		})

		stdout, _, err := executeRootCommand(
			t,
			deps,
			"task",
			"next",
			"--workspace-id",
			"ws-1",
			"--capability",
			"go",
			"--capability",
			"   ",
			"--capability",
			"typescript",
			"--priority-min",
			"3",
			"--lease-seconds",
			"120",
			"--wait",
			"--idempotency-key",
			"idem-next",
			"-o",
			"json",
		)
		if err != nil {
			t.Fatalf("task next error = %v", err)
		}
		if gotRequest.WorkspaceID != "ws-1" ||
			gotRequest.PriorityMin != 3 ||
			gotRequest.LeaseSeconds != 120 ||
			!gotRequest.Wait ||
			gotRequest.IdempotencyKey != "idem-next" ||
			len(gotRequest.RequiredCapabilities) != 2 ||
			gotRequest.RequiredCapabilities[0] != "go" ||
			gotRequest.RequiredCapabilities[1] != "typescript" {
			t.Fatalf("next request = %#v, want parsed agent claim request", gotRequest)
		}