Function waitForBuild

scaletest/autostart/run.go:201–231

waitForBuild waits for a build with the given transition to reach a terminal state. It returns nil on success, or an error if the build fails, is canceled, or the context expires. If an unexpected transition is received, it returns an error immediately.

(ctx context.Context, logger slog.Logger, updates <-chan codersdk.WorkspaceBuildUpdate, transition codersdk.WorkspaceTransition)

// fails, is canceled, or the context expires. If an unexpected transition
// is received, it returns an error immediately.
func waitForBuild(ctx context.Context, logger slog.Logger, updates <-chan codersdk.WorkspaceBuildUpdate, transition codersdk.WorkspaceTransition) error {
	for {
		select {
		case <-ctx.Done():
			return ctx.Err()
		case update, ok := <-updates:
			if !ok {
				return xerrors.New("build updates channel closed")
			}
			logger.Debug(ctx, "received build update",
				slog.F("transition", update.Transition),
				slog.F("job_status", update.JobStatus),
				slog.F("build_number", update.BuildNumber))

			if update.Transition != string(transition) {
				return xerrors.Errorf("unexpected transition: expected %s, got %s (build_number=%d)", transition, update.Transition, update.BuildNumber)
			}
			switch codersdk.ProvisionerJobStatus(update.JobStatus) {
			case codersdk.ProvisionerJobSucceeded:
				return nil
			case codersdk.ProvisionerJobFailed:
				return xerrors.Errorf("workspace build failed (transition=%s, build_number=%d)", update.Transition, update.BuildNumber)
			case codersdk.ProvisionerJobCanceled:
				return xerrors.Errorf("workspace build canceled (transition=%s, build_number=%d)", update.Transition, update.BuildNumber)
			default:
				// Intermediate states (pending, running, canceling)
				// are expected; keep waiting.
			}
		}
	}
}

func (r *Runner) Cleanup(ctx context.Context, id string, logs io.Writer) error {
	if r.workspacebuildRunner != nil {