Method createBuild

coderd/telemetry/telemetry_test.go:710–753

createBuild creates a workspace build with the given parameters, handling provisioner job creation automatically.

(
	resp dbfake.WorkspaceResponse,
	buildNumber int32,
	createdAt time.Time,
	transition database.WorkspaceTransition,
	reason database.BuildReason,
)

// createBuild creates a workspace build with the given parameters,
// handling provisioner job creation automatically.
func (h *taskTelemetryHelper) createBuild(
	resp dbfake.WorkspaceResponse,
	buildNumber int32,
	createdAt time.Time,
	transition database.WorkspaceTransition,
	reason database.BuildReason,
) (database.WorkspaceBuild, *database.WorkspaceApp) {
	job := dbgen.ProvisionerJob(h.t, h.db, nil, database.ProvisionerJob{
		Provisioner:    database.ProvisionerTypeTerraform,
		StorageMethod:  database.ProvisionerStorageMethodFile,
		Type:           database.ProvisionerJobTypeWorkspaceBuild,
		OrganizationID: h.org.ID,
	})
	bld := dbgen.WorkspaceBuild(h.t, h.db, database.WorkspaceBuild{
		WorkspaceID:       resp.Workspace.ID,
		TemplateVersionID: resp.TemplateVersion.ID,
		JobID:             job.ID,
		Transition:        transition,
		Reason:            reason,
		BuildNumber:       buildNumber,
		CreatedAt:         createdAt,
		HasAITask: sql.NullBool{
			Bool:  true,
			Valid: true,
		},
	})
	if transition == database.WorkspaceTransitionStart {
		require.NotEmpty(h.t, resp.Agents, "need at least one agent")
		agt := resp.Agents[0]
		// App IDs are regenerated by provisionerd each build.
		app := dbgen.WorkspaceApp(h.t, h.db, database.WorkspaceApp{
			AgentID: agt.ID,
		})
		_, err := h.db.UpsertTaskWorkspaceApp(h.ctx, database.UpsertTaskWorkspaceAppParams{
			TaskID:               resp.Task.ID,
			WorkspaceBuildNumber: buildNumber,
			WorkspaceAgentID:     uuid.NullUUID{UUID: agt.ID, Valid: true},
			WorkspaceAppID:       uuid.NullUUID{UUID: app.ID, Valid: true},
		})
		require.NoError(h.t, err, "failed to upsert task app")
		return bld, &app
	}
	return bld, nil
}

// nolint: dupl // Test code is better WET than DRY.
func TestTasksTelemetry(t *testing.T) {