Function TestPauseTask

coderd/aitasks_test.go:2469–2833

(t *testing.T)

}

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

	setupClient := func(t *testing.T, db database.Store, ps pubsub.Pubsub, authorizer rbac.Authorizer) *codersdk.Client {
		t.Helper()
		client, _, _ := coderdtest.NewWithAPI(t, &coderdtest.Options{
			Database:   db,
			Pubsub:     ps,
			Authorizer: authorizer,
		})
		return client
	}

	setupWorkspaceTask := func(t *testing.T, db database.Store, user codersdk.CreateFirstUserResponse) (database.Task, uuid.UUID) {
		t.Helper()
		workspaceBuild := dbfake.WorkspaceBuild(t, db, database.WorkspaceTable{
			OrganizationID: user.OrganizationID,
			OwnerID:        user.UserID,
		}).WithTask(database.TaskTable{
			Prompt: "pause me",
		}, nil).Do()
		return workspaceBuild.Task, workspaceBuild.Workspace.ID
	}

	t.Run("OK", func(t *testing.T) {
		t.Parallel()

		ctx := testutil.Context(t, testutil.WaitLong)
		client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})
		user := coderdtest.CreateFirstUser(t, client)

		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
			Parse:          echo.ParseComplete,
			ProvisionApply: echo.ApplyComplete,
			ProvisionGraph: []*proto.Response{
				{Type: &proto.Response_Graph{Graph: &proto.GraphComplete{
					HasAiTasks: true,
				}}},
			},
		})
		coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID)
		template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)

		task, err := client.CreateTask(ctx, codersdk.Me, codersdk.CreateTaskRequest{
			TemplateVersionID: template.ActiveVersionID,
			Input:             "pause me",
		})
		require.NoError(t, err)
		require.True(t, task.WorkspaceID.Valid)

		workspace, err := client.Workspace(ctx, task.WorkspaceID.UUID)
		require.NoError(t, err)
		coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID)

		resp, err := client.PauseTask(ctx, codersdk.Me, task.ID)

		// Verify that the request was accepted correctly:
		require.NoError(t, err)