Function TestTaskBlockCommandsMapRequests

internal/cli/task_test.go:342–667

(t *testing.T)

}

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

	t.Run("Should parse block request and render stable JSON", func(t *testing.T) {
		t.Parallel()

		var (
			taskID       string
			blockRequest CreateTaskBlockRequest
		)
		deps := newTestDeps(t, &stubClient{
			blockTaskFn: func(_ context.Context, id string, request CreateTaskBlockRequest) (TaskBlockRecord, error) {
				taskID = id
				blockRequest = request
				return sampleTaskBlockRecord(), nil
			},
		})

		stdout, _, err := executeRootCommand(
			t,
			deps,
			"task",
			"block",
			"task-1",
			"--kind",
			"transient",
			"--reason",
			"waiting for quota",
			"--details",
			`{"queue":"gpu"}`,
			"--expires-in",
			"30m",
			"-o",
			"json",
		)
		if err != nil {
			t.Fatalf("task block error = %v", err)
		}
		if taskID != "task-1" ||
			blockRequest.Kind != taskpkg.BlockKindTransient ||
			blockRequest.Reason != "waiting for quota" ||
			blockRequest.RunID != "" ||
			string(blockRequest.Details) != `{"queue":"gpu"}` ||
			blockRequest.ExpiresAt == nil ||
			!blockRequest.ExpiresAt.Equal(fixedTestNow.Add(30*time.Minute).UTC()) {
			t.Fatalf("block request task=%q request=%#v, want parsed transient block", taskID, blockRequest)
		}
		var output TaskBlockRecord
		if err := json.Unmarshal([]byte(stdout), &output); err != nil {
			t.Fatalf("json.Unmarshal(task block) error = %v", err)
		}
		if output.ID != "block-1" || output.TaskID != "task-1" {
			t.Fatalf("task block output = %#v, want block-1/task-1", output)
		}
	})

	t.Run("Should reject run id without agent identity before calling the daemon", func(t *testing.T) {
		t.Parallel()