Function setupCLITaskTest

cli/task_test.go:291–340

(ctx context.Context, t *testing.T, agentAPIHandlers map[string]http.HandlerFunc)

}

func setupCLITaskTest(ctx context.Context, t *testing.T, agentAPIHandlers map[string]http.HandlerFunc) setupCLITaskTestResult {
	t.Helper()

	ownerClient := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})
	owner := coderdtest.CreateFirstUser(t, ownerClient)
	userClient, _ := coderdtest.CreateAnotherUser(t, ownerClient, owner.OrganizationID)

	fakeAPI := startFakeAgentAPI(t, agentAPIHandlers)

	authToken := uuid.NewString()
	template := createAITaskTemplate(t, ownerClient, owner.OrganizationID, withSidebarURL(fakeAPI.URL()), withAgentToken(authToken))

	wantPrompt := "test prompt"
	task, err := userClient.CreateTask(ctx, codersdk.Me, codersdk.CreateTaskRequest{
		TemplateVersionID: template.ActiveVersionID,
		Input:             wantPrompt,
		Name:              "test-task",
	})
	require.NoError(t, err)

	// Wait for the task's underlying workspace to be built.
	require.True(t, task.WorkspaceID.Valid, "task should have a workspace ID")
	workspace, err := userClient.Workspace(ctx, task.WorkspaceID.UUID)
	require.NoError(t, err)
	coderdtest.AwaitWorkspaceBuildJobCompleted(t, userClient, workspace.LatestBuild.ID)

	agentClient := agentsdk.New(userClient.URL, agentsdk.WithFixedToken(authToken))
	agt := agenttest.New(t, userClient.URL, authToken, func(o *agent.Options) {
		o.Client = agentClient
	})

	coderdtest.NewWorkspaceAgentWaiter(t, userClient, workspace.ID).
		WaitFor(coderdtest.AgentsReady)

	// Report the task app as idle so that waitForTaskIdle can proceed.
	err = agentClient.PatchAppStatus(ctx, agentsdk.PatchAppStatus{
		AppSlug: "task-sidebar",
		State:   codersdk.WorkspaceAppStatusStateIdle,
		Message: "ready",
	})
	require.NoError(t, err)

	return setupCLITaskTestResult{
		ownerClient: ownerClient,
		userClient:  userClient,
		task:        task,
		agentToken:  authToken,
		agent:       agt,
	}
}

// pauseTask pauses the task and waits for the stop build to complete.
func pauseTask(ctx context.Context, t *testing.T, client *codersdk.Client, task codersdk.Task) {